如何让ggplotly图例与ggplot保持一致？需动态适配场景

这个问题被多次提及，但很难找到适配可复现案例的简洁方案。需求是实现ggplotly生成的图例与原生ggplot完全一致，且方案需支持动态场景：有时不使用多种shape美学，水平线的数量也会动态变化。

可复现代码 #

library(pacman)
pacman::p_load(dplyr, ggplot2, plotly, shiny)
set.seed(123) # 固定随机种子保证可复现
data <- iris %>% mutate(
  my_shape = sample(c(T, F), size = nrow(iris), replace = T)
)

new_data <- data.frame(
  type = c("a", "b", "c"),
  value = c(2, 3, 4)
)

g <- ggplot(data) + 
  geom_point(aes(x = Sepal.Length, y = Petal.Width, shape = my_shape, fill = Species)) + 
  geom_hline(data = new_data, aes(yintercept = value, linetype = type, color = type)) +
  scale_fill_manual(
    values = c(
      "setosa" = "olivedrab4",
      "versicolor" = "steelblue4",
      "virginica" = "red"
    )
  ) +
  scale_shape_manual(values = c("TRUE" = 24, 'FALSE' = 21)) +
  guides(fill = guide_legend(override.aes = list(shape = 21)))

plot(g)

图例效果对比 #

• 原生ggplot图例：
  

• ggplotly默认转换后的图例：
  

动态适配解决方案 #

通过手动调整plotly底层的trace和布局属性，实现和ggplot完全一致的图例效果，且自动适配shape美学是否存在、水平线数量变化的场景：

# 将ggplot对象转换为plotly对象
p <- ggplotly(g)

# 构建plotly底层对象，便于修改属性
pb <- plotly_build(p)

# 1. 调整Species（fill）的图例符号：统一使用shape=21（空心正方形）
fill_trace_indices <- which(sapply(pb$x$data, function(x) x$name %in% levels(data$Species)))
for (i in fill_trace_indices) {
  pb$x$data[[i]]$marker$symbol <- "square-open"
}

# 2. 调整my_shape的图例符号：匹配ggplot中设置的shape值（24=空心上三角，21=空心正方形）
shape_trace_indices <- which(sapply(pb$x$data, function(x) x$name %in% c("TRUE", "FALSE")))
shape_symbol_map <- c("TRUE" = "triangle-up-open", "FALSE" = "square-open")
for (i in shape_trace_indices) {
  pb$x$data[[i]]$marker$symbol <- shape_symbol_map[pb$x$data[[i]]$name]
}

# 3. 调整水平线（hline）的图例：匹配linetype和颜色，确保显示为线条
hline_trace_indices <- which(sapply(pb$x$data, function(x) x$type == "scatter" && x$mode == "lines"))
linetype_map <- c("a" = "solid", "b" = "dot", "c" = "dash") # 对应ggplot默认线型映射
for (i in hline_trace_indices) {
  pb$x$data[[i]]$showlegend <- TRUE
  pb$x$data[[i]]$line$dash <- linetype_map[pb$x$data[[i]]$name]
}

# 4. 调整图例分组间距，和ggplot布局一致
pb$x$layout$legend$tracegroupgap <- 10
pb$x$layout$legend$itemsizing <- "constant"

# 显示调整后的图
pb

方案适配性说明 #

• 如果场景中不使用shape美学，shape_trace_indices会返回空向量，对应的循环不会执行，不影响其他图例的调整；
• 水平线数量动态变化时，hline_trace_indices会自动匹配所有线条类型的trace，无需修改代码即可适配。

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