如何在R的Plotly中为热力图设置分位数定义的颜色？

你的问题核心在于错误使用了Plotly热图的color参数——热图的color参数是用来映射z值的变量，而非直接传入单个单元格的颜色值，这导致Plotly自动重新映射颜色，进而引发颜色不匹配、图例和提示框异常。下面是修正后的完整方案：

问题分析 #

原代码中直接给每个数据点分配RGB颜色并传给color=~color，但Plotly的heatmap类型会将这个参数视为分类变量，自动重新分配颜色映射规则，完全忽略你手动设置的颜色。同时这种用法也无法让Plotly正确关联z值与分位数的对应关系，导致图例和提示框逻辑混乱。

修正后的代码 #

library(dplyr)
library(plotly)

# 设置种子保证结果可复现
set.seed(123)
# 原始数据
d <- data.frame(x = 1:10, y=1:10, z = rnorm(n = 10, mean = 0, sd = 100))

# 分位数概率
qtl_val <- c(0, 0.68, 0.95, 0.99, 0.997)

# 对应分位数区间的颜色
qtl_colors <- c(
  rgb(0,0,0),
  rgb(246/256,192/256,192/256),
  rgb(244/256,135/256,134/256),
  rgb(219/256,55/256,55/256),
  rgb(164/256,30/256,34/256))

# 计算z值对应的分位数临界值
qtl_z <- quantile(d$z, probs = qtl_val)

# 构建Plotly要求的离散colorscale格式
# 格式为[[概率1, 颜色1], [概率2, 颜色1], [概率2, 颜色2], [概率3, 颜色2], ...]，实现纯色区间而非渐变
colorscale <- list()
for (i in seq_along(qtl_val)) {
  if (i == 1) {
    colorscale <- append(colorscale, list(list(qtl_val[i], qtl_colors[i])))
  } else {
    # 衔接前一个区间的终点与当前区间的起点，确保颜色是块状
    colorscale <- append(colorscale, list(list(qtl_val[i-1], qtl_colors[i-1])))
    colorscale <- append(colorscale, list(list(qtl_val[i], qtl_colors[i])))
  }
}
# 添加最后一个区间的终点（对应最大概率1）
colorscale <- append(colorscale, list(list(1, qtl_colors[length(qtl_colors)])))

# 给数据添加分位数区间标签，用于hover提示
d <- d %>%
  mutate(quantile_bin = cut(z, breaks = qtl_z, include.lowest = TRUE,
                           labels = paste0(qtl_val[-length(qtl_val)]*100, "% - ", qtl_val[-1]*100, "%")))

# 绘制热图
plot_ly(data = d, x = ~x, y = ~y, z = ~z, 
        type = 'heatmap',
        colorscale = colorscale,
        zmin = min(d$z),
        zmax = max(d$z),
        colorbar = list(
          tickvals = qtl_z,
          ticktext = paste0(qtl_val*100, "%"),
          title = "分位数"
        ),
        hoverinfo = "text",
        text = ~paste0("X: ", x, "<br>Y: ", y, "<br>Z值: ", round(z, 2), "<br>分位数区间: ", quantile_bin)
) %>%
  layout(xaxis = list(title = "X"), yaxis = list(title = "Y"))

关键修改点说明 #

1. 构建正确的colorscale：按照Plotly热图的要求，用分位数概率构建离散颜色刻度，确保每个分位数区间对应你指定的纯色。
2. 关联z值与分位数：通过quantile计算z值的分位数临界值，用于设置颜色条的刻度，让图例直接对应分位数。
3. 自定义hover提示：添加分位数区间标签，让提示框清晰展示每个单元格的z值和所属分位数区间。
4. 规范热图参数：使用z=~z作为颜色映射的核心变量，而非直接传入颜色，让Plotly正确识别颜色与z值的对应关系。

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