python - 测试给定数组的子数组

Question

这是 Numpy 初学者提出的一个基本问题:我有一个 5 行 2 列的二维数组，您可以将其视为 10 个二维向量，我想测试给定的向量是否在表。

例如:

>>> 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，但是 '[0, 2] in tableau' 也是。

目前，我计算欧氏距离是否为 0，但我确信有一个更简单的答案。

感谢您的帮助

Answer 1

您可以对匹配数组执行 bool 归约:

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