在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]