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

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),)