通過從每個列表中選擇一項來從列表列表中獲取所有可能的組合

Question

我有一個列表列表，我想通過從每個列表中選擇一個項目來獲得所有可能的組合。

例如以下列表：

l = [[1,2],[3],[4,5,6],[7],[8,9]]

output 應該是：

1,3,4,7,8
1,3,5,7,8
1,3,6,7,8
1,3,4,7,9
1,3,5,7,9
1,3,6,7,9
2,3,4,7,8
2,3,5,7,8
2,3,6,7,8
2,3,4,7,9
2,3,5,7,9
2,3,6,7,9

列表及其項目的數量是隨機的

Answer 1

使用itertools.product ：

>>> l = [[1,2],[3],[4,5,6],[7],[8,9]]
>>> from itertools import product
>>> list(product(*l))
[(1, 3, 4, 7, 8), (1, 3, 4, 7, 9), (1, 3, 5, 7, 8), (1, 3, 5, 7, 9), (1, 3, 6, 7, 8), (1, 3, 6, 7, 9), (2, 3, 4, 7, 8), (2, 3, 4, 7, 9), (2, 3, 5, 7, 8), (2, 3, 5, 7, 9), (2, 3, 6, 7, 8), (2, 3, 6, 7, 9)]

Answer 2

itertools為您完成工作

In [1]: from itertools import product

In [2]: l = [[1,2],[3],[4,5,6],[7],[8,9]]

In [3]: product(*l)
Out[3]: <itertools.product at 0x7f012ba3e780>

In [4]: list(product(*l))
Out[4]: 
[(1, 3, 4, 7, 8),
 (1, 3, 4, 7, 9),
 (1, 3, 5, 7, 8),
 (1, 3, 5, 7, 9),
 (1, 3, 6, 7, 8),
 (1, 3, 6, 7, 9),
 (2, 3, 4, 7, 8),
 (2, 3, 4, 7, 9),
 (2, 3, 5, 7, 8),
 (2, 3, 5, 7, 9),
 (2, 3, 6, 7, 8),
 (2, 3, 6, 7, 9)]