numpy 如何计算geotiff数组中值为1的相邻值

piv4azn7 于 2023-05-17 发布在其他

关注(0)|答案(1)|浏览(180)

假设我们有0和1的geotiff。

import rasterio
src = rasterio.open('myData.tif')
data = src.read(1)
data
array([[0, 1, 1, 0],
       [1, 0, 0, 1],
       [0, 0, 1, 0],
       [1, 0, 1, 1]])

对于每个像素1，我希望所有相邻像素的总和形成一个1的集群，并具有如下内容：

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

numpy

来源：https://stackoverflow.com/questions/73882042/how-to-count-the-adjacent-values-with-values-of-1-in-a-geotiff-array

1条答案

按热度按时间

9rbhqvlz1#

可以使用scipy.ndimage.label：

from scipy.ndimage import label

out = np.zeros_like(data)

labels, N = label(data)
for i in range(N):
    mask = labels==i+1
    out[mask] = mask.sum()

输出：

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

赞(0）回复(0）举报 2023-05-17

我来回答

numpy 如何计算geotiff数组中值为1的相邻值

1条答案

相关问题

热门标签

最新问答