如何为Plotly Express散点图添加按钮以切换x轴对应的数据集列值

问题原因 #

• 每次调用px.scatter指定color="Continent"时，会生成与大洲数量相等的多条trace（本数据集共5个大洲，单x字段对应5条trace），循环2个x字段后图内总共有10条trace
• 原代码中按钮的visible数组长度仅为2，与总trace数不匹配，导致切换逻辑失效；初始未正确隐藏非默认x字段的trace，所以出现多组数据叠加的问题

可行解决代码 #

import pandas as pd
import plotly.express as px
import plotly.graph_objects as go

df = pd.read_csv("https://media.githubusercontent.com/media/ajgallard/happiness_report/main/data/2018_eng.csv")
# 定义要切换的x轴字段列表
x_cols = ["GDP per capita", "Social support"]
# 统计单x字段对应的trace数量（按大洲分组的数量）
trace_per_x = len(df["Continent"].unique())

fig = go.Figure()
# 循环添加所有x字段对应的trace
for col in x_cols:
    figpx = px.scatter(df,
                       x=col,
                       y="Score",
                       size="Population",
                       color="Continent",
                       hover_name="Country/Region",
                       size_max=60,
                       color_discrete_sequence=px.colors.qualitative.G10)
    # 先把所有trace设置为不可见
    fig.add_traces(figpx.update_traces(visible=False).data)

# 生成按钮配置
buttons = []
for idx, col in enumerate(x_cols):
    # 构造visible数组：当前x字段对应的trace设为True，其余为False
    visible = [False]*len(fig.data)
    visible[idx*trace_per_x : (idx+1)*trace_per_x] = [True]*trace_per_x
    buttons.append({
        "label": col,
        "method": "update",
        "args": [
            {"visible": visible},
            {"xaxis": {"title": {"text": col}}} # 切换时同步更新x轴标题
        ]
    })

# 更新布局添加按钮，设置初始显示第一个x字段的trace
fig.update_layout(
    updatemenus=[{
        "buttons": buttons,
        "direction": "down",
        "showactive": True,
        "x": 0.1,
        "y": 1.15
    }],
    xaxis_title = x_cols[0]
)
# 初始显示第一个x字段的trace
fig.update_traces(visible=True, selector=slice(0, trace_per_x))

fig.show()

实现说明 #

• 先统计单x字段对应的trace数量，构造长度和总trace数匹配的visible数组，保证切换时只有对应x字段的trace可见
• 切换时同步更新x轴标题，符合原有图表的格式逻辑
• 保留了原图表的颜色、大小、悬浮提示等所有原有样式，仅新增x轴切换功能

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