基于terra包的栅格堆栈NA同步掩膜高效实现方案问询

我有一个包含145个栅格的raster stack，要求如果某单元格在任意一层中为NA，就将该单元格在所有其他层中也设为NA。我已经实现了一种处理方法，但不确定效率如何，希望能得到更优的方案。

我的实现代码 #

library(terra)
a <- rast(ncol = 2, nrow = 2)
values(a) <- c(1,2,3,4)
names(a) <- "layer_one"

b <- rast(ncol = 2, nrow = 2)
values(b) <- c(1,2,3,4)
names(b) <- "layer_two"

c <- rast(ncol = 2, nrow = 2)
values(c) <- c(1,2,3,NA)
names(c) <- "layer_three"

z <- c(a,b,c)

my_layer_mask <- app(z, mean)

masked_z <- mask(z, my_layer_mask)

values(masked_z)

运行结果：

layer_one layer_two layer_three
[1,]         1         1           1
[2,]         2         2           2
[3,]         3         3           3
[4,]        NA        NA          NA

更优方案 #

你的方法通过计算均值生成掩码，虽然能实现需求，但对于145层的栅格栈来说，计算均值的额外开销没必要——我们只需要判断单元格是否在所有层都非NA即可，用更直接的逻辑运算能大幅提升效率：

library(terra)
# 生成示例数据（同你的代码）
a <- rast(ncol = 2, nrow = 2)
values(a) <- c(1,2,3,4)
names(a) <- "layer_one"

b <- rast(ncol = 2, nrow = 2)
values(b) <- c(1,2,3,4)
names(b) <- "layer_two"

c <- rast(ncol = 2, nrow = 2)
values(c) <- c(1,2,3,NA)
names(c) <- "layer_three"

z <- c(a,b,c)

# 生成掩码：所有层都无NA的单元格标记为TRUE，否则FALSE
mask_layer <- sum(is.na(z)) == 0

# 应用掩码
masked_z <- mask(z, mask_layer)

values(masked_z)

方案优势 #

• 向量化运算：sum(is.na(z))是terra原生的向量化操作，直接对整个栅格栈批量计算，比app(z, mean)这种逐单元格调用函数的方式快得多，当栅格数量多（145层）、像素量大时，性能差距会非常明显。
• 逻辑更直接：我们只需要统计每个单元格的NA数量，若数量为0则保留，否则设为NA，完全贴合需求，不需要额外的均值计算步骤。

另外，也可以用app(z, function(x) all(!is.na(x)))生成掩码，但性能不如sum(is.na(z)) == 0——前者是逐单元格调用R函数，后者是terra底层优化的向量化运算。

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