有没有一种方法可以查看Numpy中的矩阵是否包含特定矢量?
即
X = np.array([[1, 1, 1], [2, 2, 2], [3, 3, 3]])
v = np.array([1, 1, 1])
我希望能够测试:bool = v
中的X
。我知道这不适用于Numpy,想知道是否可以在没有令人讨厌的循环的情况下测试它?感谢您的帮助!
答案 0 :(得分:0)
您可以使用all
汇总行,然后查看是否有所有列都匹配的行。
np.any(np.all(np.isin(X,v,True),axis=1))
我应该提到,这是假设您的行是独特且唯一的。