使用数组操作过滤 numpy

Question

我正在使用 python 并尝试过滤二维数组以仅包含具有特定总和且不包含元素 0 的数组。

其他教程似乎展示了如何使用 numpy.where 过滤数组以获取满足条件的某些元素，但我试图只获取满足条件的某些数组，当然不使用循环，而是使用 numpy方法。

类似这个操作，但是使用数组和numpy：

import itertools

list_o_tuples = list(filter(lambda x: sum(x)==10 and 0 not in x, 
                    itertools.combinations(range(10),3)))
#returns [(1, 2, 7), (1, 3, 6), (1, 4, 5), (2, 3, 5)]

Answer 1

我想这就是你想要的：

test = np.array(list(itertools.combinations(range(10),3)))
mask = (test.sum(axis=1) == 10) & (test.all(axis=1))
test[mask]

为了提高安全性/可读性，您可能需要使用(test != 0).all(axis=1) 而不是test.all(axis=1)。