小编典典

是否有一个 NumPy 函数来返回数组中某物的第一个索引?

all

我知道 Python 列表有一种方法可以返回某物的第一个索引:

>>> l = [1, 2, 3]
>>> l.index(2)
1

NumPy 数组有类似的东西吗?


阅读 113

收藏
2022-03-08

共1个答案

小编典典

是的,给定一个数组
,array和一个item要搜索的值,您可以np.where用作:

itemindex = numpy.where(array==item)

结果是一个元组,首先是所有行索引,然后是所有列索引。

例如,如果一个数组是二维的,并且它在两个位置包含您的项目,那么

array[itemindex[0][0]][itemindex[1][0]]

将等于您的项目,因此将是:

array[itemindex[0][1]][itemindex[1][1]]
2022-03-08