pythonでリスト内のリストでの組み合わせ

Question

python初学者です

リスト内での組み合わせを求めたいのですが例えば、

[['3', '4', '1', '5', '6', '2'], ['0', '1', '7', '5', '2', '1'], ['5', '1', '3', '0', '8', '4']]

このようなリストがあるとき、二番目のリストは無視して一番目と三番目のリストから3つずつ数字を取り出して組み合わせを求めたいのですがどのような方法があるでしょうか？

またリスト内で取り出す数字を制限することはできるでしょうか？
例えば1番目のリストから数字を取り出すとき隣り合った数字からしか取り出せないようにしたい時、どのような方法が考えられますか

例
[3, 4, 1], [4, 1, 5],[1, 5, 6]...

Answer 1

３つの異なる要素の組み合わせをすべて作るには、３重のforループを使います。

l0 = range(5)   # [0, 1, 2, 3, 4]
len0 = len(l0)

for i in range(len0):
    for j in range(i + 1, len0):
        for k in range(j + 1, len0):
            print(l0[i],l0[j],l0[k])
# 0 1 2
# 0 1 3
# 0 1 4
# ...
# 2 3 4

２つのリストからそれぞれ３つ取り出すとなると６重ループになり少々見苦しいので、generator functionを定義するとすっきりします。

ls = [['3', '4', '1', '5', '6', '2'], ['0', '1', '7', '5', '2', '1'], ['5', '1', '3', '0', '8', '4']]

def triples(lst):
    n = len(lst)
    for i in range(n):
        for j in range(i + 1, n):
            for k in range(j + 1, n):
                yield (lst[i], lst[j], lst[k])

for x, y, z in triples(ls[0]):
    for a, b, c in triples(ls[2]):
        print(f'({x},{y},{z}),({a},{b},{c})')
# (3,4,1),(5,1,3)
# (3,4,1),(5,1,0)
# ...
# (5,6,2),(0,8,4)

itertoolsライブラリのcombinations関数を使っても同じことができます。

from itertools import combinations

ls = [['3', '4', '1', '5', '6', '2'], ['0', '1', '7', '5', '2', '1'], ['5', '1', '3', '0', '8', '4']]

for x, y, z in combinations(ls[0], 3):
    for a, b, c in combinations(ls[2], 3):
        print(f'({x},{y},{z}),({a},{b},{c})')
# (3,4,1),(5,1,3)
# (3,4,1),(5,1,0)
# ...
# (5,6,2),(0,8,4)

---

例えば1番目のリストから数字を取り出すとき隣り合った数字からしか取り出せないようにしたい時、どのような方法が考えられますか

他の回答者さんがすでにお答えになっているので省略します。

Answer 2

このようにできると思います。

>>> a = [['3', '4', '1', '5', '6', '2'], ['0', '1', '7', '5', '2', '1'], ['5', '1', '3', '0', '8', '4']]
>>> print([[a[0][i:i+3],a[2][i:i+3]] for i in range(4)])
[[['3', '4', '1'], ['5', '1', '3']], [['4', '1', '5'], ['1', '3', '0']], [['1', '5', '6'], ['3', '0', '8']], [['5', '6', '2'], ['0', '8', '4']]]

説明すると思います。
まずはpythonのリスト内包表記という機能って分かりますか？
上のコードと下のコードは同じです。

b = []
for i in range(4):
    b.append([a[0][i:i+3],a[2][i:i+3]])
print(b)

リスト内包表記で一行だけで上のコードができますよ。

Answer 3

隣接するものだけでいいのならば数学的な意味でのコンビネーションではないですよね?

ならばこうでしょうか?

lists = [['3', '4', '1', '5', '6', '2'],
         ['0', '1', '7', '5', '2', '1'],
         ['5', '1', '3', '0', '8', '4']]

results = []

for i in (0,2):
    for j in range(len(lists[i])-2):
        results.append([lists[i][j],lists[i][j+1],lists[i][j+2]])

print(results)

結果

[['3', '4', '1'], ['4', '1', '5'], ['1', '5', '6'], ['5', '6', '2'], ['5', '1', '3'], ['1', '3', '0'], ['3', '0', '8'], ['0', '8', '4']]

もう少し短くできるかもしれませんけど。。(^ ^;

追記：

6つの数字をどう出力するか(2次元リスト?一つのリスト?)も書いてほしかったです。。(^ ^;

コンビネーションはitertoolsで作れます。

from itertools import combinations

lists = [['3', '4', '1', '5', '6', '2'],
         ['0', '1', '7', '5', '2', '1'],
         ['5', '1', '3', '0', '8', '4']]

for c0 in combinations(lists[0],3):
    for c2 in combinations(lists[2],3):
        print(list(c0+c2))

結果は長いので省略します。

隣接のほうですけど、lists[0]の数字とlists[2]の数字はもちろん隣接していませんよね?
それを同時に出力して意味があるのでしょうか..?

lists = [['3', '4', '1', '5', '6', '2'],
         ['0', '1', '7', '5', '2', '1'],
         ['5', '1', '3', '0', '8', '4']]

adjoin_results = []

for j in range(len(lists[0])-2):
    adjoin_results.append([lists[0][j],lists[0][j+1],lists[0][j+2],
                           lists[2][j],lists[2][j+1],lists[2][j+2]])

print(adjoin_results)

結果

[['3', '4', '1', '5', '1', '3'], ['4', '1', '5', '1', '3', '0'], ['1', '5', '6', '3', '0', '8'], ['5', '6', '2', '0', '8', '4']]

Comments

[退会済みユーザー]

1番目と3番目のリストから3つづつ、つまり合わせると6つの数字で結果を出したいです
質問がわかりにくくてすみません・・・
また、隣接するものだけの時とそれ以外の場合とでそれぞれ回答をいただけるとありがたいです
質問がわかりにくくて申し訳ないです