numpy获取数组指定值的索引并返回索引列表

### 使用 NumPy 获取数组中指定元素的所有索引为了找到 NumPy 数组中特定值的所有索引位置并将这些索引作为列表返回，`numpy.where()` 函数非常有用。此函数可以根据给定条件筛选出符合条件的元素的位置。对于一维数组而言，该方法会返回单个包含所有匹配项索引的一维元组；而对于多维情况，则会产生多个元组，每个代表不同维度上的坐标集合[^4]。下面展示了一个具体的例子，说明怎样利用 `numpy.where()` 来实现这一目标： ```python import numpy as np def find_indices(array, value): """ 查找数组中等于value的所有元素的索引. 参数: array (np.ndarray): 输入的NumPy数组. value : 要查找的目标值. 返回: list of tuples: 所有匹配到的元素所在的索引组成的列表. 对于二维数组来说，每一个tuple分别表示行列索引. """ indices = np.where(array == value) result = list(zip(*indices)) return result # 创建一个简单的二维NumPy数组用于测试 test_array = np.array([[1, 2, 3], [4, 5, 6]]) print(find_indices(test_array, 5)) ``` 上述代码定义了一个名为 `find_indices` 的辅助函数，接受任意形状的 NumPy 数组以及想要查询的具体数值作为输入参数。内部调用了 `numpy.where()` 方法来定位所有的匹配项，并最终将它们组合成易于理解的形式——即由各维度上相应坐标的 tuple 组成的一个 Python 列表。

创作声明：本文部分内容由AI辅助生成（AIGC），仅供参考

下一篇 opencv显示大图c++