Plotly绘制3D热力图：实现颜色渐变与Z轴独立

我明白你想要的效果——让Plotly 3D表面图的高度对应freq列，同时颜色渐变独立绑定variable列，就像你用ggplot2实现的热力图那样。其实Plotly的add_surface函数支持通过surfacecolor参数自定义颜色数据源，完美解决这个问题，下面是具体的实现步骤和代码：

步骤1：准备数据（重现你的数据集） #

先确保我们的数据集一致，这里加上随机种子让结果可复现：

set.seed(123)
dat <- data.frame(
  "cat1" = sort(rep(seq(1:6), 6)), 
  "cat2" = rep(seq(1:6), 6), 
  "variable" = round(rnorm(36), 1), 
  "freq" = seq(100, (100-35))
)

步骤2：生成高度和颜色对应的矩阵 #

分别将freq和variable转换为与cat1、cat2对应的二维矩阵：

# 对应Z轴高度的freq矩阵
mat_freq <- tapply(dat$freq, list(dat$cat1, dat$cat2), sum)
# 对应颜色渐变的variable矩阵
mat_variable <- tapply(dat$variable, list(dat$cat1, dat$cat2), sum)

步骤3：用Plotly绘制独立控制的3D热力图 #

核心就是通过surfacecolor参数传入颜色矩阵，实现Z轴高度与颜色的解耦：

library(plotly)

plot_ly() %>%
  add_surface(
    z = ~mat_freq,          # Z轴高度绑定freq数据
    surfacecolor = ~mat_variable, # 颜色渐变绑定variable数据
    colorscale = "Viridis", # 可选：指定你喜欢的颜色方案
    colorbar = list(title = "Variable") # 给颜色条添加标题
  ) %>%
  layout(
    scene = list(
      zaxis = list(title = "Freq"), # Z轴（高度）标题
      xaxis = list(title = "cat1"),
      yaxis = list(title = "cat2")
    )
  )

关键说明 #

• surfacecolor是实现独立控制的核心：它接受一个与z矩阵维度完全相同的矩阵，专门用来定义表面的颜色，这样就彻底把高度（Z轴）和颜色的数据源分开了。
• 你可以根据需求替换colorscale的值（比如"RdBu"、"Plasma"等），或者自定义颜色渐变方案。
• 如果需要添加类似ggplot2里的文本标签，可以结合add_text函数实现，不过3D场景下的文本需要指定x/y/z坐标，有需要的话可以再补充细节。

内容的提问来源于stack exchange，提问作者P. Denelle