我试图通读matplotlib.animation.funcanimation
的文档,以使热图动画化。
当我运行下面的代码时,我没有得到任何错误,并且确实出现了一个热图曲线,但它似乎没有动画。
import numpy as np
import matplotlib.pyplot as plt
import seaborn as sns
import matplotlib
# generate random noise for the heatmap
rnd_data = np.random.normal(0, 1, (500, 100, 100))
fig, ax = plt.subplots(figsize=(12,10))
def my_func(i):
sns.heatmap(rnd_data[i])
anim = matplotlib.animation.FuncAnimation(fig=fig, func=my_func, frames=200, interval=500, blit=False)
plt.show()
此代码的结果似乎是rnd_data
的单帧,即第一个数组rnd_data[0]
。 我尝试将帧数
和间隔数
更改为更大的数字,因为我认为它的动画速度太快,我看不见,但这似乎没有起作用。
我是不是做错了什么? 我想,当我为这样的数据集绘制热图时,我应该能够看到图的像素像白噪音一样变化和移动,但它没有工作。 如何使热图动画化?
为了正确运行动画,您必须使用:
sns.heatmap(rnd_data[..., i])
以便指定热图沿第三轴随时间变化。
完整代码如下,为了正确添加颜色条,我更改了一些内容:
import numpy as np
import matplotlib.pyplot as plt
import seaborn as sns
from matplotlib.animation import FuncAnimation
# generate random noise for the heatmap
rnd_data = np.random.normal(0, 1, (500, 100, 100))
def my_func(i):
ax.cla()
sns.heatmap(rnd_data[..., i],
ax = ax,
cbar = True,
cbar_ax = cbar_ax,
vmin = rnd_data.min(),
vmax = rnd_data.max())
grid_kws = {'width_ratios': (0.9, 0.05), 'wspace': 0.2}
fig, (ax, cbar_ax) = plt.subplots(1, 2, gridspec_kw = grid_kws, figsize = (12, 8))
anim = FuncAnimation(fig = fig, func = my_func, frames = 200, interval = 50, blit = False)
plt.show()
这给了我一个动画: