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