如何查找 NumPy 数组中元素的第一个索引？

在 NumPy 数组中查找项目的第一个索引

在 Python 列表中，index() 方法允许我们检索第一个项目一个元素的出现。 NumPy 能为数组提供类似的功能吗？

答案

是的，NumPy 提供了一种方便的方法来定位数组中项目的第一个索引。您可以按如下方式使用 np.where 函数：

import numpy as np

array = np.array([1, 2, 3])
item = 2

itemindex = np.where(array == item)

np.where 函数返回一个元组，其中包含找到该项目的行索引和列索引。

示例

如果您的数组是二维的并且包含位于两个位置的项目，然后：

array[itemindex[0][0]][itemindex[1][0]]  # equals item
array[itemindex[0][1]][itemindex[1][1]]  # also equals item

这使您可以根据第一次出现轻松访问所需的元素。

以上是如何查找 NumPy 数组中元素的第一个索引？的详细内容。更多信息请关注PHP中文网其他相关文章！