python - 在python中生成二进制数的二维数组

Question

这就是我在 Python 中尝试的。`

input = []
for i in range(10):
  n = getBin(i, 4)
  input.append(n)
print input

它给出的是:

['0000', '0001', '0010', '0011', '0100', 
 '0101', '0110', '0111', '1000', '1001']

我需要的是:

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

Answer 1

使用itertools.product具有 (0, 1) 和 4 次重复:

input = [list(x) for x in itertools.product((0, 1), repeat=4)]

如果您愿意使用 tuple 的 list 而不是 list，您可以简单地执行以下操作:

input = list(itertools.product((0, 1), repeat=4))

或者最简单的是，如果您无论如何都要迭代它，则无需将其设为列表:

input = itertools.product((0, 1), repeat=4)

最后，(0, 1) 可能是 range(2)，但这很难说是改进

<小时/>

itertools.product 通常会尝试以您指定的相同格式返回。因此，通过输入一个字符串，它会返回一个字符串。给它一个列表，它...几乎返回一个列表(返回一个元组)