利用np.where函数,np.where(condition)中的condition表示筛选元素的条件,可以是一个判断式
- 找某个元素在数组中的索引
import numpy as np x = np. array ([4, 7, 7, 7, 8, 8, 8]) print(np.where(x==8)) output:(array([4, 5, 6],dtype=int64),)
- 找某个元素在数组中的第一个索引位置
#find first index position where x is equal to 8 print(np.where(x==8)[0][0]) output:4