2018-07-24

python中获取list中满足条件的下标

a=np.array([1,0,0,1])  #必须转换成array

print(np.argwhere(a==1)) #list中全是1的下标

输出结果：[[0] [3]]