基于Plotly自动创建下拉选择按钮的技术实现问题

问题背景 #

我有一个记录5年内与49个不同委员会互动次数的DataFrame，简化示例数据如下：

library(dplyr)
df <- data.frame(Year = c(2019, 2020, 2021, 2022, 2023),
                 Committees = c("House Foreign Affairs",
                                "House Foreign Affairs",
                                "House Foreign Affairs",
                                "House Foreign Affairs",
                                "House Foreign Affairs",
                                "House Judiciary",
                                "House Judiciary",
                                "House Judiciary",
                                "House Judiciary",
                                "House Judiciary",
                                "Senate Appropriations",
                                "Senate Appropriations",
                                "Senate Appropriations",
                                "Senate Appropriations",
                                "Senate Appropriations",
                                "Senate Oversight",
                                "Senate Oversight",
                                "Senate Oversight",
                                "Senate Oversight",
                                "Senate Oversight"), 
                 n = c(4,8,4,6,2,7,4,8,2,8,1,4,3,6,4,8,4,3,8,7))

df <- df %>% 
  mutate(across(c(1:2), factor)) %>%
  arrange(Year)

用ggplot2绘制折线图时，49个类别会导致图表杂乱，因此尝试转为Plotly交互式图表并通过下拉按钮切换委员会，但代码报错Error: object 'y_axis_var_names' not found，出错代码如下：

create_buttons <- function(df, y_axis_var_names) {
  lapply(
    y_axis_var_names,
    FUN = function(var_name, df) {
      button <- list(
        method = 'restyle',
        args = list(list(y = list(df[, var_name]),x = list(df[, var_name]))),
        label = sprintf('Show %s', var_name)
      )
    },
    df
  )
  
}

y_axis_var_names <- c("House Foreign Affairs",
                        "House Judiciary",
                        "Senate Appropriations",
                        "Senate Oversight")

plot %>%
  ggplotly(type = 'scatter',
           mode = 'lines') %>%
  layout(xaxis = list(domain = c(0.1, 1)),
         yaxis = list(title = "y"),
         updatemenus = list(
      list(
        y = 0.7,
        buttons = create_buttons(plot, y_axis_var_names))))

错误原因 #

1. 参数传递错误：调用create_buttons时传入的是ggplot对象plot，但函数需要原始数据框df，导致内部无法正确获取数据
2. 数据筛选逻辑错误：原数据是长格式，不能直接用df[, var_name]获取y值，需要筛选对应委员会的n列数据
3. 函数无返回值：lapply内部的函数没有返回button对象，导致生成的按钮列表为空
4. ggplotly参数冲突：无需手动指定type='scatter'和mode='lines'，ggplot的图层已经定义了这些属性

修正后的完整代码 #

library(dplyr)
library(ggplot2)
library(plotly)

# 1. 准备数据（调整年份格式为数值，适配Plotly）
df <- data.frame(Year = c(2019, 2020, 2021, 2022, 2023),
                 Committees = c("House Foreign Affairs",
                                "House Foreign Affairs",
                                "House Foreign Affairs",
                                "House Foreign Affairs",
                                "House Foreign Affairs",
                                "House Judiciary",
                                "House Judiciary",
                                "House Judiciary",
                                "House Judiciary",
                                "House Judiciary",
                                "Senate Appropriations",
                                "Senate Appropriations",
                                "Senate Appropriations",
                                "Senate Appropriations",
                                "Senate Appropriations",
                                "Senate Oversight",
                                "Senate Oversight",
                                "Senate Oversight",
                                "Senate Oversight",
                                "Senate Oversight"), 
                 n = c(4,8,4,6,2,7,4,8,2,8,1,4,3,6,4,8,4,3,8,7))

df <- df %>% 
  mutate(Year = as.numeric(as.character(Year)),
         Committees = factor(Committees)) %>%
  arrange(Year)

# 2. 修正按钮生成函数
create_buttons <- function(df, committee_names) {
  lapply(committee_names, function(name) {
    # 筛选当前委员会的互动次数数据
    filtered_data <- df[df$Committees == name, ]
    list(
      method = 'restyle',
      args = list(
        list(y = list(filtered_data$n), x = list(filtered_data$Year))
      ),
      label = name
    )
  })
}

# 3. 生成Plotly交互式图表
committee_list <- unique(df$Committees)

# 基础ggplot图（默认显示第一个委员会）
base_plot <- df %>%
  filter(Committees == committee_list[1]) %>%
  ggplot(aes(Year, n)) +
  geom_line(color = "#1f77b4") +
  geom_point(color = "#1f77b4") +
  theme_bw() +
  labs(y = "互动次数", x = "年份")

# 转为Plotly并添加下拉按钮（含重置选项）
ggplotly(base_plot) %>%
  layout(
    updatemenus = list(
      list(
        y = 0.9,
        x = 0.1,
        showactive = TRUE,
        buttons = c(
          # 重置按钮：显示所有委员会
          list(
            method = 'restyle',
            args = list(
              list(y = lapply(committee_list, function(x) df$n[df$Committees == x]),
                   x = lapply(committee_list, function(x) df$Year[df$Committees == x]),
                   color = lapply(1:length(committee_list), function(i) ggplot2::scale_color_hue()(length(committee_list))[i])
              )
            ),
            label = "显示所有委员会"
          ),
          # 各委员会切换按钮
          create_buttons(df, committee_list)
        )
      )
    )
  )

针对49个委员会的优化方案 #

• 分组下拉菜单：按House/Senate拆分多个下拉按钮，减少单个菜单的选项数量，提升操作便捷性
• 固定颜色映射：预先给每个委员会分配固定颜色，切换时保持颜色一致，增强用户记忆
• 增强hover提示：在ggplot中添加aes(text = paste("年份:", Year, "\n互动次数:", n))，再在ggplotly中设置tooltip = "text"，显示更详细的交互信息
• 按钮样式优化：通过direction = "down"设置下拉方向，添加activebackground属性高亮当前选中按钮

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