这是 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,但也给出“[0, 2] in tableau”。
目前,我计算欧氏距离是否为 0,但我确信有一个更简单的答案。
谢谢你的帮助
您可以对匹配数组执行布尔归约:
([0, 1] == tableau).all(axis=1).any()
本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系:hwhale#tublm.com(使用前将#替换为@)