R语言中如何迭代将向量列表值存入数据框列表的对应列？

核心优化方案 #

你不需要单独存储中间变量a再逐行回写，你已经写出了核心的变量修改逻辑，仅需将lapply的返回结果直接赋值给原列表对象，一行代码即可完成30个插补数据框的所有变量修改：

# 你已经定义的修改变量函数
modify_vars <- function(x) {
  x$var2 <- ifelse(x$var1 == 1, 1, x$var2)
  x$var3 <- ifelse(x$var2 == 1, 1, x$var3)
  x$var4 <- ifelse(x$var3 == 1, 1, x$var4)
  return(x)
}
# 批量修改所有插补数据框，直接覆盖原列表即可
list.of.imputed.dfs <- lapply(list.of.imputed.dfs, modify_vars)

原理说明 #

lapply会自动遍历list.of.imputed.dfs中的每一个数据框，将其传入modify_vars函数完成变量修改，最终将所有修改后的数据框重新整合成列表，赋值后直接替换原列表对象，不需要手动写90行赋值代码。

通用扩展写法 #

如果后续需要调整修改的变量序列，可以用更通用的函数实现，避免硬编码变量名：

# 通用修改函数：可指定前置判断变量、要修改的变量序列
modify_vars_general <- function(df, start_col = "var1", target_cols = c("var2", "var3", "var4")) {
  # 初始化标记：前置变量为1的观测需要修改后续所有变量
  modify_flag <- df[[start_col]] == 1
  for (col in target_cols) {
    df[[col]] <- ifelse(modify_flag, 1, df[[col]])
    # 更新标记：只要当前变量为1，后续所有变量都要设为1
    modify_flag <- modify_flag | (df[[col]] == 1)
  }
  return(df)
}
# 调用方式一致
list.of.imputed.dfs <- lapply(list.of.imputed.dfs, modify_vars_general)

可选for循环实现 #

如果你偏好for循环的写法，也可以用如下代码实现同等效果，不需要逐行写30次赋值：

for (i in seq_along(list.of.imputed.dfs)) {
  list.of.imputed.dfs[[i]] <- modify_vars(list.of.imputed.dfs[[i]])
}

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