python numpy查找元素位置

可以使用numpy库中的where函数来查找元素位置。

例如，假设有一个numpy数组a：

import numpy as np

a = np.array([1, 2, 3, 4, 5])

要查找元素3的位置，可以使用where函数：

index = np.where(a == 3)
print(index)

输出结果为：

(array([2]),)

其中，array([2])表示元素3在数组a中的位置为第3个元素（数组下标从0开始计数）。

如果要查找多个元素的位置，可以将多个元素放在一个列表中，然后使用numpy的in1d函数来判断元素是否在数组中，再使用where函数来查找位置。

例如，假设要查找元素2和4的位置：

b = np.array([2, 4])
index = np.where(np.in1d(a, b))
print(index)

输出结果为：

(array([1, 3]),)

其中，array([1, 3])表示元素2和4在数组a中的位置分别为第2个和第4个元素。