如何在R中用kableExtra动态添加分组行标签

问题描述 #

我有一个包含分组列pos的数据框，希望用R的kableExtra包生成带分组行标签的HTML表格。已知可以用pack_rows()手动实现，但实际数据中pos的唯一值数量和每组行数都不固定，需要自动化的动态实现方法。示例数据及初始代码如下：

library(tidyverse)
library(kableExtra)

set.seed(10)
name <- paste("Name", LETTERS[1:15])
var1 <- rep(rnorm(15, 100, 20))
var2 <- rep(rnorm(15, 100, 20))
var3 <- rep(rnorm(15, 100, 20))
var4 <- rep(rnorm(15, 100, 20))
pos <- c(rep("DEF", 5), rep("MID", 5), rep("FWD", 5))

df <- data.frame(name, var1, var2, var3, var4, pos)

agg_df <- bind_rows(
  df,
  df %>% group_by(pos) %>%
    summarise_at(vars(var1:var4), mean) %>%
      mutate(name = "mean")
  ) %>% 
  arrange(pos, name)

agg_df %>%
  kbl() %>%
  kable_styling(bootstrap_options = c("striped", "hover", "condensed"),
                fixed_thead = TRUE)

动态实现方案 #

核心思路是先计算每个pos分组的总行数（包含均值行），再通过批量调用pack_rows()完成分组标签的添加，无需手动指定每个分组的索引。

步骤1：计算分组行数与名称 #

从处理好的agg_df中提取每个分组的名称和对应的总行数：

# 获取每个pos分组的总行数（含均值行）
group_row_counts <- agg_df %>%
  group_by(pos) %>%
  tally() %>%
  pull(n)

# 获取所有分组名称
group_names <- unique(agg_df$pos)

步骤2：批量添加分组行标签 #

可以用两种方式实现批量调用：

方式1：使用purrr::reduce2()（函数式编程风格）

agg_df %>%
  select(-pos) %>%  # 移除pos列，避免重复显示分组信息
  kbl() %>%
  kable_styling(bootstrap_options = c("striped", "hover", "condensed"),
                fixed_thead = TRUE) %>%
  # 动态遍历分组，添加分组行标签
  purrr::reduce2(
    .x = group_names,
    .y = group_row_counts,
    .f = function(kable_obj, group_name, row_count) {
      pack_rows(kable_obj, group_name, row_count = row_count)
    }
  )

方式2：使用Base R循环

# 先创建基础的kable表格对象
kable_table <- agg_df %>%
  select(-pos) %>%
  kbl() %>%
  kable_styling(bootstrap_options = c("striped", "hover", "condensed"),
                fixed_thead = TRUE)

# 循环遍历每个分组，添加分组行标签
for (i in seq_along(group_names)) {
  kable_table <- pack_rows(kable_table, group_names[i], row_count = group_row_counts[i])
}

# 输出最终表格
kable_table

说明 #

• 两种方式都能自动适配pos分组数量和每组行数的变化，无需手动修改代码
• 必须移除pos列，否则表格会重复显示分组信息，分组标签的展示会更简洁清晰

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