如何使用ggplot2设置自定义非线性色阶条？

你可以通过手动控制色阶的映射位置，或者创建自定义数学变换函数来实现非线性的色阶条。下面是两种实用的方案，你可以根据自己的需求调整：

方法1：手动指定非线性色阶映射（推荐） #

这种方式能让你精准控制色阶在特定区间的细腻程度，比如在你关注的数值区间增加更多颜色断点，让颜色变化更密集。示例如下：

library(ggplot2)
library(scales)

a = data.frame(
  x=c(1,2,3,3,4,5,3,2,3,4), 
  y=c(1,2,3,6,4,5,3,2,3,2), 
  z=c(-4,-3.5,-2,-1,0,0.2,0.45,0.6,0.9,1)
)

# 自定义z值断点：在需要细腻区分的区间（比如0附近）添加更多节点
custom_z_breaks <- c(-5, -4, -3, -2, -1, -0.5, 0, 0.2, 0.4, 0.6, 0.8, 1)
# 将z断点转换为0-1范围的数值，告诉ggplot每个颜色对应的位置
custom_values <- rescale(custom_z_breaks, to = c(0,1), from = c(-5,1))

ggplot()+ 
  geom_point(data=a, aes(x,y,fill=z), colour = "black", pch = 21, size = 3, stroke = 0.1)+ 
  scale_fill_gradientn(
    colors = c("blue", "lightblue", "white", "pink", "red"), # 可按需调整颜色
    breaks = custom_z_breaks,
    limits = c(-5,1), 
    values = custom_values, # 核心参数：实现非线性色阶的关键
    guide = guide_colorbar(barwidth = 0.8, barheight = 18), 
    oob = scales::squish, 
    trans = "identity"
  )

关键说明： #

• custom_z_breaks：你可以在需要重点区分的数值区间（比如接近0的部分）添加更多断点，让该区间的颜色过渡更平滑细腻。
• custom_values：通过rescale函数将自定义的z断点映射到0-1的范围，让ggplot明确知道每个颜色对应的具体z值位置，从而实现非线性渐变效果。

方法2：创建自定义数学变换函数 #

如果你希望用数学变换（比如对数、平方根）来实现非线性色阶，可以用scales::trans_new创建专属变换函数。比如让正数部分的色阶更密集，负数部分保持线性：

library(ggplot2)
library(scales)

a = data.frame(
  x=c(1,2,3,3,4,5,3,2,3,4), 
  y=c(1,2,3,6,4,5,3,2,3,2), 
  z=c(-4,-3.5,-2,-1,0,0.2,0.45,0.6,0.9,1)
)

# 定义自定义非线性变换：正数用平方根压缩，负数保持线性
custom_trans <- trans_new(
  name = "custom_nonlinear",
  transform = function(x) {
    ifelse(x >= 0, sqrt(x), x)
  },
  inverse = function(x) {
    ifelse(x >= 0, x^2, x) # 逆变换用于色阶条的标签显示
  }
)

ggplot()+ 
  geom_point(data=a, aes(x,y,fill=z), colour = "black", pch = 21, size = 3, stroke = 0.1)+ 
  scale_fill_gradientn(
    colors = c("blue","white","red"), 
    breaks=c(-5,-4,-3,-2,-1,0,0.2,0.4,0.6,0.8,1),
    limits=c(-5,1), 
    guide = guide_colorbar(barwidth = 0.8, barheight = 18), 
    oob = scales::squish, 
    trans = custom_trans # 应用自定义变换
  )

关键说明： #

• 你可以修改transform中的函数逻辑，实现不同的非线性效果，比如对负数部分用对数变换（注意要处理0和负数的边界问题）。

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