在NumPy中,可以使用np.where()
函數來查找元素索引。np.where()
函數返回滿足指定條件的元素的索引。以下是使用np.where()
函數查找元素索引的示例:
import numpy as np
arr = np.array([1, 2, 3, 4, 5, 6])
# 查找元素3的索引
index = np.where(arr == 3)
print(index) # 輸出 (array([2]),)
# 查找元素大于4的索引
index = np.where(arr > 4)
print(index) # 輸出 (array([4, 5]),)
# 查找元素不等于2的索引
index = np.where(arr != 2)
print(index) # 輸出 (array([0, 2, 3, 4, 5]),)
注意,np.where()
函數返回的是一個元組,其中包含滿足條件的元素的索引數組。如果要獲取索引數組,可以使用index[0]
來獲取。