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

2023-12-16 17:33:27

记录 Python 使用 numpy 库 判断一个 点是否在 np.darray 中的一个容易犯的错误
简单地使用 i n in 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)  

文章来源:https://blog.csdn.net/OrdinaryMatthew/article/details/135034865
本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。