numpy根据取值索引位置

import numpy as np

# 示例的NumPy数组
numpy_array = np.array([[1, 2, 3], [4, 5, 6], [7, 2, 9]])

# 使用np.where()函数找到元素等于2的索引位置
indices = np.where(numpy_array == 2)

# 输出索引位置
print(indices)

上述代码将输出以下结果：

(array([0, 2], dtype=int64), array([1, 1], dtype=int64))

np.where()函数返回一个元组，包含满足条件的元素的行索引和列索引。在这个例子中，元素等于2的索引位置是行索引0、列索引1，以及行索引2、列索引1。

注意：返回的索引是从0开始的。如果要获得每个匹配元素的具体坐标，可以使用np.transpose()函数来转置索引元组：

coordinates = np.transpose(indices)
print(coordinates)

这将输出：

[[0 1]
 [2 1]]

这样，我们得到了每个元素等于2的具体坐标位置。第一行的元素等于2位于行索引0、列索引1，第二行的元素等于2位于行索引2、列索引1。