如何用xarray数据集结合Cartopy PlateCarree投影制作带色条的动画？

核心实现方案 #

结合xarray、cartopy（处理投影）和matplotlib.animation（生成动画），重点解决坐标系统一和色条帧间一致性问题，以下是完整可运行的代码示例：

完整代码示例 #

假设你的Xarray数据集为ds，目标变量为data_var，时间维度为time：

import xarray as xr
import cartopy.crs as ccrs
import matplotlib.pyplot as plt
from matplotlib.animation import FuncAnimation

# 加载数据集（替换为你的数据路径或已加载的ds）
# ds = xr.open_dataset("your_data_file.nc")

# 1. 初始化画布与投影轴
fig, ax = plt.subplots(figsize=(10, 6), subplot_kw={"projection": ccrs.PlateCarree()})
ax.coastlines(linewidth=0.8)
ax.gridlines(draw_labels=True, linestyle="--")

# 2. 计算全局颜色范围（确保所有帧色条一致）
vmin = ds["data_var"].min().values
vmax = ds["data_var"].max().values

# 3. 绘制初始帧（暂不添加色条）
im = ds["data_var"].isel(time=0).plot(
    ax=ax,
    transform=ccrs.PlateCarree(),  # 数据原始坐标系（需与你的数据匹配，经纬度数据用这个）
    vmin=vmin,
    vmax=vmax,
    cmap="viridis",
    add_colorbar=False
)

# 4. 添加固定色条
cbar = fig.colorbar(im, ax=ax, orientation="horizontal", pad=0.08)
cbar.set_label(ds["data_var"].attrs.get("units", "Value"))  # 自动读取变量单位

# 5. 定义帧更新函数
def update(frame_idx):
    # 清除轴内数据（保留海岸线、网格线）
    ax.clear()
    ax.coastlines(linewidth=0.8)
    ax.gridlines(draw_labels=True, linestyle="--")
    
    # 绘制当前帧数据
    ds["data_var"].isel(time=frame_idx).plot(
        ax=ax,
        transform=ccrs.PlateCarree(),
        vmin=vmin,
        vmax=vmax,
        cmap="viridis",
        add_colorbar=False
    )
    
    # 设置帧标题（显示当前时间）
    current_time = ds["time"].isel(time=frame_idx).dt.strftime("%Y-%m-%d %H:%M").values
    ax.set_title(f"Data: {current_time}", fontsize=12)
    return im,

# 6. 生成并展示动画
ani = FuncAnimation(
    fig,
    update,
    frames=len(ds["time"]),
    interval=400,  # 帧间隔（毫秒）
    blit=True
)

# 保存动画（可选）
# ani.save("data_animation.mp4", writer="ffmpeg", dpi=100)
# ani.save("data_animation.gif", writer="pillow", dpi=100)

plt.show()

关键细节说明 #

• 坐标系匹配：transform参数必须设置为数据的原始坐标系（经纬度数据用ccrs.PlateCarree()），轴的projection是目标展示坐标系（这里指定为PlateCarree），两者不可混淆。
• 色条一致性：提前计算全局vmin/vmax，避免每帧重新计算导致色条缩放，保证动画对比性。
• 轴元素重置：ax.clear()会清除所有轴内容，因此每次更新需重新绘制海岸线、网格线。
• 保存依赖：保存MP4需安装ffmpeg，保存GIF需安装pillow，可通过pip install ffmpeg-python pillow完成安装。

常见问题排查 #

• 投影异常：检查数据的经纬度范围是否正确，确认transform参数与数据坐标系匹配。
• 色条不显示：确保每帧绘制时设置add_colorbar=False，仅在初始阶段添加一次色条。
• 动画卡顿：可通过降采样时间维度（如ds = ds.isel(time=slice(0, None, 2))）或降低dpi优化。

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