python 数组组合_python-查找数组中某些元素的所有组合

一种方法是使用掩码以ls中的值填充数组的无效条目,其次数是ls的排列次数.

但是,通过将这些置换的长度设置为arr中无效条目的数量,可以使此操作更加可靠.这样,我们还要考虑len(ls)> (x == None).sum().

def fill_combs(x, fill, replace=None):

from itertools import permutations

m = x == replace

for i in permutations(fill, int(m.sum())):

x_ = x.copy()

x_[m] = np.array(i)

yield x_.astype(int)

样品运行：

arr = np.array([

[1, None, 3],

[9, 4, None],

])

ls = [9, 8]

list(fill_with_permut(arr, ls))

输出：

[array([[1, 9, 3],

[9, 4, 8]]),

array([[1, 8, 3],

[9, 4, 9]])]

或更大的ls：

ls = [3,5,2]

list(fill_with_permut(arr, ls))

[array([[1, 3, 3],

[9, 4, 5]]),

array([[1, 3, 3],

[9, 4, 2]]),

array([[1, 5, 3],

[9, 4, 3]]),

array([[1, 5, 3],

[9, 4, 2]]),

array([[1, 2, 3],

[9, 4, 3]]),

array([[1, 2, 3],

[9, 4, 5]])]