如何用Matplotlib制作Xarray数据动画？帧堆叠及无输出问题求助

你的动画出现无输出、帧堆叠的问题，主要源于colorbar重复创建、动画更新逻辑不规范和月份索引显示错误这几个关键点，以下是针对性的修正方案：

核心问题拆解 #

• 帧堆叠：每次调用pr_image.plot()都会生成新的colorbar，ax.clear()仅清除轴内绘图元素，无法移除已创建的colorbar，导致多帧后colorbar堆叠。
• 动画无输出：update函数返回ax而非具体的艺术家对象，Matplotlib无法跟踪需要更新的元素；部分后端对动画渲染的支持需要明确配置。
• 月份显示错误：frame是0-based索引，但实际月份是1-12，标题显示Month: 0不符合业务逻辑。

修正后的代码 #

import rioxarray
import xarray
import matplotlib.pyplot as plt
from matplotlib.animation import FuncAnimation

# 读取并预处理数据
fn = 'C:/.../pr_2020.nc'
pr = rioxarray.open_rasterio(fn)
pr_masked = pr.where(pr != pr.attrs['_FillValue'])
pr_monthly = pr_masked.groupby('day.month').mean()

# 计算全局降水率范围，固定colorbar标尺（增强动画连贯性）
vmin = pr_monthly.min().item()
vmax = pr_monthly.max().item()

fig, ax = plt.subplots()
im = None
cbar = None

def update(frame):
    global im, cbar
    # 清除轴内所有元素
    ax.clear()
    # 移除上一帧的colorbar，避免堆叠
    if cbar is not None:
        cbar.remove()
    
    # 获取当前帧对应的实际月份（1-12）
    current_month = pr_monthly['month'].values[frame]
    pr_image = pr_monthly.sel(month=current_month)
    
    # 绘制图像，手动控制colorbar生成
    im = pr_image.plot(ax=ax, vmin=vmin, vmax=vmax, add_colorbar=False)
    cbar = fig.colorbar(im, ax=ax)
    # 设置正确的月份标题
    ax.set_title(f"Month: {current_month}")
    
    # 返回需要更新的艺术家对象，配合blit=True提升渲染效率
    return [im, cbar, ax.title]

# 创建动画，frames直接遍历月份索引
ani = FuncAnimation(fig, update, frames=range(len(pr_monthly)), interval=500, blit=True)

# 显示动画
plt.show()

# 保存动画（需提前安装ffmpeg：conda install ffmpeg 或 pip install ffmpeg-python）
# ani.save('us_precipitation_monthly.mp4', writer='ffmpeg', fps=2)

关键修改说明 #

1. 固定颜色标尺：用全局的vmin和vmax统一所有帧的colorbar范围，避免每帧颜色标尺波动，便于对比降水率变化。
2. 手动管理colorbar：每次更新时移除旧colorbar，手动创建新的colorbar，彻底解决堆叠问题。
3. 正确的月份映射：通过pr_monthly['month'].values[frame]获取1-12的实际月份，修正标题显示错误。
4. 规范动画更新逻辑：返回具体的艺术家对象（图像、colorbar、标题）并启用blit=True，让Matplotlib仅更新变化的部分，确保动画正常渲染且更流畅。
5. 移除冗余初始绘制：FuncAnimation会自动调用update(0)生成第一帧，无需提前手动绘制，避免重复渲染。

额外注意事项 #

• 如果动画仍无法显示，可在代码开头添加plt.switch_backend('TkAgg')或plt.switch_backend('QtAgg')切换后端（需对应安装Tkinter或PyQt）。
• 保存动画前确保已安装ffmpeg，否则会保存失败。

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