【python】查找array中非零元素的横纵坐标

为了获取图中非零元素的坐标，numpy提供了nonzero函数可以得到对应点的行列索引位置

data = np.zeros((5,5))
data[1,2] = 1
data[3,4] = 2

row,col = np.nonzero(data)
print(row,col)
# >>> [1 3] [2 4]
# [行号.........] [列号.........]
# 1,3是两个点位置的行号

print(data[row,col])
# >>> [1. 2.]
# 基于索引读取数据

值得注意的是，对于图像来说，行、列与x、y坐标是需要交换的，使用时候务必注意。

reference

pic from Go