如何用R循环处理数据框中n列连续数值列并导出CSV？

问题背景 #

现有以Grp为ID列、后续为若干数值列（数量不固定，从5到500+）的数据框，需按每n=3个连续数值列分组，每次选取ID列加该组数值列，执行pivot_longer操作后导出为单独CSV文件，替代手动重复操作。

示例数据框：

data <- data.frame("Grp" = rep(c("GRP_A1", "GRP_A2", "GRP_A3", "GRP_A4", "GRP_A5"), each = 5),
  A = rep(1:5, each = 5), 
  B = rep(6:10, each = 5), 
  C = rep(11:15, each = 5), 
  D = rep(16:20, each = 5),
  E = rep(21:25, each = 5), 
  G = rep(26:30, each = 5), 
  H = rep(31:35, each = 5),
  I = rep(36:40, each = 5))

设定参数：

n <- 3

解决方案 #

以下两种方法均可自动完成批量处理，适配任意数量的数值列：

方法1：基础循环实现 #

library(tidyverse)

# 获取数值列的起始索引（ID列为第1列，数值列从第2列开始）
num_cols_start <- 2
num_cols_total <- ncol(data) - 1  # 总数值列数，减去ID列

# 生成每组数值列的起始位置
group_starts <- seq(num_cols_start, num_cols_start + num_cols_total - 1, by = n)

# 循环处理每组
for (i in seq_along(group_starts)) {
  start_col <- group_starts[i]
  # 计算当前组的结束列，避免超出数据框总列数
  end_col <- min(start_col + n - 1, ncol(data))
  
  # 选取目标列并转换为长格式
  df_processed <- data %>%
    select(1, start_col:end_col) %>%
    pivot_longer(cols = -Grp, names_to = "Trait", values_to = "Value")
  
  # 导出CSV文件，文件名按序号自动生成
  write.csv(df_processed, paste0("DF", i, ".csv"), row.names = FALSE)
}

方法2：purrr函数式编程（更简洁） #

library(tidyverse)

n <- 3
num_cols <- names(data)[-1]  # 获取所有数值列的列名

# 将数值列名按每n个为一组拆分
col_groups <- split(num_cols, ceiling(seq_along(num_cols)/n))

# 遍历每个分组完成处理与导出
walk(col_groups, function(cols, idx) {
  df_processed <- data %>%
    select(Grp, all_of(cols)) %>%
    pivot_longer(cols = -Grp, names_to = "Trait", values_to = "Value")
  
  write.csv(df_processed, paste0("DF", idx, ".csv"), row.names = FALSE)
}, idx = seq_along(col_groups))

代码说明 #

• 两种方法均支持自动处理最后一组不足n个数值列的情况
• 使用cols = -Grp替代固定列索引，避免因列数变化导致出错
• 文件名按DF1.csv、DF2.csv...的规则自动生成，无需手动命名

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