如何在R语言的数据框中为每个分组单独添加列标题

实现逻辑 #

首先将数据框所有列转为字符型，避免插入字符串格式的列名时出现类型不兼容报错；随后按group列拆分数据框，为每个子分组的头部插入列名行；最后合并所有处理后的子数据框即可得到目标结果。

测试用示例数据 #

df <- data.frame(
  "c1" = seq(1,5,1),
  "c2" = seq(1,5,1),
  "c3" = seq(1,5,1),
  "c4" = seq(1,5,1),
  "c5" = seq(1,5,1),
  "group" = c("A","A","B","B","C")
)

实现代码 #

方法1：tidyverse 方案（使用你提到的group_split()函数） #

需要先加载dplyr和purrr包：

library(dplyr)
library(purrr)

# 所有列转字符型
df_char <- df %>% mutate(across(everything(), as.character))

# 分组处理并合并
result <- df_char %>%
  group_split(group, .keep = TRUE) %>%
  map_dfr(~ bind_rows(as.list(names(df)), .x))

方法2：基础R方案（无需加载第三方包） #

# 所有列转字符型
df_char <- as.data.frame(lapply(df, as.character), stringsAsFactors = FALSE)
# 按group拆分
df_list <- split(df_char, df_char$group)
# 每个子分组插入列名后合并
result <- do.call(rbind, lapply(df_list, function(x) rbind(names(df), x)))
# 重置行名
rownames(result) <- NULL

注意事项 #

最终输出的所有列都是字符型，如果后续需要对数值列做计算，可筛选掉列名行后单独转换列类型即可。

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