我想生成一个列表列表,表示数字0和1的所有可能组合.列表的长度为n.

输出应该如下所示.对于n = 1：

[ [0], [1] ]

对于n = 2：

[ [0,0], [0, 1], [1,0], [1, 1] ]

对于n = 3：

[ [0,0,0], [0, 0, 1], [0, 1, 1]... [1, 1, 1] ]

我查看了itertools.combinations,但这会产生元组,而不是列表. [0,1]和[1,0]是不同的组合,而只有一个元组(0,1)(顺序无关紧要).

任何提示或建议？我尝试了一些递归技术,但我还没有找到解决方案.

解决方法:

你在找itertools.product(...).

>>> from itertools import product
>>> list(product([1, 0], repeat=2))
[(1, 1), (1, 0), (0, 1), (0, 0)]

如果要将内部元素转换为列表类型,请使用列表推导

>>> [list(elem) for elem in product([1, 0], repeat =2)]
[[1, 1], [1, 0], [0, 1], [0, 0]]

或者使用map()

>>> map(list, product([1, 0], repeat=2))
[[1, 1], [1, 0], [0, 1], [0, 0]]