- 方法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的值的位置索引