python - 如何获得n个二进制值的所有组合？

Question

这个问题在这里已经有了答案:

How do I compute all possibilities for an array of numbers/bits (in python, or any language for that matter) (5 个回答)

关闭 9年前.

在 Python 中，我怎样才能得到 n 二进制值 0 和 1 的所有组合？

比如如果n = 3，我想有

[ [0,0,0], [0,0,1], [0,1,0], [0,1,1], ... [1,1,1] ]  #total 2^3 combinations

我该怎么做？

Answer 1

使用 itertools.product

import itertools
lst = list(itertools.product([0, 1], repeat=3))

这将产生一个元组列表(参见 here)

您可以轻松地将其更改为使用变量repeat:

n = 3
lst = list(itertools.product([0, 1], repeat=n))

如果你需要一个列表列表，那么你可以使用 map 函数(感谢@Aesthete)。

lst = map(list, itertools.product([0, 1], repeat=n))

或者在 Python 3 中:

lst = list(map(list, itertools.product([0, 1], repeat=n)))
# OR
lst = [list(i) for i in itertools.product([0, 1], repeat=n)]

请注意，使用 map 或列表推导式意味着您不需要将产品转换为列表，因为它将遍历 itertools.product 对象和生成一个列表。