如何用Plotly为Pandas多列数据添加下拉选择菜单绘制图表

我有一个Pandas DataFrame，想要用Plotly绘制柱状图，并且通过下拉菜单选择不同的股票名称列来查看对应数据。尝试相关方法后，不确定如何把动态变化的列名添加到updatemenus中，附上当前代码和参考DataFrame：

import plotly.graph_objects as go
# plotly setup
fig = go.Figure()

# Add traces, starting from First Column. Each column is a stock data.
for mycolumn in df.columns[1:]:
    fig.add_traces( go.Scatter(x=list(df.Quarters), y=list(df[mycolumn])) )

updatemenus = None

for mycolumn in df.columns[1:]:
    # construct menus
    updatemenus = [{'buttons': 
                    [{'method': 'update',
                      'label' : column, # what?
                      'args'  : [{'y': [values_1, values_1b]},] # what?
                     }
                    'direction': 'down',
                    'showactive': True,}
                  ]

# update layout with buttons, and show the figure
fig.update_layout(updatemenus=updatemenus)
fig.show()

参考DataFrame：

核心思路 #

1. 先添加所有股票的柱状图轨迹，默认仅显示第一个股票的数据
2. 遍历股票列动态生成下拉按钮，点击按钮时切换对应轨迹的可见状态并更新图表标题

修正后的完整代码 #

import plotly.graph_objects as go
import pandas as pd

# 替换为你的DataFrame加载逻辑
# df = pd.read_csv("your_data_source.csv")

fig = go.Figure()

stock_columns = df.columns[1:]
# 批量添加柱状图轨迹，默认只显示第一个股票
for idx, col in enumerate(stock_columns):
    fig.add_trace(go.Bar(
        x=df['Quarters'],
        y=df[col],
        name=col,
        visible=(idx == 0)
    ))

# 动态生成下拉菜单按钮
buttons = []
for idx, col in enumerate(stock_columns):
    buttons.append({
        'label': col,
        'method': 'update',
        'args': [
            # 仅让当前选中的股票轨迹可见
            {'visible': [i == idx for i in range(len(stock_columns))]},
            # 更新图表标题，明确当前查看的股票
            {'title': f'季度数据：{col}'}
        ]
    })

# 配置下拉菜单的位置和样式
updatemenus = [{
    'buttons': buttons,
    'direction': 'down',
    'showactive': True,
    'x': 0.1,
    'xanchor': 'left',
    'y': 1.15,
    'yanchor': 'top'
}]

# 完善图表整体布局
fig.update_layout(
    updatemenus=updatemenus,
    title='季度股票数据',
    xaxis_title='季度',
    yaxis_title='数值'
)

fig.show()

关键说明 #

• 动态列适配：通过遍历df.columns[1:]生成按钮和轨迹，无需硬编码列名，自动适配列名变化的场景
• 可见性控制：利用visible参数实现轨迹的显示/隐藏切换，默认仅展示第一个股票
• 交互体验优化：点击按钮时同步更新标题，让用户清晰知晓当前查看的股票

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