在R语言中编写循环批量复制type列至每6列之后

需求说明 #

现有数据集capacity_ex，需要将其中的type列复制到每6列之后（如1_q6、2_q6列的后面），目标格式参考capacity_want。由于实际数据规模较大，无法手动操作，此前编写的循环代码报错，寻求可行的实现方案。

示例数据 #

原数据集 #

capacity_ex <- data.frame("type" = c("p", "t", "t", "r,"),
                         "1_q1" = 1:4,
                         "1_q2" = c("a", "b", "c", "d"),
                         '1_q3' = 10:13,
                         '1_q4' = 100:103,
                         '1_q5' = 110:113,
                         '1_q6' = 11:14,
                         "2_q1" = 22:25,
                         "2_q2" = c("i", "j", "k", "l"),
                         '2_q3' = 20:23,
                         '2_q4' = 200:203,
                         '2_q5' = 210:213,
                         '2_q6' = 21:24,
                          "3_q1" = 90:93,
                          "3_q2" = c("p", "q", "r", "s"), 
                          '3_q3' = 10:13,
                          '3_q4' = 300:303,
                          '3_q5' = 310:313,
                          '3_q6' = 31:34,check.names = FALSE)

期望格式 #

capacity_want <- data.frame("type" = c("p", "t", "t", "r,"),
                         "1_q1" = 1:4,
                         "1_q2" = c("a", "b", "c", "d"),
                         '1_q3' = 10:13,
                         '1_q4' = 100:103,
                         '1_q5' = 110:113,
                         '1_q6' = 11:14,
                         "type" = c("p", "t", "t", "r,"),
                         "2_q1" = 22:25,
                         "2_q2" = c("i", "j", "k", "l"),
                         '2_q3' = 20:23,
                         '2_q4' = 200:203,
                          '2_q5' = 210:213,
                          '2_q6' = 21:24,
                          "type" = c("p", "t", "t", "r,"),
                          "3_q1" = 90:93,
                          "3_q2" = c("p", "q", "r", "s"), 
                           '3_q3' = 10:13,
                          '3_q4' = 300:303,
                           '3_q5' = 310:313,
                          '3_q6' = 31:34,check.names = FALSE)

注：type列可添加后缀保证列名唯一。

错误代码分析 #

你提供的代码存在以下问题：

1. 语法错误：seq(2, (ncol(capacity_ex), 6) 括号使用错误，正确的seq参数格式应为seq(from, to, by)，此处应为seq(2, ncol(capacity_ex), 6)。
2. 逻辑错误：lapply每次都是基于原始capacity_ex执行mutate，而非累积修改后的结果，导致插入操作无法叠加。
3. 函数用法错误：do.call与lapply的组合逻辑错误，lapply返回多个独立的数据框，无法直接通过do.call合并得到目标结果。
4. 列名重复：未处理重复type列的命名问题，可能引发后续数据操作异常。

正确实现方案 #

方案一：反向插入法（推荐，避免位置偏移） #

从后往前插入type列，避免插入操作导致后续列位置偏移，同时自动添加后缀保证列名唯一：

library(dplyr)

# 复制原数据集作为操作基础
capacity_result <- capacity_ex

# 计算需要插入的位置：原数据中每个6列组的最后一列位置（1_q6在第7列，2_q6在第13列，以此类推）
insert_positions <- seq(7, ncol(capacity_ex), 6)

# 反向遍历插入位置，避免位置偏移
for (pos in rev(insert_positions)) {
  # 插入type列并添加后缀
  suffix <- which(insert_positions == pos)
  capacity_result <- capacity_result %>%
    mutate(!!paste0("type_", suffix) := type, .after = pos)
}

方案二：构造列索引重组法 #

通过拆分列组、构造新的列顺序来重组数据框，适合需要严格控制列顺序的场景：

# 获取原数据列名
col_names <- colnames(capacity_ex)

# 拆分除type外的列，每6列为一组
groups <- split(col_names[-1], ceiling(seq_along(col_names[-1])/6))

# 构造新的列顺序：初始type + 每组6列 + 带后缀的type
new_col_order <- c("type")
for (i in seq_along(groups)) {
  new_col_order <- c(new_col_order, groups[[i]], paste0("type_", i))
}
# 移除最后多余的type后缀列（因为最后一组不需要再插type）
new_col_order <- new_col_order[-length(new_col_order)]

# 重组数据框
capacity_result <- capacity_ex[, new_col_order, drop = FALSE]

# 复制type值到后缀列
for (i in seq_along(groups)) {
  capacity_result[[paste0("type_", i)]] <- capacity_result$type
}

方案三：dplyr链式操作（适合组数较少的情况） #

library(dplyr)

capacity_result <- capacity_ex %>%
  # 在1_q6后插入type_1
  mutate(type_1 = type, .after = "1_q6") %>%
  # 在2_q6后插入type_2
  mutate(type_2 = type, .after = "2_q6")
# 若有更多组，继续添加mutate语句即可

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