numpy返回符合特定条件的索引方法

利用np.nonzero函数 按照轴顺序输出

a = np.array([[1 , 2, 3, 4], [5, 6, 7, 8], [9, 10, 11, 12]])
print(a>=5)
print( np.nonzero(a>=5))

[[False False False False]
 [ True  True  True  True]
 [ True  True  True  True]]
(array([1, 1, 1, 1, 2, 2, 2, 2]), array([0, 1, 2, 3, 0, 1, 2, 3]))

输出位置的话 用zip函数

b=np.nonzero(a>=5)
list_of_coordinates= list(zip(b[0], b[1]))
for coord in list_of_coordinates:
...     print(coord)
(0, 0)
(0, 1)
(0, 2)
(0, 3)