np.where函数的使用

np.where（）

a = np.array([1,1,2,2,3,3])
np.where(a==1)[0]
Out[7]: array([0, 1], dtype=int64)
print(np.where(a==1)[0])
[0 1]

小结：由此可见，np.where返回的是一个元组，需要加【0】索引才能返回的是位置