Bootstrap

python查找元素/值在数组中的索引

利用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
    
;