Question

我想生成一个列表，列出数字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）（顺序无关紧要）。

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

Answer 1

您正在寻找itertools.product(...)。

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

如果要将内部元素转换为list类型，请使用列表理解

>>> [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]]

Answer 2

使用itertools.product，将repeat分配给n。

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

演示：

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

Answer 3

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

要获取列表清单，您可以执行以下操作：

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

Answer 4

只是为了增加一些多样性，这是实现这一目标的另一种方式：

>>> [map(int, format(i, "03b")) for i in range(8)]
[[0, 0, 0], [0, 0, 1], [0, 1, 0], [0, 1, 1], [1, 0, 0], [1, 0, 1], [1, 1, 0], [1, 1, 1]]

如何生成列表组合？

4 个答案: