在 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中文网其他相关文章！
