R Plotly热力图如何用ifelse实现标注文字颜色的条件设置？

问题背景 #

需要实现：热力图数据中Z值大于5时，标注文字显示白色；Z值≤5时显示黑色，但直接在add_annotations的font参数中使用公式映射无法生效。

问题原因 #

Plotly的add_annotations函数不支持在font参数内直接用公式（~）语法绑定数据列，必须通过显式赋值的方式设置每个标注的字体颜色。

解决方法 #

方法一：提前生成颜色列，循环添加标注 #

先在数据框中生成对应标注颜色的列，再逐个添加标注并指定颜色：

library(tibble)
library(plotly)

df <- tibble(Y=c("A","B","C"),
             X=c(1,2,3),
             Z=c(1,5,10))

# 生成标注颜色列
df <- df %>% mutate(text_color = ifelse(Z > 5, "white", "black"))

# 初始化热力图
p <- plot_ly(df, x = ~X, y = ~Y, z = ~Z, type = 'heatmap')

# 循环添加每个标注
for (i in 1:nrow(df)) {
  p <- p %>% add_annotations(
    x = df$X[i],
    y = df$Y[i],
    text = df$Z[i],
    showarrow = FALSE,
    font = list(color = df$text_color[i])
  )
}

p

方法二：构建Plotly对象后手动修改标注颜色 #

先创建基础热力图和标注，再通过plotly_build获取底层对象，批量修改标注的字体颜色：

library(tibble)
library(plotly)

df <- tibble(Y=c("A","B","C"),
             X=c(1,2,3),
             Z=c(1,5,10))

# 创建带标注的基础热力图
p <- df %>%
  plot_ly(x = ~X, y = ~Y, z = ~Z, type = 'heatmap') %>%
  add_annotations(text = ~Z, showarrow = FALSE)

# 构建Plotly对象并修改颜色
p_build <- plotly_build(p)
text_colors <- ifelse(df$Z > 5, "white", "black")
for (i in seq_along(text_colors)) {
  p_build$x$layout$annotations[[i]]$font$color <- text_colors[i]
}

p_build

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