如何用Plotly Express/图形对象制作带状态配色的地图线路动画？

问题背景 #

我正在做个人项目，制作动画地图展示北美一组线性基础设施项目的发展历程。这些项目有所有者/开发者，会经历开发、建设、运营阶段，也可能失败终止，在地图上以LineString形式呈现。

用geopandas dataframe存储数据，循环生成各年份带项目状态的数据集，整理经纬度后用Plotly绘图。用px.line_mapbox做动画时，不设置color参数能正常显示线路，但开启color=statuses按状态配色后，同状态的线路会被错误连接。尝试切换到Plotly Graphics Objects（plotly.go）实现，但设置动画帧时遇到困难，现有示例多针对点而非线路，官方文档也没覆盖相关内容。

一、修复Plotly Express的线路错误连接问题 #

• 问题根源：px.line_mapbox会自动按color分组，把同组内的所有坐标点连在一起，破坏了每条LineString的独立结构。
• 解决步骤：
  1. 给每条线路添加唯一标识符（比如项目ID）；
  2. 同时设置color（按状态配色）和line_group（按项目ID分组）参数；
  3. 用color_discrete_map手动绑定状态对应的颜色。
• 代码示例：

  # 假设GeoDataFrame包含project_id（唯一ID）、status（状态）列，且已拆分出lat/lon列
  fig = px.line_mapbox(
      df,
      lat="lat",
      lon="lon",
      animation_frame="year",
      color="status",
      line_group="project_id",  # 核心：强制按项目ID保留线路独立性
      color_discrete_map={
          "开发": "#FFA500",
          "建设": "#00FF00",
          "运营": "#0000FF",
          "终止": "#FF0000"
      },
      mapbox_style="carto-positron",
      zoom=3
  )
  fig.show()
  

• 原理：line_group参数会让Plotly优先按项目ID分组渲染线路，color仅负责颜色分配，不会改变线路的连接逻辑。

二、Plotly Graphics Objects（Go）实现线路动画 #

如果必须用Go实现，核心思路是为每个动画帧单独生成所有线路的Scattermapbox trace，再通过frames参数管理各帧内容：

1. 先整理数据：按年份分组，每个年份下的每个项目都要有对应的经纬度坐标数组（LineString拆分为lat/lon列表）；
2. 代码示例：

import plotly.graph_objects as go

# 假设已整理好按年份分组的数据：year_data = {2020: [{"project_id": "p1", "status": "开发", "lats": [...], "lons": [...]}], ...}
status_colors = {"开发": "#FFA500", "建设": "#00FF00", "运营": "#0000FF", "终止": "#FF0000"}

# 构建初始帧（以最早年份为例）
initial_year = min(year_data.keys())
initial_traces = []
for proj in year_data[initial_year]:
    initial_traces.append(go.Scattermapbox(
        lon=proj["lons"],
        lat=proj["lats"],
        mode="lines",
        line=dict(color=status_colors[proj["status"]], width=3),
        name=f"{proj['project_id']} ({proj['status']})"
    ))

# 构建所有动画帧
frames = []
for year in year_data.keys():
    frame_traces = []
    for proj in year_data[year]:
        frame_traces.append(go.Scattermapbox(
            lon=proj["lons"],
            lat=proj["lats"],
            mode="lines",
            line=dict(color=status_colors[proj["status"]], width=3),
            name=f"{proj['project_id']} ({proj['status']})"
        ))
    frames.append(go.Frame(data=frame_traces, name=str(year)))

# 组装最终图表
fig = go.Figure(
    data=initial_traces,
    frames=frames,
    layout=go.Layout(
        mapbox=dict(style="carto-positron", zoom=3, center=dict(lat=40, lon=-95)),
        updatemenus=[dict(
            type="buttons",
            buttons=[dict(label="播放", method="animate", args=[None, dict(frame=dict(duration=500, redraw=True), fromcurrent=True)])]
        )],
        sliders=[dict(
            steps=[dict(method="animate", args=[[str(year)], dict(mode="immediate")]) for year in year_data.keys()],
            currentvalue=dict(prefix="年份: ")
        )]
    )
)
fig.show()

• 注意事项：每个帧的trace必须完整包含该年份所有项目的线路，不能复用前一帧的内容；确保每个项目的经纬度数组是LineString的连续坐标，不能打乱顺序。

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