在numpy
中,可以使用numpy.where()
函数来获取满足条件的元素的索引。这个函数将返回一个包含满足条件的元素索引的元组。
例如,我们有一个numpy
数组arr
:
import numpy as np
arr = np.array([1, 2, 3, 4, 5])
我们想获取所有大于2的元素的索引,可以使用numpy.where()
函数:
indexes = np.where(arr > 2)
print(indexes)
输出:
(array([2, 3, 4]),)
这表示满足条件的元素的索引是2、3和4。如果我们想直接获取索引值而不是一个元组,可以使用numpy.where()
的第一个元素:
indexes = np.where(arr > 2)[0]
print(indexes)
输出:
[2 3 4]