使用pivot_longer后自动重排tibble因子水平匹配ggplot图例顺序

需求：使用pivot_longer处理tibble后，希望根据Count列中每个因子的最大值自动重排Name的因子水平，无需手动指定顺序，最终让ggplot绘图时图例与图表中的线条顺序对齐。

用户原始代码：

df <- tibble(Year=c(1900, 1901, 1903), Foo=c(9,30, 60), Bar=c(3,20, 50)) %>%
 pivot_longer(!Year, names_to = "Name", values_to = "Count")
df$Name <- factor(df$Name)

当前levels(df$Name)为默认字母顺序[1] "Bar" "Foo"，期望重排为[1] "Foo" "Bar"（按Count最大值排序）。

解决方案 #

方法1：使用forcats::fct_reorder()（推荐，tidyverse风格） #

fct_reorder()可以直接根据指定变量的统计量重排因子水平，无需额外计算步骤，完美适配管道流：

library(tidyverse)

df <- tibble(Year=c(1900, 1901, 1903), Foo=c(9,30, 60), Bar=c(3,20, 50)) %>%
  pivot_longer(!Year, names_to = "Name", values_to = "Count") %>%
  # 按Count的最大值降序排列Name的因子水平
  mutate(Name = fct_reorder(Name, Count, .fun = max, .desc = TRUE))

验证：levels(df$Name)会返回[1] "Foo" "Bar"，符合预期。

方法2：手动计算最大值后重排（无需额外包） #

如果不想依赖forcats包，可以手动分组计算每个Name的最大Count，再以此排序因子水平：

library(tidyverse)

df <- tibble(Year=c(1900, 1901, 1903), Foo=c(9,30, 60), Bar=c(3,20, 50)) %>%
  pivot_longer(!Year, names_to = "Name", values_to = "Count") %>%
  group_by(Name) %>%
  mutate(max_count = max(Count)) %>%
  ungroup() %>%
  # 按max_count降序提取唯一Name作为因子水平
  mutate(Name = factor(Name, levels = unique(Name[order(-max_count)]))) %>%
  select(-max_count) # 移除临时计算列

绘图验证 #

使用你的ggplot代码绘图，图例顺序会与线条的最大值顺序完全对齐：

p <- ggplot(df, aes(x = Year, y = Count)) +
  geom_line(aes(colour = Name))
p

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