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