我试图在jupyter中通过seaborn热力图使HTML(anim.to_html5_video)
动画生效。
- 首先,我从文档中获取了可行的示例,并制作了“纯matplotlib” image map animated example,它能够工作,但存在小问题(动画单元格中的“寄生输出”)
- 然后,我试图让它与seaborn.heatmap一起工作...但失败了。动画看起来像是“无限镜子”——显然matplotlib轴/绘图组合有问题,但我无法解决。
常见的初始化单元格:
import pandas as pd
import seaborn as sns
import numpy as np
%matplotlib inline
#%matplotlib notebook # Tried both, not needed for animation.
import matplotlib.pyplot as plt
from matplotlib import animation, rc
from IPython.display import HTML
动画可以工作,但是存在“不想要的静态输出图像”:
fig, ax = plt.subplots()
nx = 50
ny = 50
line2d, = ax.plot([], [], lw=2)
def init():
line2d.set_data([], [])
ax.imshow(np.zeros((nx, ny)))
return (line2d,)
def animate(i):
data = np.random.rand(nx, ny)
ax.set_title('i: ' + str(i))
ax.imshow(data)
return (line2d,)
anim = animation.FuncAnimation(fig, animate, init_func=init, frames=10, interval=1000, blit=False)
HTML(anim.to_html5_video())
所以,我的Jupyter设置(软件包,ffmpeg等)看起来都没问题。
但是,我不知道如何使用seaborn.heatmap:
fig, ax = plt.subplots()
nx = 50
ny = 50
line2d, = ax.plot([], [], lw=2)
ax_global = ax
def init_heatmap():
line2d.set_data([], [])
sns.heatmap(np.zeros((nx, ny)), ax=ax_global)
return (line2d,)
def animate_heatmap(i):
data = np.random.rand(nx, ny)
sns.heatmap(data, ax=ax_global)
ax.set_title('Frame: ' + str(i))
return (line2d,)
anim = animation.FuncAnimation(fig, animate_heatmap, init_func=init_heatmap,
frames=10, interval=1000, blit=True)
HTML(anim.to_html5_video())
两个样本都已准备好在Github上测试
当然,我希望看到随机地图和“稳定的热轴”动画,但是得到了这个https://vimeo.com/298786185/