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)