本文介绍两种方法来查找NumPy二维数组中满足特定条件（例如大于1）的元素位置索引：一种是使用已废弃的matplotlib.mlab.find函数；另一种是利用numpy自带的nonzero函数自定义实现。
- 方法1：
使用matplotlib.mlab.find，不过在最近版本中，已取消find函数
import matplotlib.mlab as mlab
np_array = 一个2维数组
idx = mlab.find(np_array > 1) # idx为np_array中，大于1的值的位置索引
- 方法2：
自建函数find代替
def find(condition):
res = np.nonzero(condition)
return res
np_array = 一个2维数组
idx = find(np_array > 1) # idx为np_array中，大于1的值的位置索引
扫一扫
372
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
