Pythonで複数の配列の全ての組み合わせを取得したい

Question

Pythonで複数の配列をかけ合わせた全ての組み合わせを取得したいです。
どのように記述すればアウトプットが得られるでしょうか。

#入力データ
ramen = ["豚骨","醤油"]
station = ["新宿","渋谷","池袋"]

##実現したいアウトプット
豚骨 新宿
豚骨 渋谷
豚骨 池袋
新宿 豚骨
渋谷 豚骨
池袋 豚骨
醤油 新宿
醤油 渋谷
醤油 池袋
新宿 醤油
渋谷 醤油
池袋 醤油

Comments

[Ryupe]

一つ質問ですが、ramenとstationの要素の組み合わせは6通りのように思いますが、実現したいアウトプットは一体どういう条件でしょうか？

[loveone]

説明が足らず失礼しました。組み合わせは6通りですが、要素の順番を入れ替えたパターンも全て取得したいため、合計では12パターンのアウトプットを実現したいです。

Answer 1

回答

itertools.product をお使いください。

簡単な例:

from itertools import product

A = ["A", "B"]
B = [1, 2, 3]

print(list(product(A, B)))
# [('A', 1), ('A', 2), ('A', 3), ('B', 1), ('B', 2), ('B', 3)]

リストA、リストB があったとき、{(x, y)|x ∈ A ∧ y ∈ B}∪{(x, y)|x ∈ B ∧ y ∈ A} は次のように作れます。

from itertools import product

ramen = ["豚骨", "醤油"]
station = ["新宿", "渋谷", "池袋"]

key_words = list(product(ramen, station)) + list(product(station, ramen))

for k in key_words:
    print(k)
# ('豚骨', '新宿')
# ('豚骨', '渋谷')
# ('豚骨', '池袋')
# ('醤油', '新宿')
# ('醤油', '渋谷')
# ('醤油', '池袋')
# ('新宿', '豚骨')
# ('新宿', '醤油')
# ('渋谷', '豚骨')
# ('渋谷', '醤油')
# ('池袋', '豚骨')
# ('池袋', '醤油')

参考文献

• itertools.product — Python 3.7.4 ドキュメント

Comments

[loveone]

ありがとうございます！実現したいアウトプットが再現できました！
ドキュメントもよく読ませていただきます！
ベストアンサーにさせていただきました！

Answer 2

直積と順列を組合せると以下のようにも書けます。

import itertools
for v in itertools.product(["豚骨","醤油"],["新宿","渋谷","池袋"]):
    for p in itertools.permutations(v):
        print(p)
"""
('豚骨', '新宿')
('新宿', '豚骨')
('豚骨', '渋谷')
('渋谷', '豚骨')
('豚骨', '池袋')
('池袋', '豚骨')
('醤油', '新宿')
('新宿', '醤油')
('醤油', '渋谷')
('渋谷', '醤油')
('醤油', '池袋')
('池袋', '醤油')
"""