numpy如何获取array中数组元素的索引位置

要获取numpy数组中元素的索引位置可以使用numpy模块中的where()函数。下面是获取索引位置的详细攻略：

步骤1：导入numpy模块

import numpy as np

步骤2：创建numpy数组

arr = np.array([10, 20, 30, 40, 50])

步骤3：使用where()函数获取数组元素索引位置

index = np.where(arr == 30)

步骤4：打印索引位置

print(index)

以上代码输出结果为：(array([2]),)，这表示元素30在数组中的索引位置为2。

这里还提供另外一个示例：

步骤1：导入numpy模块

import numpy as np

步骤2：创建numpy数组

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

步骤3：使用where()函数获取数组元素索引位置

index = np.where(arr == 2)

步骤4：打印索引位置

print(index)

以上代码输出结果为：(array([1]), array([1]))，这表示元素2在数组中的索引位置为(1,1)。

需要注意的是，当where()函数返回的是一个二维数组时，第一个数组表示行数，第二个数组表示列数。