numpy.ndarray查找元素index

可以使用`numpy.where`函数来查找数组中指定元素的索引。

例如，查找数组`arr`中值为3的元素的索引：

import numpy as np

arr = np.array([1, 2, 3, 4, 5, 3, 6, 7])
idx = np.where(arr == 3)
print(idx)

输出：

(array([2, 5]),)

其中`np.where`返回一个元组，元组中的第一个元素是一个数组，表示符合条件的元素的索引。在这个例子中，值为3的元素在数组中的索引为2和5。