如何在ggplot中为geom_segment添加基于变量的两端渐变色？

要实现线段从var1对应颜色渐变到var2对应颜色的效果，无需手动为15个比较设置渐变，可以通过扩展数据生成渐变点或者使用ggforce包的专用函数来实现，以下是两种可行方案：

方案1：数据扩展+基础geom_segment #

这种方法通过将每个线段拆分为多个连续的小线段，每个小线段的颜色逐步从起始色过渡到结束色，兼容性强，无需额外依赖（仅需tidyverse）。

完整代码： #

library(tidyverse)

# 定义颜色映射（与原代码保持一致）
color_map <- c(
  'ControlD1' = "limegreen", 
  'HSP90iD1' = "firebrick1", 
  'PRRTD1' = "deepskyblue", 
  'CombinationD1' = "orchid1", 
  'HSP90iD3' = "firebrick4", 
  'PRRTD3' = "deepskyblue3", 
  'CombinationD3' = "purple3"
)

# 处理比较数据：移除分组，添加颜色和数值化的x轴位置
tb_links <- as.data.frame(unique(tb_deg[, c("comparison", "var1", "var2")])) %>%
  ungroup() %>%
  mutate(
    start_color = color_map[as.character(var1)],
    end_color = color_map[as.character(var2)],
    # 将离散x轴转为数值，匹配原scale_x_discrete的顺序
    x1 = match(as.character(var1), rev(levels(md$group))),
    x2 = match(as.character(var2), rev(levels(md$group)))
  )

# 扩展每个线段为100个渐变点（n值越大，渐变越平滑）
tb_gradient <- tb_links %>%
  rowwise() %>%
  mutate(
    x = list(seq(x1, x2, length.out = 100)),
    color = list(colorRampPalette(c(start_color, end_color))(100))
  ) %>%
  unnest(c(x, color))

# 绘制图形
p_left <- ggplot() +
  # 绘制渐变线段（连接相邻的渐变点）
  geom_segment(
    data = tb_gradient,
    aes(x = x, xend = lead(x), y = comparison, yend = comparison, color = color),
    size = 2, show.legend = FALSE, na.rm = TRUE
  ) +
  # 绘制原始分组点
  geom_point(data = tb_deg, aes(y=comparison, x=var1, fill=var1), shape=21, size=5, show.legend = FALSE) +
  geom_point(data = tb_deg, aes(y=comparison, x=var2, fill=var2), shape=21, size=5, show.legend = FALSE) +
  # 颜色映射设置
  scale_fill_manual(values = color_map) +
  scale_color_identity() +
  # 原代码中的轴与主题设置
  scale_x_discrete(limits = rev(levels(md$group))) +
  theme_void() +
  labs(x=NULL, y=NULL, title = NULL) +
  theme(
    legend.position = "none",
    aspect.ratio = 2,
    axis.text.x = element_text(angle=90, hjust=1),
    axis.text.y = element_blank()
  )

print(p_left)

方案2：使用ggforce包的geom_link2 #

ggforce包的geom_link2专门用于绘制渐变线段，代码更简洁，无需手动扩展数据。

完整代码： #

library(tidyverse)
library(ggforce)

# 定义颜色映射
color_map <- c(
  'ControlD1' = "limegreen", 
  'HSP90iD1' = "firebrick1", 
  'PRRTD1' = "deepskyblue", 
  'CombinationD1' = "orchid1", 
  'HSP90iD3' = "firebrick4", 
  'PRRTD3' = "deepskyblue3", 
  'CombinationD3' = "purple3"
)

# 处理数据，添加数值化的x轴位置
tb_links <- as.data.frame(unique(tb_deg[, c("comparison", "var1", "var2")])) %>%
  ungroup() %>%
  mutate(
    x1 = match(as.character(var1), rev(levels(md$group))),
    x2 = match(as.character(var2), rev(levels(md$group)))
  )

# 绘制图形
p_left <- ggplot(tb_deg) +
  # 绘制渐变线段：stat(index)为每个线段内0到1的插值，按comparison分组实现独立渐变
  ggforce::geom_link2(
    data = tb_links,
    aes(
      y = comparison,
      x = x1,
      xend = x2,
      colour = stat(index),
      group = comparison
    ),
    size = 2, show.legend = FALSE, n = 100,
    # 手动指定每个线段的起始和结束颜色
    start_color = color_map[tb_links$var1],
    end_color = color_map[tb_links$var2]
  ) +
  # 绘制原始分组点
  geom_point(aes(y=comparison, x=var1, fill=var1), shape=21, size=5, show.legend = FALSE) +
  geom_point(aes(y=comparison, x=var2, fill=var2), shape=21, size=5, show.legend = FALSE) +
  # 颜色设置
  scale_fill_manual(values = color_map) +
  scale_colour_identity(guide = "none") +
  # 原代码中的轴与主题设置
  scale_x_discrete(limits = rev(levels(md$group))) +
  theme_void() +
  labs(x=NULL, y=NULL, title = NULL) +
  theme(
    legend.position = "none",
    aspect.ratio = 2,
    axis.text.x = element_text(angle=90, hjust=1),
    axis.text.y = element_blank()
  )

print(p_left)

说明： #

• 两种方案均无需手动为15个比较单独设置渐变，自动根据var1和var2的颜色映射生成对应渐变
• 方案1无需额外安装包，兼容性更好；方案2代码更简洁，适合快速实现
• 调整n参数可控制渐变平滑度，数值越大渐变越自然

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