二进制图像中所有可能的组合

Question

我试图在一个形状为(n，10)的数组中创建所有可能的0和1的组合。例如，如果我们假设一个像这样的任意组合:np.array([0，0，1，1，0，0，1，1，0，0])，我如何生成所有可能的组合(这将产生2^10=1024数组)？

Answer 1

您可以使用itertools模块中的排列：

import numpy as np 
import itertools

list = np.array([0, 0, 1, 1, 0, 0, 1, 1, 0, 0])
combs = itertools.permutations(list)

for i in combs:
    print(i)

Answer 2

是的，您可以将 itertools.product（） 与 repeat 参数一起使用来生成所需的输出：

import numpy as np
from itertools import product

np.array(list(product([0, 1], repeat=10)))

这将输出:

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