Plotly动画散点图：如何让图例显示所有颜色类别？

问题核心是Plotly动画默认只展示当前帧中存在的类别，若某一日期帧里只有一种"Highest Vol"类别，图例就只会显示该类别。以下是两种可行的解决办法：

方法一：预处理数据，让每个日期帧包含所有类别 #

通过扩展数据集，确保每个Date对应的所有"Highest Vol"类别都有记录（无数据的行用NaN填充，不会显示散点但会保留图例条目）。

示例代码：

import pandas as pd
import plotly.express as px

# 假设你的原始数据集是df
all_vol_categories = df['Highest Vol'].unique()
# 生成所有Date和Highest Vol的组合
full_index = pd.MultiIndex.from_product(
    [df['Date'].unique(), all_vol_categories],
    names=['Date', 'Highest Vol']
)
# 扩展数据集，缺失的数据用NaN填充
expanded_df = full_index.to_frame(index=False).merge(df, on=['Date', 'Highest Vol'], how='left')

# 绘制动画散点图
fig = px.scatter(
    expanded_df,
    x='你的X轴字段名',
    y='你的Y轴字段名',
    animation_frame='Date',
    color='Highest Vol',
    hover_data=['需要显示的额外字段']  # 可选
)
fig.show()

方法二：手动修改Trace属性，强制显示全类别 #

如果不想修改原始数据，可以在绘图后遍历所有Trace，设置每个类别对应的Trace都显示图例。

示例代码：

import plotly.express as px

# 用原始数据绘制动画图
fig = px.scatter(
    df,
    x='你的X轴字段名',
    y='你的Y轴字段名',
    animation_frame='Date',
    color='Highest Vol'
)

# 获取所有Highest Vol的唯一类别
all_cats = df['Highest Vol'].unique()

# 遍历每个类别，设置对应Trace的showlegend为True
for category in all_cats:
    for trace in fig.data:
        if trace.name == category:
            trace.showlegend = True

# 调整图例布局，避免重复和混乱
fig.update_layout(
    legend={
        'tracegroupgap': 0,
        'itemsizing': 'constant'
    }
)

fig.show()

两种方法都能让图例显示所有"Highest Vol"对应的颜色类别，方法一的交互逻辑更贴合Plotly原生设计，方法二适合快速调整现有图表。

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