如何移除ggplot图例中未在数据中出现的自定义配色项

动态移除未出现分类的图例项 #

你可以通过动态过滤预定义配色向量的方式，仅保留当前数据中存在的分类对应的配色，多余的分类自然不会出现在图例中，修改后的函数示例如下：

library(tidyverse)

scale_color_sex <- function(data, col, ...){
  # 预定义性别配色
  sex_pal <- c(
    female = "#9986A5", 
    hermaphroditic = "#79402E", 
    male = "#CCBA72",
    none = "#0F0D0E",
    .default = "red" # ggplot2 3.5.0+ 支持，未匹配分类的默认颜色
  )
  # 提取当前数据中性别列的非缺失唯一值
  present_vals <- unique(na.omit(data[[col]]))

  scale_color_manual(
    ...,
    values = sex_pal,
    limits = present_vals,
    drop = TRUE
  )
}

# 调用示例
starwars %>% 
  ggplot(aes(x = height, y = birth_year, color = sex)) +
  geom_point() + 
  scale_color_sex(data = starwars, col = "sex")

如果你的ggplot2版本低于3.5.0，可以将默认颜色逻辑写在配色过滤环节：

scale_color_sex <- function(data, col, ...){
  sex_pal <- c(
    female = "#9986A5", 
    hermaphroditic = "#79402E", 
    male = "#CCBA72",
    none = "#0F0D0E"
  )
  present_vals <- unique(na.omit(data[[col]]))
  # 匹配存在的分类，未匹配的分类赋值红色
  used_pal <- purrr::map_chr(present_vals, ~ifelse(.x %in% names(sex_pal), sex_pal[.x], "red"))
  names(used_pal) <- present_vals

  scale_color_manual(..., values = used_pal)
}

正则匹配分类值分配配色 #

配色向量的键本身不支持直接写正则表达式，但你可以自定义匹配逻辑实现该需求，示例如下：

scale_color_sex_regex <- function(data, col, ...){
  present_vals <- unique(na.omit(data[[col]]))
  # 定义正则-配色对应规则
  match_rules <- list(
    ".*[Ff]emale.*" = "#9986A5",
    ".*[Mm]ale.*" = "#CCBA72",
    ".*hermaphroditic.*" = "#79402E",
    ".*none.*" = "#0F0D0E"
  )
  # 遍历分类值匹配对应颜色
  used_pal <- purrr::map_chr(present_vals, function(val){
    for (pattern in names(match_rules)) {
      if (stringr::str_detect(val, pattern)) return(match_rules[[pattern]])
    }
    return("red")
  })
  names(used_pal) <- present_vals

  scale_color_manual(..., values = used_pal)
}

配色方案优化建议 #

• 优先选择色弱友好的配色，可使用scales::show_col()预览配色效果，用colorspace::deutan()测试红绿色弱辨识度，避免使用区分度过低的颜色组合。
• 可以将配色逻辑封装为独立的调色板函数，同时适配color和fill两种标度，提升复用性。
• 如果分类取值固定，建议将配色向量存储为全局变量，确保多个图表的配色一致性。

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