检查矩阵是否包含有效元素

Question

我有这个数组

scale=np.array([-3,0,2,4,7,10,12])

这个矩阵

matrix=np.array([[17, 10, 10],
       [10, 12, 12],
       [ 7,  7,  4],
       [-3, 11,  2]])

现在我想知道矩阵中行的索引，它们不包含任何规模的元素。输出应为：

array([0,3])

我已尝试使用np.where，np.all和np.any而不解决问题。

你有一个简单的解决方案吗？

Answer 1

您正在寻找：

np.where(~np.in1d(matrix, scale).reshape(matrix.shape).all(axis=1))
#(array([0, 3], dtype=int64),)