Matlab中find在python中的替代

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