如何在Plotly动画散点图中使用多个DataFrame展示多组带时间维度的XY坐标并区分标记

我明白你的痛点——把多组数据硬塞到同一个DataFrame里确实会丢失分组信息，没法灵活设置标记样式。其实有两种更优雅的方法来实现你的需求，分别用Plotly Express（更简单）和Plotly Graph Objects（更灵活）来做，我给你详细拆解一下：

方法一：用Plotly Express（推荐，代码更简洁） #

核心思路是把三组数据整理成长格式（long-format）的DataFrame，给每组数据加一个group标识列，让Plotly自动识别分组并帮你区分样式。

import pandas as pd
import plotly.express as px

# 假设x1_trim、y1_trim等都是与time_trim长度一致的列表
# 为每组数据单独构造DataFrame并添加分组标识
df1 = pd.DataFrame({"x": x1_trim, "y": y1_trim, "time": time_trim, "group": "Group 1"})
df2 = pd.DataFrame({"x": x2_trim, "y": y2_trim, "time": time_trim, "group": "Group 2"})
df3 = pd.DataFrame({"x": x3_trim, "y": y3_trim, "time": time_trim, "group": "Group 3"})

# 合并成一个长格式DataFrame
df = pd.concat([df1, df2, df3], ignore_index=True)

# 设置坐标轴范围
x_range = [-1, 1]
y_range = [-1, 1]

# 绘制动画散点图，通过分组区分样式
fig = px.scatter(
    df,
    x="x",
    y="y",
    animation_frame="time",
    color="group",  # 按分组自动设置颜色
    symbol="group",  # 按分组自动设置标记符号
    size_max=10,
    range_x=x_range,
    range_y=y_range,
    title="多组数据的时间动画散点图"
)

# 添加绿色矩形
fig.add_shape(
    type="rect",
    x0=-0.5, y0=-0.5, x1=0.5, y1=0.5,
    line=dict(color="Green", width=2)
)

# 自定义每组的样式（可选，比如手动指定颜色和符号）
fig.update_traces(selector={"name": "Group 1"}, marker=dict(color="crimson", symbol="circle"))
fig.update_traces(selector={"name": "Group 2"}, marker=dict(color="royalblue", symbol="square"))
fig.update_traces(selector={"name": "Group 3"}, marker=dict(color="orange", symbol="diamond"))

fig.show()

这种方法的优势是Plotly会自动处理动画逻辑，每个时间帧会自动筛选对应分组的数据，你只需要通过color、symbol等参数就能轻松区分各组，后期修改样式也非常灵活。

方法二：用Plotly Graph Objects（更灵活，适合复杂自定义） #

你之前尝试用Graph Objects失败，是因为错误地在add_trace里使用了animation_frame参数——这个参数是Plotly Express专属的，Graph Objects需要手动构建frames数组来定义每个时间帧的内容。

import plotly.graph_objects as go

# 初始化基础图的trace（使用第一帧的数据）
fig = go.Figure()
fig.add_trace(go.Scatter(
    x=[x1_trim[0]], y=[y1_trim[0]],
    mode="markers", name="Group 1",
    marker=dict(color="crimson", size=10)
))
fig.add_trace(go.Scatter(
    x=[x2_trim[0]], y=[y2_trim[0]],
    mode="markers", name="Group 2",
    marker=dict(color="royalblue", size=10)
))
fig.add_trace(go.Scatter(
    x=[x3_trim[0]], y=[y3_trim[0]],
    mode="markers", name="Group 3",
    marker=dict(color="orange", size=10)
))

# 为每个时间点创建对应的frame
frames = []
for i, t in enumerate(time_trim):
    frame = go.Frame(
        data=[
            go.Scatter(x=[x1_trim[i]], y=[y1_trim[i]]),
            go.Scatter(x=[x2_trim[i]], y=[y2_trim[i]]),
            go.Scatter(x=[x3_trim[i]], y=[y3_trim[i]])
        ],
        name=str(t)
    )
    frames.append(frame)

# 将frames添加到图中
fig.frames = frames

# 设置动画控制组件和坐标轴范围
fig.update_layout(
    xaxis=dict(range=[-1,1], title="X坐标"),
    yaxis=dict(range=[-1,1], title="Y坐标"),
    title="多组数据的时间动画散点图",
    updatemenus=[{
        "buttons": [
            {"label": "播放", "method": "animate", "args": [None, {"frame": {"duration": 500, "redraw": True}, "fromcurrent": True}]},
            {"label": "暂停", "method": "animate", "args": [[None], {"frame": {"duration": 0, "redraw": False}, "mode": "immediate"}]}
        ],
        "direction": "left",
        "pad": {"r": 10, "t": 87},
        "showactive": False,
        "type": "buttons",
        "x": 0.1,
        "xanchor": "right",
        "y": 0,
        "yanchor": "top"
    }]
)

# 添加绿色矩形
fig.add_shape(
    type="rect",
    x0=-0.5, y0=-0.5, x1=0.5, y1=0.5,
    line=dict(color="Green", width=2)
)

fig.show()

这种方法适合需要高度自定义动画逻辑的场景，比如不同时间帧有不同的样式变化，或者需要精细控制动画的播放速度、切换逻辑等。

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