ggplot2绘制折线图如何按y值正负分区着色消除颜色跳变

问题说明 #

绘制折线图的需求：

• 所有y > 0的线条部分显示为深绿色
• 所有y ≤ 0的线条部分显示为红色

原有代码存在的缺陷：
geom_line() 基于离散数据点连接线段，不会对线段中间的连续值做颜色判断。原有逻辑仅在离散采样点上判断y > 0，判断结果会直接应用到该点之后的整段线段。当相邻两个采样点的y值正负异号时，线条穿过y=0轴时不会立即切换颜色，会保持当前段的颜色直到下一个采样点，出现红绿线条跨轴跳变的问题。

原有可复现代码如下：

library(tidyverse)
library(zoo)

set.seed(4133)

x = seq(1,100)
y = rnorm(100) %>%
  rollmean(.,k = 5,na.pad = T,fill = 0)

data = data.frame(x,y)

ggplot(data,aes(x = x, y = y)) +
  geom_line(color = ifelse(y > 0,"darkgreen","red")) +
  geom_hline(yintercept = 0)

修复方案 #

核心思路 #

找到所有线段与y=0轴的交点，把跨0的长线段从交点位置拆分为两个独立短段，拆分后每一段内部的y值正负完全一致，再给每段匹配对应颜色即可消除跨轴颜色跳变问题。

修复后代码 #

library(tidyverse)
library(zoo)

set.seed(4133)
x = seq(1,100)
y = rnorm(100) %>%
  rollmean(.,k = 5,na.pad = T,fill = 0)
data = data.frame(x,y)

# 计算所有跨0线段与y=0轴的交点
cross_points <- data %>%
  mutate(
    x_next = lead(x),
    y_next = lead(y)
  ) %>%
  # 筛选相邻点y值异号的跨0线段
  filter(y * y_next < 0) %>%
  mutate(
    # 线性插值计算交点坐标
    x_cross = x - y * (x_next - x)/(y_next - y),
    y_cross = 0
  ) %>%
  select(x = x_cross, y = y_cross)

# 合并原始点和交点，按x轴排序后给每段分配分组和颜色
plot_data <- bind_rows(data, cross_points) %>%
  arrange(x) %>%
  mutate(
    seg_id = cumsum(is.na(lag(y)) | y * lag(y) < 0),
    seg_col = ifelse(y > 0, "darkgreen", "red")
  )

# 绘图
ggplot(plot_data, aes(x = x, y = y)) +
  geom_line(aes(group = seg_id, color = I(seg_col)), linewidth = 1) +
  geom_hline(yintercept = 0)

实现说明 #

• 不需要对x轴做高密度插值，仅补全所有跨0位置的交点即可，渲染效率高，颜色切换位置完全精准
• 拆分后每段线段内部y值正负统一，不存在跨0情况，从根源上避免了颜色赋值错位的问题

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