R Plotly下拉菜单更新UMAP散点图：颜色方案与图例问题

我用R的Plotly生成UMAP图，想通过下拉菜单切换不同变量来改变散点颜色。目前代码能切换颜色，但显示的是红色，不是想要的Set1配色，而且没有对应颜色含义的图例。现有代码如下：

vars_df$orig.ident <- as.factor(vars_df$orig.ident)
vars_df$singlr_labels <- as.factor(vars_df$singlr_labels)

# Define dropdown buttons with explicit color mapping
dropdown_buttons <- list(
  list(method = "restyle",
       args = list("marker.color", list(as.numeric(vars_df$orig.ident))),
       label = "orig.ident"),
  list(method = "restyle",
       args = list("marker.color", list(as.numeric(vars_df$singlr_labels))),
       label = "singlr_labels")
)

# Your plot
vars_umap <- plot_ly(data = vars_df,
                     x = ~UMAP_1,
                     y = ~UMAP_2,
                     type = 'scatter',
                     mode = 'markers',
                     alpha = 0.7,
                     marker = list(color = as.numeric(vars_df$orig.ident), colors = "Set1"),
                     showlegend = TRUE
                      
) %>% layout(
  xaxis = list(
    tickvals = 'none',  
    showticklabels = FALSE  
  ),
  yaxis = list(
    tickvals = 'none', 
    showticklabels = FALSE  
  )
) %>%
layout(title = 'UMAP Plot',
       updatemenus = list(
         list(
           buttons = dropdown_buttons,
           direction = "down",
           showactive = TRUE
         )
       ))

vars_umap

问题根源 #

1. 配色失效：直接传递因子的数值编码给marker.color时，Plotly的restyle方法不会自动应用colors = "Set1"的配色规则，只会默认用红色系渲染数值。
2. 图例缺失：仅传递数值无法让Plotly识别分类层级，自然生成不了对应图例。

修复后的完整代码 #

# 先安装依赖包（如果未安装）
if (!require(RColorBrewer)) install.packages("RColorBrewer")
library(RColorBrewer)

# 转换因子变量
vars_df$orig.ident <- as.factor(vars_df$orig.ident)
vars_df$singlr_labels <- as.factor(vars_df$singlr_labels)

# 自定义函数：根据因子变量生成对应数量的Set1配色
get_set1_palette <- function(factor_col) {
  n_categories <- nlevels(factor_col)
  brewer.pal(n_categories, "Set1")
}

# 为两个变量生成对应的Set1配色映射
orig_palette <- get_set1_palette(vars_df$orig.ident)
names(orig_palette) <- levels(vars_df$orig.ident)

singlr_palette <- get_set1_palette(vars_df$singlr_labels)
names(singlr_palette) <- levels(vars_df$singlr_labels)

# 构建下拉按钮：直接传递每个数据点的颜色值
dropdown_buttons <- list(
  list(
    method = "restyle",
    args = list(
      "marker.color",
      lapply(vars_df$orig.ident, function(x) orig_palette[as.character(x)])
    ),
    label = "orig.ident"
  ),
  list(
    method = "restyle",
    args = list(
      "marker.color",
      lapply(vars_df$singlr_labels, function(x) singlr_palette[as.character(x)])
    ),
    label = "singlr_labels"
  )
)

# 初始化UMAP图：绑定分类变量生成图例，应用Set1配色
vars_umap <- plot_ly(data = vars_df,
                     x = ~UMAP_1,
                     y = ~UMAP_2,
                     type = 'scatter',
                     mode = 'markers',
                     alpha = 0.7,
                     color = ~orig.ident,  # 绑定分类变量自动生成图例
                     colors = orig_palette,  # 应用预定义的Set1配色
                     showlegend = TRUE
) %>%
  layout(
    xaxis = list(tickvals = NULL, showticklabels = FALSE),
    yaxis = list(tickvals = NULL, showticklabels = FALSE),
    title = 'UMAP Plot',
    updatemenus = list(
      list(
        buttons = dropdown_buttons,
        direction = "down",
        showactive = TRUE
      )
    )
  )

vars_umap

关键修改说明 #

• 用RColorBrewer生成对应分类数量的Set1配色，确保每个类别有唯一颜色。
• 下拉菜单中不再传递因子的数值编码，而是直接传递每个数据点对应的具体颜色值。
• 初始化图时通过color = ~orig.ident绑定分类变量，让Plotly自动生成对应图例。
• 替换了原代码中错误的tickvals = 'none'为tickvals = NULL，符合Plotly的参数要求。

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