如何在Shiny的selectInput交互下固定Plotly折线图的国家颜色

问题 #

我需要展示不同国家随时间变化的预期寿命，通过shiny::selectInput实现多选功能，但存在一个问题：当选中的国家数量变化时，对应折线的颜色会改变。例如仅选中阿富汗时，其折线为橙色；同时选中阿富汗和阿尔巴尼亚时，阿富汗的折线变为浅蓝色，这会让用户感到困惑。希望让每个国家的颜色独立于交互选中的国家数量，保持固定不变。

解决方法 #

要实现固定颜色，核心是为每个国家预先分配唯一的颜色，并建立国家-颜色的映射关系，让Plotly始终根据这个映射来渲染颜色，而非动态分配。具体步骤：

• 提取所有国家列表，基于调色板为每个国家生成固定颜色（调色板颜色不足时循环扩展）
• 创建命名向量存储国家与颜色的对应关系
• 在Plotly绘图时，将这个命名向量作为colors参数传入

修改后的完整代码 #

library(shiny)
library(bslib)
library(dplyr)
library(plotly)
library(gapminder)

# 为所有国家创建固定颜色映射
all_countries <- unique(gapminder$country)
# 扩展Paired调色板，确保每个国家都有专属颜色
color_palette <- rep(RColorBrewer::brewer.pal(12, "Paired"), length.out = length(all_countries))
country_color_map <- setNames(color_palette, all_countries)

ui <- page_sidebar(
    titlePanel("Life Expectations (1952-2007)"),
    sidebar = sidebar(
        selectInput(
            inputId = "country",
            label = "Country",
            choices = all_countries,
            multiple = TRUE
        )
    ),
    card(plotlyOutput("p"))
)

server <- function(input, output, session) {

    selected_data <- reactive({
        req(input$country)
        gapminder |>
            select(year, lifeExp, country) |>
            filter(country %in% input$country) |>
            arrange(year) |>
            na.omit() |>
            droplevels()
    })

    output$p <- renderPlotly({
        req(selected_data())
        plot_ly(
            data = selected_data(),
            x = ~year,
            y = ~lifeExp,
            color = ~country,
            colors = country_color_map,  # 使用预定义的颜色映射
            type = 'scatter',
            mode = 'lines+markers',
            marker = list(size = 10)
        )
    })
}

shinyApp(ui, server)

关键说明 #

• country_color_map是命名向量，每个国家名称对应一个固定颜色，即使调色板颜色数量少于国家总数，rep(..., length.out = length(all_countries))会循环扩展调色板，保证每个国家都有专属颜色
• Plotly在渲染时会根据country字段的值匹配country_color_map中的颜色，无论选中多少国家，每个国家的颜色都不会改变

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