使用按钮过滤plotly中的不同数据

import plotly.graph_objects as go
import dash
import dash_core_components as dcc
import dash_html_components as html

# 创建一个Dash应用
app = dash.Dash(__name__)

# 创建数据
data1 = [1, 2, 3, 4, 5]
data2 = [2, 4, 6, 8, 10]
data3 = [3, 6, 9, 12, 15]

# 创建图表
fig = go.Figure()

# 添加初始数据
fig.add_trace(go.Scatter(x=[1, 2, 3, 4, 5], y=data1, name='Data 1'))

# 创建布局
app.layout = html.Div([
    html.H1("使用按钮过滤plotly中的不同数据"),
    dcc.Checklist(
        id='data-selector',
        options=[
            {'label': 'Data 1', 'value': 'data1'},
            {'label': 'Data 2', 'value': 'data2'},
            {'label': 'Data 3', 'value': 'data3'}
        ],
        value=['data1'],  # 默认选择Data 1
        labelStyle={'display': 'inline-block'}
    ),
    dcc.Graph(id='graph')
])

# 回调函数，根据按钮选择更新图表
@app.callback(
    dash.dependencies.Output('graph', 'figure'),
    [dash.dependencies.Input('data-selector', 'value')]
)
def update_graph(selected_data):
    traces = []
    for data in selected_data:
        if data == 'data1':
            traces.append(go.Scatter(x=[1, 2, 3, 4, 5], y=data1, name='Data 1'))
        elif data == 'data2':
            traces.append(go.Scatter(x=[1, 2, 3, 4, 5], y=data2, name='Data 2'))
        elif data == 'data3':
            traces.append(go.Scatter(x=[1, 2, 3, 4, 5], y=data3, name='Data 3'))
    
    return {'data': traces}

# 运行应用
if __name__ == '__main__':
    app.run_server(debug=True)