你可以使用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的元素的索引。