在 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
我怎样才能做到这一点?
在 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
我怎样才能做到这一点?
import itertools
lst = list(itertools.product([0, 1], repeat=3))
这将产生一个元组列表(见这里)
您可以轻松地将其更改为使用变量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
对象并生成列表。
无需使用任何内置功能或智能技术,我们就可以得到这样的结果。
def per(n):
for i in range(1<<n):
s=bin(i)[2:]
s='0'*(n-len(s))+s
print (map(int,list(s)))
per(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]
以下将为您提供所有此类组合
bin = [0,1]
[ (x,y,z) for x in bin for y in bin for z in bin ]