测试给定数组的子数组

Question

这是初学者Numpy用户的一个基本问题:我有一个5行2列的2D数组,你可以看到10个2d向量,我想测试一个给定的向量是否在表中.

例如 :

>>> tableau = array(range(10), dtype = uint8)
>>> tableau.shape = (5,2)
>>> print tableau
[[ 0  1]
 [ 2  3]
 [ 4  5]
 [ 6  7]
 [ 8  9]]
>>> [0, 1] in tableau
True

最后一行给出了True,但在'tableau'中也是'[0,2].

目前,我计算欧几里得距离是否为0,但我确信这是一个更简单的答案.

谢谢你的帮助

Answer 1

您可以对匹配数组执行布尔减少:

([0, 1] == tableau).all(axis=1).any()