使用 numpy时 in 的一个容易忽略错误

最新推荐文章于 2025-07-27 16:46:34 发布

培之

最新推荐文章于 2025-07-27 16:46:34 发布

阅读量449

点赞数 6

CC 4.0 BY-SA版权

文章标签： numpy ubuntu

本文链接：https://blog.youkuaiyun.com/OrdinaryMatthew/article/details/135034865

文章讲述了在使用Python的numpy库时，如何避免错误地用in判断一个点是否在多维度数组(np.darray)中，强调了应该使用`np.any(np.all(...,axis=1))`的正确范式。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

记录 Python 使用 numpy 库判断一个点是否在 np.darray 中的一个容易犯的错误
简单地使用 $in$ 判断会有问题，如果 np.darray 是两个维度的，只要这个点任意一维度在 np.darray 能找到，就会是 True。所以要用下面的范式

import numpy as np

kp0_numpy_2 = np.array([[5, 7],
                        [30, 4],
                        [11, 0],
                        [457, 65]])
point = [1,4]

flag = point  in kp0_numpy_2
print(flag)


flag = np.any(np.all(point == kp0_numpy_2, axis=1))
print(flag)