如何在RStudio中用循环高效创建目标DataFrame？

问题背景 #

在RStudio中通过重复代码生成ciecg、ciecghol、ciecgholec三个DataFrame，合并后得到包含基线与最新阶段统计值的目标结果。尝试用循环优化重复代码时，遇到维度不匹配及非数值参数报错，需要用循环直接生成合并后的目标DataFrame替代重复代码。

原重复代码 #

# 生成ciecg
ciecg <- data.frame(
  group = "cg",
  baseline_mean = mean(data_cg$baseline),
  baseline_sd = sd(data_cg$baseline),
  latest_mean = mean(data_cg$latest),
  latest_sd = sd(data_cg$latest)
)

# 生成ciecghol
ciecghol <- data.frame(
  group = "chol",
  baseline_mean = mean(data_chol$baseline),
  baseline_sd = sd(data_chol$baseline),
  latest_mean = mean(data_chol$latest),
  latest_sd = sd(data_chol$latest)
)

# 生成ciecgholec
ciecgholec <- data.frame(
  group = "cholec",
  baseline_mean = mean(data_cholec$baseline),
  baseline_sd = sd(data_cholec$baseline),
  latest_mean = mean(data_cholec$latest),
  latest_sd = sd(data_cholec$latest)
)

# 合并得到目标结果
target_df <- rbind(ciecg, ciecghol, ciecgholec)

预期输出 #

> target_df
    group baseline_mean baseline_sd latest_mean latest_sd
1      cg          12.3         2.1        10.5       1.8
2    chol           8.7         1.5         7.2       1.3
3  cholec           5.4         0.9         4.8       0.7

尝试的错误代码及报错 #

错误代码 #

groups <- c("cg", "chol", "cholec")
data_list <- list(data_cg, data_chol, data_cholec)
target_df <- data.frame()

for (i in 1:length(groups)) {
  temp <- data.frame(
    group = groups[i],
    baseline_mean = mean(data_list[i]$baseline),
    baseline_sd = sd(data_list[i]$baseline),
    latest_mean = mean(data_list[i]$latest),
    latest_sd = sd(data_list[i]$latest)
  )
  target_df <- rbind(target_df, temp)
}

报错信息 #

Error in mean(data_list[i]$baseline) : 
  argument is not numeric or logical: returning NA
In addition: Warning message:
In data_list[i]$baseline :
  $ operator is invalid for atomic vectors

（或出现维度不匹配报错：Error in rbind(deparse.level, ...) : numbers of columns of arguments do not match）

解决方案 #

方法1：修正循环逻辑的基础实现 #

核心问题是列表索引错误（用[[i]]而非[i]提取数据框），同时提前定义结果DataFrame的列结构避免维度问题：

# 定义分组名称与对应的数据框列表
group_names <- c("cg", "chol", "cholec")
data_frames <- list(data_cg, data_chol, data_cholec)

# 初始化指定列类型的空结果DataFrame
target_df <- data.frame(
  group = character(),
  baseline_mean = numeric(),
  baseline_sd = numeric(),
  latest_mean = numeric(),
  latest_sd = numeric(),
  stringsAsFactors = FALSE
)

# 循环生成并合并行
for (i in seq_along(group_names)) {
  current_df <- data_frames[[i]]
  temp_row <- data.frame(
    group = group_names[i],
    baseline_mean = mean(current_df$baseline, na.rm = TRUE), # 加入na.rm处理缺失值
    baseline_sd = sd(current_df$baseline, na.rm = TRUE),
    latest_mean = mean(current_df$latest, na.rm = TRUE),
    latest_sd = sd(current_df$latest, na.rm = TRUE)
  )
  target_df <- rbind(target_df, temp_row)
}

方法2：用tidyverse工具简化实现 #

如果熟悉tidyverse，可用purrr::map_dfr替代循环，代码更简洁：

library(purrr)
library(dplyr)

# 命名数据框列表，方便自动提取分组名
data_frames <- list(
  cg = data_cg,
  chol = data_chol,
  cholec = data_cholec
)

# 批量处理并合并为DataFrame
target_df <- map_dfr(data_frames, function(df) {
  tibble(
    baseline_mean = mean(df$baseline, na.rm = TRUE),
    baseline_sd = sd(df$baseline, na.rm = TRUE),
    latest_mean = mean(df$latest, na.rm = TRUE),
    latest_sd = sd(df$latest, na.rm = TRUE)
  )
}, .id = "group")

错误原因说明 #

1. 非数值参数报错：data_list[i]返回的是子列表而非单个数据框，无法用$提取列，需用[[i]]取出列表中的数据框元素。
2. 维度不匹配报错：初始化空DataFrame时未指定列类型，循环中rbind易因临时行的列类型不一致导致维度冲突，提前定义列结构可避免此问题。

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