如何为数据框列表创建实现API递归计算的迭代函数？

方法1：基础R实现（与原循环逻辑完全对齐） #

步骤1：定义单个数据框处理函数 #

该函数接收数据框和衰减系数k，按公式计算api列后返回新数据框：

calculate_api <- function(df, k = 0.9) {
  n_rows <- nrow(df)
  api <- numeric(n_rows)
  # 设置第一个api值
  api[1] <- k * df$precip[1]
  # 循环计算后续值（行数大于1时执行）
  if (n_rows > 1) {
    for (i in 2:n_rows) {
      api[i] <- k * api[i-1] + df$precip[i]
    }
  }
  df$api <- api
  return(df)
}

步骤2：批量应用到数据框列表 #

用lapply遍历列表，将函数应用到每个数据框：

# 使用默认k=0.9计算
list_Df_with_api <- lapply(list_Df, calculate_api)

# 自定义k值（例如k=0.85）
list_Df_with_api <- lapply(list_Df, calculate_api, k = 0.85)

方法2：Tidyverse风格实现（更简洁） #

如果习惯使用tidyverse工具链，可结合purrr::map和accumulate实现递归计算：

library(tidyverse)

k <- 0.9

list_Df_with_api <- map(list_Df, ~ .x %>%
                          mutate(api = {
                            api_vec <- numeric(nrow(.))
                            api_vec[1] <- k * precip[1]
                            if (nrow(.) > 1) {
                              # 用accumulate处理递归逻辑
                              api_vec[-1] <- accumulate(precip[-1], ~ k * .x + .y, .init = api_vec[1])[-1]
                            }
                            api_vec
                          }))

结果验证 #

取列表中第一个数据框查看前几行，确认api列计算符合预期：

head(list_Df_with_api[[1]], 5)

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