如何在ggplot2中为图例添加层级子分类并实现主分类单一颜色块展示

嘿，我明白你想要的效果——既保留每个主栖息地类型的总面积柱状图，又能在图例里清晰展示主分类下的子分类，而且每个主分类只显示一个对应颜色块对吧？我给你一个自动化的解决方案，完美适配大规模数据：

先回顾你的需求和现有问题 #

你有一组层级化的栖息地数据：

set.seed(1)
hab_dat <- data.frame(
  habitat_type = rep(c("sea", "coast", "land"), times = 1, each = 3),
  habitat_name = c("rocky", "sandy", "seaweed", "beach", "pebbles", "rockpools", "fields", "hills", "forest"),
  area_km2 = sample(10:40, size =9)
)

你最初的代码已经能画出正确的总面积柱状图，但图例不够直观：

hab_dat %>% 
  group_by(habitat_type) %>% 
  summarise(area_km2 = sum(area_km2)) %>% 
  ggplot(aes(x = habitat_type, y = area_km2, fill = habitat_type)) +
  geom_bar(stat = "identity") +
  scale_fill_manual(values = c("gold", "forestgreen", "blue"))

你尝试的堆叠版本虽然接近，但出现了缺少主分类标题和同颜色多色块的问题，而且还改变了图表主体的展示形式——这不是你想要的。

自动化解决方案 #

我们可以通过手动构建层级图例标签，结合ggplot2的自定义图例功能来实现需求，同时完全保留原图表的主体样式：

步骤1：计算主分类总面积 #

先保留你最初的求和逻辑，得到每个栖息地类型的总面积数据：

library(tidyverse)

# 计算各主栖息地类型的总面积
hab_total <- hab_dat %>% 
  group_by(habitat_type) %>% 
  summarise(total_area = sum(area_km2))

步骤2：构建层级化图例标签 #

自动生成主分类+子分类的层级标签，主分类用粗体突出，子分类缩进显示：

# 整理主分类与子分类的映射关系，生成层级标签
hab_legend_labels <- hab_dat %>% 
  group_by(habitat_type) %>% 
  summarise(sub_habitats = paste0("  - ", habitat_name, collapse = "\n")) %>% 
  mutate(full_label = paste0("**", habitat_type, "**\n", sub_habitats))

# 构建颜色映射（支持手动指定或自动生成）
# 手动指定颜色（和你原来的设置一致）
color_map <- c(sea = "blue", coast = "gold", land = "forestgreen")
# 如果是大规模数据，可自动生成颜色：
# color_map <- setNames(scales::hue_pal()(n_distinct(hab_dat$habitat_type)), unique(hab_dat$habitat_type))

步骤3：绘制图表并自定义图例 #

用scale_fill_manual指定标签，再通过guides()调整图例的显示样式：

ggplot(hab_total, aes(x = habitat_type, y = total_area, fill = habitat_type)) +
  geom_bar(stat = "identity") +
  # 关联颜色和层级标签
  scale_fill_manual(values = color_map,
                    labels = setNames(hab_legend_labels$full_label, hab_legend_labels$habitat_type)) +
  # 自定义图例外观
  guides(fill = guide_legend(
    title = "Habitat Categories",
    keyheight = unit(2.5, "lines"), # 调整图例高度以容纳多行标签
    label.hjust = 0, # 标签左对齐
    label.position = "right"
  )) +
  # 图表标签美化
  labs(x = "Main Habitat Type", y = "Total Area (km²)", title = "Total Area of Island Habitats") +
  theme_minimal() +
  theme(
    legend.text = element_text(lineheight = 1.3) # 调整行高让子分类显示更清晰
  )

效果亮点 #

• 图表主体完全保留了你最初想要的单柱总面积样式，没有改变展示逻辑
• 图例是层级化的：每个主分类显示对应颜色块，下方缩进列出所属子分类，清晰直观
• 完全自动化，不管你有多少主分类和子分类，代码都能自动适配，不需要手动修改标签或颜色

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