如何使用Numpy查找数组中指定元素的位置？

针对你给出的示例数组，我来一步步演示怎么查找元素5的具体位置：

首先，先导入NumPy并创建你的示例数组：

import numpy as np

arr = np.array([1, 7, 10, 5, 2, 9, 8])

方法1：使用np.where() #

np.where()是最常用的查找方法之一，它会返回一个包含索引的元组。对于一维数组来说，我们可以直接提取元组的第一个元素得到索引集合：

index_tuple = np.where(arr == 5)
print(index_tuple)  # 输出：(array([3]),)
# 若要获取具体的索引数值，可这样提取：
print(index_tuple[0][0])  # 输出：3

如果数组中有多个匹配的目标元素，np.where()会一次性返回所有对应的索引，非常方便批量查找。

方法2：使用np.argwhere() #

np.argwhere()会返回一个二维数组，每个匹配元素的索引单独作为数组的一行。这种格式在处理多维数组时会更直观，能清晰保留维度信息：

index_array = np.argwhere(arr == 5)
print(index_array)  # 输出：[[3]]
# 提取具体索引数值：
print(index_array[0][0])  # 输出：3

方法3：使用np.flatnonzero() #

这个方法专门用来返回数组中非零元素的索引，我们可以把arr == 5生成的布尔数组传入，间接得到匹配元素的索引：

index_list = np.flatnonzero(arr == 5)
print(index_list)  # 输出：[3]
print(index_list[0])  # 输出：3

需要注意的是，如果要查找的元素不在数组中，这些方法都会返回空数组，不会抛出错误，你可以通过判断返回结果是否为空来确认元素是否存在。

内容的提问来源于stack exchange，提问作者AufsMaulwurf