获取numpy数组中某个元素或者满足某个条件的元素的index索引

import numpy as np
a=[1,2,3,4]
b=np.array(a)
print('b==2:')
print(np.argwhere(b==2))
print('b>2:')
print(np.argwhere(b>2))

输出如下所示：

b==2:
[[1]]
b>2:
[[2]
 [3]]

使用np.argwhere() 加条件，即可得到满足条件的索引。（当数组较大时慎用此方法，因为速度较慢）