你的位置:首页 > 信息动态 > 新闻中心
信息动态
联系我们

Matlab中find在python中的替代

2021/12/24 4:57:05
  • 方法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的值的位置索引