numpy如何根据值找索引

你可以使用numpy.where()函数来根据值找到索引。该函数返回一个包含满足条件的元素的索引的元组。

下面是一个示例：

import numpy as np

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

# 找到值为2的元素的索引
indexes = np.where(arr == 2)

print(indexes)

输出结果为：

(array([1, 3]),)

这表示索引1和索引3处的元素的值为2。请注意，np.where()返回的是一个元组，因此可以使用indexes[0]来获取索引的数组。

如果你想根据多个条件来查找索引，你可以将多个条件组合起来，例如：

indexes = np.where((arr == 2) & (arr < 4))

这将找到值为2且小于4的元素的索引。