为了获取图中非零元素的坐标,numpy提供了
nonzero
函数可以得到对应点的行列索引位置
1. 寻找非零元素的索引值
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坐标
是需要交换的,使用时候务必注意。