Python-是否有NumPy函数返回数组中某物的第一个索引？

一尘不染

python

我知道有一种用于Python列表的方法来返回某些内容的第一个索引：

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

NumPy数组有类似的东西吗？

阅读 508

2020-02-16

共1个答案

一尘不染

是的，在给定NumPy数组array和值的情况下，这是搜索的答案item：

itemindex = numpy.where(array==item)

结果是具有所有行索引，然后是所有列索引的元组。

例如，如果一个数组是二维的，并且它在两个位置包含你的商品，则

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

将等于你的项目，因此

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

2020-02-16