在R语言中使用条件语句处理含NA值的3个栅格数据

我帮你梳理下问题，然后一步步解决NA值处理和代码里的小问题哈！

首先，你的代码里有个致命的小错误：循环里硬写了i <- 1，这会导致循环根本不会遍历name_BSA里的所有元素，只会一直处理第1个，先把这个改掉！

接下来是NA值的核心问题：栅格数据的NA处理不能用普通的is.na(BSA[,])，因为这是把整个栅格转成向量后判断整体有没有NA，而我们需要逐像元判断哪些位置三个栅格都有有效值（或者根据你的需求调整，比如允许两个有值就计算）。

下面是修改后的完整代码，我给你两种思路，你可以根据自己的计算需求选择：

思路1：只保留三个栅格都有有效值的像元进行计算 #

用overlay函数逐像元处理，自定义NA判断逻辑：

for(i in 1:length(name_BSA)){ 
  # 去掉硬编码的i<-1，让循环正常遍历每个元素
  if(days_BSA[i] == days_WSA[i] & days_WSA[i] == days_FDS[i]){ 
    BSA <- raster(list_BSA[i]) 
    WSA <- raster(list_WSA[i]) 
    FDS <- raster(list_FDS[i]) 
    
    # 自定义你的计算表达式，这里示例用BSA + WSA * FDS，你替换成自己的逻辑
    result <- overlay(BSA, WSA, FDS, fun = function(b, w, f) {
      # 判断当前像元三个栅格是否都不为NA
      valid_pixel <- !is.na(b) & !is.na(w) & !is.na(f)
      # 初始化结果为NA
      res <- rep(NA, length(b))
      # 只在有效像元上执行计算
      res[valid_pixel] <- b[valid_pixel] + w[valid_pixel] * f[valid_pixel]
      return(res)
    })
    
    # 可以在这里把结果保存到文件，或者做后续处理
    # writeRaster(result, filename = paste0("calculated_result_", i, ".tif"), overwrite = TRUE)
  } 
}

思路2：允许部分栅格为NA时计算（比如两个有值就计算） #

如果你的业务逻辑允许部分NA（比如只要BSA和WSA有值就计算，不管FDS），只需要修改valid_pixel的判断条件即可：

valid_pixel <- (!is.na(b) & !is.na(w)) | (!is.na(b) & !is.na(f)) | (!is.na(w) & !is.na(f))

补充：用calc函数处理合并后的Brick #

如果你习惯先把三个栅格合并成Brick，也可以用calc函数实现同样的逻辑：

for(i in 1:length(name_BSA)){ 
  if(days_BSA[i] == days_WSA[i] & days_WSA[i] == days_FDS[i]){ 
    BSA <- raster(list_BSA[i]) 
    WSA <- raster(list_WSA[i]) 
    FDS <- raster(list_FDS[i]) 
    
    brick_data <- brick(BSA, WSA, FDS)
    result <- calc(brick_data, fun = function(x) {
      # complete.cases判断每行（每个像元的三个值）是否都不为NA
      valid_pixel <- complete.cases(x)
      res <- rep(NA, nrow(x))
      # x[,1]是BSA，x[,2]是WSA，x[,3]是FDS，替换成你的表达式
      res[valid_pixel] <- x[valid_pixel, 1] + x[valid_pixel, 2] * x[valid_pixel, 3]
      return(res)
    })
  } 
}

关键提醒 #

• 不要用is.na(BSA[,])这种方式判断栅格的NA，这是判断整个栅格是否存在NA，而非逐像元判断，完全不符合你的需求。
• raster包的overlay和calc函数都是专门为栅格数据设计的逐像元处理工具，能高效处理大栅格数据，避免手动遍历像元的性能问题。

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