python - 如何使用 contourf() 制作动画？

Question

我正在尝试为一些时间相关数据的空间坐标的 wigner 函数设置动画。 wigner 函数是二维的，所以我使用 contourf() 来绘制它。我将数据存储在 HDF5 文件中，可以即时制作 Wigner 分布，但我不知道如何制作动画。我能找到的所有动画教程和示例(例如 this one 和 this one)都严格适用于线图。具体来说，他们的 animate(i) 函数使用 line.set_data()，我似乎找不到 contourf() 的等效函数.

如何为使用 contourf() 制作的图像制作动画？

什么是 contourf() 等同于 set_data()？

Answer 1

FuncAnimation 有一个简单的方法:您必须具有清除轴并根据帧号绘制新轮廓的功能。不要忘记将 blit 设置为 False。

import numpy as np
import matplotlib.pyplot as plt
import matplotlib.animation as animation

DATA = np.random.randn(800).reshape(10,10,8)


fig,ax = plt.subplots()

def animate(i):
       ax.clear()
       ax.contourf(DATA[:,:,i])
       ax.set_title('%03d'%(i)) 

interval = 2#in seconds     
ani = animation.FuncAnimation(fig,animate,5,interval=interval*1e+3,blit=False)

plt.show()