如何在FuncAnimation中传递参数实现动画？（时变地理绘图场景）

嘿，我来帮你梳理下怎么在FuncAnimation里正确传递参数并搞定你的时变地理绘图动画～

首先，你已经用fargs=(ax,)把ax传递给animate函数了，这一步是完全正确的！不过要让动画正常跑起来，还需要调整几个细节：

1. 修复动画帧重叠问题 #

每次绘制新帧前，必须清空轴上的旧内容，不然所有时间步的图层会叠在一起。在animate函数开头加上ax.clear()：

def animate(tindex, ax):
    ax.clear()  # 清空上一帧的绘图内容
    summasel = summaout.isel(time = tindex)['pptrate'].drop('time').to_dataframe()
    summasel['hru_id2'] = summasel.index
    polydat2 = pd.merge(polydat, summasel)
    # 捕获绘图返回的艺术家对象，用于blit更新
    plot_obj = polydat2.plot(ax = ax, column = 'pptrate', cmap='viridis_r', scheme='quantiles')
    # 可以添加时间步标题，让动画更直观
    ax.set_title(f"Time Step: {tindex + 1}")
    return (plot_obj,)

2. 确认参数传递的正确姿势 #

你用fargs=(ax,)的方式是标准用法：FuncAnimation会把frames里的每个值作为第一个参数传给animate，再把fargs里的参数依次往后传递。比如frames=40，就会依次把0到39（默认从0开始的整数）作为tindex传入，再加上ax参数。

如果想避免依赖全局变量（比如summaout、polydat），可以把它们也加入参数列表：

def animate(tindex, ax, summaout, polydat):
    ax.clear()
    summasel = summaout.isel(time = tindex)['pptrate'].drop('time').to_dataframe()
    summasel['hru_id2'] = summasel.index
    polydat2 = pd.merge(polydat, summasel)
    plot_obj = polydat2.plot(ax = ax, column = 'pptrate', cmap='viridis_r', scheme='quantiles')
    ax.set_title(f"Time Step: {tindex + 1}")
    return (plot_obj,)

# 创建动画时把额外参数加到fargs里
a = ani.FuncAnimation(fig, animate, frames=40, fargs=(ax, summaout, polydat), blit=True)

3. 适配Blitting的细节 #

当blit=True时，animate函数必须返回所有需要更新的艺术家对象。如果你的绘图包含颜色条这类额外元素，也要把它们加入返回元组：

def animate(tindex, ax, summaout, polydat):
    ax.clear()
    summasel = summaout.isel(time = tindex)['pptrate'].drop('time').to_dataframe()
    summasel['hru_id2'] = summasel.index
    polydat2 = pd.merge(polydat, summasel)
    # 开启图例/颜色条
    plot_obj = polydat2.plot(ax = ax, column = 'pptrate', cmap='viridis_r', scheme='quantiles', legend=True)
    # 获取颜色条对应的轴对象
    cbar = ax.get_figure().get_axes()[1]
    ax.set_title(f"Time Step: {tindex + 1}")
    return (plot_obj, cbar,)

4. 动画输出测试 #

最后用HTML(a.to_html5_video())渲染动画是没问题的，要是在Jupyter环境里，记得先运行%matplotlib notebook或者%matplotlib inline（后者可能需要额外设置才能正常显示动画）。

这样调整后，你的时变地理绘图动画应该就能正常播放啦～

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