检查Numpy中2D数组是否包含特定的1D数组

时间:2018-06-25 20:09:32

标签: python arrays numpy

有没有一种方法可以查看Numpy中的矩阵是否包含特定矢量?

X = np.array([[1, 1, 1], [2, 2, 2], [3, 3, 3]]) 

v = np.array([1, 1, 1])

我希望能够测试:bool = v中的X。我知道这不适用于Numpy,想知道是否可以在没有令人讨厌的循环的情况下测试它?感谢您的帮助!

1 个答案:

答案 0 :(得分:0)

您可以使用all汇总行,然后查看是否有所有列都匹配的行。

np.any(np.all(np.isin(X,v,True),axis=1))

我应该提到,这是假设您的行是独特且唯一的。