Python Plotly:使用按钮更新子图集并任意分配轨迹

import plotly.graph_objects as go
from plotly.subplots import make_subplots
import random

# 创建子图布局
fig = make_subplots(rows=2, cols=2)

# 创建按钮
buttons = []
for i in range(4):
    buttons.append(
        dict(
            label=f"子图 {i+1}",
            method="update",
            args=[{"visible": [False] * 4}, {"title": f"子图 {i+1}"}],
            args2=[{"visible": [False] * 4, f"visible[{i}]": True}]
        )
    )

# 添加按钮到布局
fig.update_layout(
    updatemenus=[
        dict(
            type="buttons",
            buttons=buttons,
            direction="down",
            showactive=True,
            x=1.1,
            y=1.2
        )
    ]
)

# 添加随机轨迹到子图
for i in range(4):
    trace = go.Scatter(
        x=[random.randint(1, 10) for _ in range(10)],
        y=[random.randint(1, 10) for _ in range(10)],
        mode="markers",
        name=f"轨迹 {i+1}",
        visible=False if i != 0 else True
    )
    fig.add_trace(trace, row=(i // 2) + 1, col=(i % 2) + 1)

# 更新布局
fig.update_layout(showlegend=True)

# 显示图表
fig.show()