要查找特定值的索引,可以使用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])
億速云公眾號
手機網站二維碼
Copyright ? Yisu Cloud Ltd. All Rights Reserved. 2018 版權所有
廣州億速云計算有限公司粵ICP備17096448號-1 粵公網安備 44010402001142號增值電信業務經營許可證編號:B1-20181529