要查找特定值的索引,可以使用np.where()函數。例如,要查找數組中值為5的索引,可以這樣做:
np.where()
import numpy as np arr = np.array([1, 2, 3, 4, 5, 6]) index = np.where(arr == 5) print(index)
這將打印出值為5的索引,即array([4])。如果數組中有多個相同的值,np.where()將返回所有這些值的索引。
array([4])