强制 Plotly 相关热图色标在零时为白色 - R

Force Plotly correlation heatmap colorscale to be white at zero - R

我在 plotly 中生成了相关热图。比例尺从 -1 到 1。随着相关性正向增强,图块的颜色会变深红色。随着相关性负向增强,图块的颜色会变深蓝色。我需要零值才能变成白色。然而,颜色条只是根据数据集的分布选择零颜色。如何将零值强制为白色并位于颜色条的中间?我尝试使用 ,但无法使其适用于此热图。请帮帮我,我要疯了!

library(plotly)
library(magrittr)

# compute a correlation matrix
correlation <- round(cor(mtcars), 3)
nms <- names(mtcars)


colorlength <- 100

null_value <- (0 - min(correlation)) / (max(correlation) - min(correlation))        
border <- as.integer(null_value * colorlength)
colorscale <- as.list(1:colorlength)

#colorscale below zero
s <- scales::seq_gradient_pal("blue", "white", "Lab")(seq(0,1,length.out=border))
for (i in 1:border) {
  colorscale[[i]] <- c((i - 1) / colorlength, s[i])
}

#colorscale above zero
s <- scales::seq_gradient_pal("white", "red", "Lab")(seq(0,1,length.out=colorlength - border))
for (i in 1:(colorlength - border)) {
  colorscale[[i + border]] <- c((i + border) / colorlength, s[i])
}



plot_ly(x = nms, y = nms, z = correlation, 
            key = correlation, type = "heatmap", source = "heatplot",color = ~correlation,
            colorscale = colorscale,
            colorbar = list(len=1)) %>%
      layout(xaxis = list(title = ""), 
             yaxis = list(title = ""))

如何使用 ggplotggplotly:

library(tidyverse);
gg <- correlation %>%
    as.data.frame() %>%
    rownames_to_column("x") %>%
    gather(y, val, -x) %>%
    ggplot(aes(x, y, fill = val)) +
    geom_tile() +
    scale_fill_gradientn(
        colours = c("blue", "white", "red"),
        limits = c(-1, 1))

# Create plotly object
library(plotly);
ggplotly(gg);

说明:在ggplot中我们可以使用scale_fill_gradientn明确设置limits,确保白色对应于0的值。 ggplotly 然后将 ggplot 对象转换为 plotly 对象。

使用 ggplotly 非常简单,但我认为您也可以使用 plotly 来完成。你可以试试这个:

col3 <- colorRamp(c("red", "white", "blue"))

plot_ly(x = nms, y = nms, z = correlation, 
        key = correlation, type = "heatmap", source = "heatplot",color = col3,
        colorscale = colorscale,
        colorbar = list(len=1, limits = c(-1, 1))) %>%
    layout(xaxis = list(title = ""), 
           yaxis = list(title = ""))

使用 plotly,使用 graph_objects.Heatmap 对象,因为您知道热图值的限制,您可以设置 zmaxzmidzmin 参数分别为 1, 0, -1colorscale 参数为中间白色的任何色标,如 RdBu.