多重リストでの選択肢を列挙したい

Question

[[0,1],[2,3],[4],[5,6]]などを入力して、[[0,2,4,5],[0,2,4,6],[0,3,4,5],[0,3,4,6],...,[1,3,4,6]を出力するように、選択肢を列挙するpythonのコードが書けなくて困っています。入力するリスト長や、要素であるリストの長さは未定です。for文をネストしていくのだと、リストの要素数が変わることに対応できません・・・

Answer 1

itertools.productを使いましょう。

>>> import itertools
>>> x = [[0,1],[2,3],[4],[5,6]]
>>> print(list(itertools.product(*x)))
[(0, 2, 4, 5), (0, 2, 4, 6), (0, 3, 4, 5), (0, 3, 4, 6), (1, 2, 4, 5), (1, 2, 4, 6), (1, 3, 4, 5), (1, 3, 4, 6)]

です。

Comments

[sho0423]

ありがとうございます！
早くご回答いただけ方にBAを送らせていただいたのですが、大変助かりました。
一つ質問なのですが、引数にアスタリスクをつけて可変長にしているのはなぜですか？

Answer 2

itertools.product を使うとタプル型でお望みのものが出力されます：

from itertools import product

li = [[0,1],[2,3],[4],[5,6]]

print(list(product(*li))

出力：

[(0, 2, 4, 5), (0, 2, 4, 6), (0, 3, 4, 5), (0, 3, 4, 6), (1, 2, 4, 5), (1, 2, 4, 6), (1, 3, 4, 5), (1, 3, 4, 6)]

後は適宜内包表記などを用いてリストの中身をリスト型等に変えてやれば良いです。

Comments

[sho0423]

ありがとうございます！
liの前についているアスタリスクはどういう意味なのでしょうか？
itertools自体は使おうと試行錯誤していたのですが、その表記をググれませんでした・・・

[kay_ventris4]

*はシーケンスのアンパックをしています。
https://note.nkmk.me/python-tuple-list-unpack/ なんかが参考になるかもしれません。

[sho0423]

親切にありがとうございます、勉強します！