怎么返回np.array中指定元素的index

在NumPy库中，如果你想要找到数组`np.array`中特定元素的索引，可以使用`numpy.where()`函数，这个函数返回的是满足特定条件的所有元素的坐标。例如，假设有一个一维数组`arr`：

import numpy as np

arr = np.array([1, 2, 3, 4, 5])
index = np.where(arr == 3)  # 查找值为3的元素的索引
print(index)  # 输出：(array([2]),)

这里，`index`会是一个包含元组的数组，每个元组表示对应元素的位置，元组的第一个元素是行索引，第二个元素（如果有多个维度）是列索引。对于一维数组，它只是一个包含单个元素的元组。

如果你想获取二维数组中特定值所在的行和列的索引，同样使用`numpy.where()`，但它将返回一个包含两个数组的结果：

arr_2d = np.array([[1, 2], [3, 4], [5, 6]])
row_index, col_index = np.where(arr_2d == 4)
print(row_index, col_index)  # 输出：(array([1]), array([1]))