自定义R函数计算SAPEI指数的WSD模块遇问题，求技术协助

原函数存在几个关键问题：

• 循环内部直接调用return()，导致函数在第一次迭代时就终止，完全没有处理n参数对应的滑动累积计算
• 未计算n天尺度的累积水分平衡，仅返回了单天的P-PET值
• 干湿状态的判断基于单天水分平衡，而非论文要求的n天累积WSD值

根据Li等人2021年论文中SAPEI的定义，WSD是指定时间尺度（如3个月）内降水与潜在蒸散量差值的累积值。以下是两种可行的实现方式：

方式1：使用zoo包高效计算滑动累积（推荐） #

zoo包的rollsum函数可快速实现滑动窗口求和，避免手动循环的低效：

library(zoo)

WSD <- function(P, PET, n) {
  # 计算单天水分平衡
  wat_bal <- P - PET
  
  # 计算n天滑动累积的WSD，align="right"表示以当前日期为窗口终点
  # fill=NA表示窗口不足n天时填充NA（比如前n-1天没有足够的前期数据）
  wsd <- rollsum(wat_bal, k = n, align = "right", fill = NA)
  
  # 基于累积WSD判断干湿状态
  condition <- ifelse(wsd > 0, "wet", "dry")
  
  # 返回结果数据框
  return(data.frame(wat_bal, wsd, condition))
}

方式2：基础R手动循环实现 #

若不想依赖第三方包，可使用基础R循环完成滑动累积：

WSD <- function(P, PET, n) {
  wat_bal <- P - PET
  len <- length(wat_bal)
  wsd <- rep(NA, len)
  
  # 从第n天开始计算累积
  for(i in n:len) {
    # 取当前日期往前n天的范围（包括当天）求和
    wsd[i] <- sum(wat_bal[(i - n + 1):i])
  }
  
  condition <- ifelse(wsd > 0, "wet", "dry")
  return(data.frame(wat_bal, wsd, condition))
}

注意事项： #

• 确保输入的P和PET是按时间顺序排列的向量，且长度一致
• 若使用月尺度（如3个月），需将n设为对应天数（如90天，或根据实际月份天数调整），或先将数据聚合为月尺度再计算
• 前n-1天因没有足够的前期数据，WSD值会设为NA，可根据需求处理这些缺失值

内容的提问来源于stack exchange，提问作者Fabián