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

Question

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

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

NumPy 数组有类似的东西吗？

Answer 1

是的，给定一个数组 array 和一个值 item 进行搜索，您可以使用 np.where如:

itemindex = numpy.where(array == item)

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

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

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

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

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