如何为 R 和 plotly 中的热图设置分位数定义的 colors？

Question

我想要 plot 中的矩阵 R 和 plotly，其中色标基于分位数值。 这是一个独立的可重现示例：

library(dplyr)
library(plotly)

# Data to plot
d <- data.frame(x = 1:10, y=1:10, z = rnorm(n = 10, mean = 0, sd = 100))

# Quantile values
qtl_val <- c(0, 0.68, 0.95, 0.99, 0.997)

# Colors to use
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))

# Function to assign a color to each quantile window
set_color <- function(z,qtl_leg){
  for( i in 1:nrow(qtl_leg)){
    min_val <- qtl_leg$val[i]
    max_val <- ifelse( !is.na(qtl_leg$lead_val[i]), qtl_leg$lead_val[i], Inf)
    if( z >= min_val & z <= max_val ) return(qtl_leg$color[i])
  }
}

qtl_leg <- data.frame(val=round(as.numeric(quantile(d$z, probs=qtl_val)))) %>%
  mutate(lead_val = lead(val)) %>%
  mutate(color = qtl_colors)

d <- d %>%
  rowwise() %>%
  mutate(color = set_color(z,qtl_leg) ) %>%
  as.data.frame()

# Draw the matrix
plot_ly(data=d, x=~x, y=~y, z=~z, color=~color, type='heatmap' )

plot 与 data.frame 中设置的 colors 不匹配，图例很奇怪，提示工具文本也是如此。 简而言之，我出了点问题，但我不知道是什么。

Answer 1

您可以使用zauto = FALSE创建自定义比例并指定zmin和zmax ，如下所示：

library(plotly)
library(dplyr)
plot_ly(data=d, x=~x, y=~y, z=~z, 
        color=~color, type='heatmap', 
        zauto = FALSE,
        zmin = qtl_val[1],
        zmax = qtl_val[5])

Output：