能否一步完成多边形栅格化(rasterize)与掩膜(mask)操作？

测试数据 #

library(terra)

# 创建模板栅格（中间区域有数据，其余为NA）
template_raster <- rast(xmin=0, xmax=10, ymin=0, ymax=10, crs="+proj=longlat +datum=WGS84 +no_defs +type=crs", resolution = 0.1)
values(template_raster) <- c(rep(NA,4000),rep(10,2000),rep(NA,4000))

# 创建待栅格化的多边形
polygon_to_rasterize <- as.polygons(ext(3, 6, 3, 6), crs="+proj=longlat +datum=WGS84")
polygon_to_rasterize$polygon_val <- 20

# 可视化测试数据
plot(template_raster)
plot(polygon_to_rasterize, add=T)

问题需求 #

需要将多边形栅格化，但仅保留模板栅格中有数据的区域，且结果与模板栅格尺寸一致。当前通过两步实现，但处理50000+图层时，临时文件会带来IO开销，希望一步完成。

一步实现方案 #

利用terra的链式调用，将rasterize和mask操作合并为一步，无需生成中间临时文件：

# 一步完成栅格化+掩膜，直接得到目标结果
result <- mask(
  rasterize(polygon_to_rasterize, template_raster, field = "polygon_val"),
  template_raster
)

# 如需直接写入最终文件，可添加filename参数，全程无临时文件
# result <- mask(
#   rasterize(polygon_to_rasterize, template_raster, field = "polygon_val"),
#   template_raster,
#   filename = "final_result.tif"
# )

# 可视化结果
plot(result)

原理说明 #

• rasterize先基于模板栅格的尺寸、分辨率和坐标系生成多边形的栅格化结果
• mask直接以模板栅格的非NA区域为掩码，过滤掉栅格化结果中模板栅格为NA的部分
• 链式调用全程在内存中完成（或直接写入最终文件），避免了中间临时文件的创建与读写，大幅提升批量处理的效率

内容的提问来源于stack exchange，提问作者TheRealJimShady