web-dev-qa-db-ja.com

Python 2つのリストをすべての可能な順列とマージする

2つのリストをすべての可能な組み合わせにマージするための最良の方法を見つけようとしています。したがって、次のような2つのリストから始めると、

list1 = [1, 2]
list2 = [3, 4]

結果のリストは次のようになります。

[[[1,3], [2,4]], [[1,4], [2,3]]]

つまり、基本的にリストのリストを作成し、2つの間のすべての潜在的な組み合わせを作成します。

私はitertoolsを使って作業してきましたが、これは答えを保持していると確信していますが、このように動作させる方法は思いつきません。一番近いのは：

list1 = [1, 2, 3, 4]
list2 = [5, 6, 7, 8]
print list(itertools.product(list1, list2))

生産したもの：

[(1, 5), (1, 6), (1, 7), (1, 8), (2, 5), (2, 6), (2, 7), (2, 8), (3, 5), (3, 6), (3, 7), (3, 8), (4, 5), (4, 6), (4, 7), (4, 8)]

そのため、各リスト内のアイテムの可能な組み合わせはすべて実行されますが、結果のリストのすべては実行されません。どうすればそれを実現できますか？

編集：最終目標は、各リストを個別に処理して効率を判断できるようにすることです（実際に作業しているデータはより複雑です）。したがって、上記の元の例では、次のように機能します。

list1 = [1, 2]
list2 = [3, 4]

Get first merged list: [[1,3], [2, 4]]
    Do stuff with this list
Get second merged list: [[1,4], [2, 3]]
    Do stuff with this list

上記の「リストのリストのリスト」出力を取得した場合、それをforループに入れて処理することができます。他の形式の出力でも機能しますが、最も簡単に機能するようです。

pythonlistitertools
17
GeoJunkie

repeat最初のリスト、permutate 2番目、Zipすべて一緒

>>> from itertools import permutations, repeat
>>> a = [1, 2, 3]
>>> b = [4, 5, 6]
>>> list(list(Zip(r, p)) for (r, p) in Zip(repeat(a), permutations(b)))
[[(1, 4), (2, 5), (3, 6)],
 [(1, 4), (2, 6), (3, 5)],
 [(1, 5), (2, 4), (3, 6)],
 [(1, 5), (2, 6), (3, 4)],
 [(1, 6), (2, 4), (3, 5)],
 [(1, 6), (2, 5), (3, 4)]]

[〜＃〜] edit [〜＃〜]：Peter Ottenが指摘したように、内部のZipとrepeatは不要です。

[list(Zip(a, p)) for p in permutations(b)]
21
pacholik

受け入れられた答えは、次のように簡略化できます。

a = [1, 2, 3]
b = [4, 5, 6]
[list(Zip(a, p)) for p in permutations(b)]

（list（）呼び出しは、Python 2）では省略できます）

12
Peter Otten

リストジェネレーターを使用して、ネストされたリストを作成してみてください。

>>> [[[x,y] for x in list1] for y in list2]
[[[1, 3], [2, 3]], [[1, 4], [2, 4]]]
>>>

または、1行のリストが必要な場合は、括弧を削除するだけです。

>>> [[x,y] for x in list1 for y in list2]
[[1, 3], [1, 4], [2, 3], [2, 4]]
8
Sdwdaw

目的の出力が得られるようにコードを編集しました。

list1 = [1,2]
list2 = [3,4]
combined = []

for a in list1:
    new_list = []
    for b in list2:
        new_list.append([a, b])
    combined.append(new_list)

print combined
3
Joe Smart

これを使用して、リストの組み合わせを含む2つのリストメンバーのすべての順列を作成して、リストを作成できます。

lst1 = [1,2]
lst2 = [3,4]

#lst = [[j,k] for j in lst1 for k in lst2] # [[1,3],[1,4],[2,3],[2,4]]
lst = [[[j,k] for j in lst1] for k in lst2] # [[[1,3],[2,3]],[[1,4],[2,4]]]
print lst
2
Semih Yagcioglu

@pacholikの答えは異なる長さのリストをカバーしていないので、2つの変数を持つリスト内包表記を使用した私の解決策があります：

first_list = [1, 2, 3]
second_list = ['a', 'b']

combinations = [(a,b) for a in first_list for b in second_list]

出力は次のようになります。

[(1, 'a'), (1, 'b'), (2, 'a'), (2, 'b'), (3, 'a'), (3, 'b')]
2
Minato

これを試して：

combos=[]
for i in list1:
      for j in list2:
          combos.append([i,j])
print combos
1
pydude