在python中以2D数组打印组合

Question

我有这个数组：

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

我想显示数组的所有可能组合，例如：

0 1 2
3 1 2
0 4 2
3 4 2

此数组的行和列不是常量。它会根据用户的输入进行更改。

Answer 1

使用itertools.product：

from itertools import product
l = [[0, 3], [1, 4], [2]]

for prod in product(*l):
    print(prod)