为动画重绘 Seaborn 图形

Question

一些seaborn 方法，例如JointPlot create new figures on each call。这使得无法创建像 matplotlib 这样的简单动画，其中对 plt.cla() 或 plt.clf() 的迭代调用允许更新图形的内容，而无需每次都关闭/打开窗口。

我目前看到的唯一解决方案是：

for t in range(iterations):
    # .. update your data ..

    if 'jp' in locals():
        plt.close(jp.fig)

    jp = sns.jointplot(x=data[0], y=data[1])
    plt.pause(0.01)

这是因为我们在创建新窗口之前关闭了前一个窗口。但当然，这远非理想。

有没有更好的方法？可以以某种方式直接在先前生成的Figure 对象上完成绘图吗？或者有没有办法阻止这些方法在每次调用时生成新的数字？

Answer 1

不幸的是，sns.jointplot 自己创建了一个图形。为了使联合图动画化，因此可以重用这个创建的图形，而不是在每次交互中重新创建一个新图形。

jointplot 在内部创建一个JointGrid，因此直接使用它并分别绘制关节轴和边缘是有意义的。然后在动画的每一步中更新数据，清除轴并设置它们，就像在创建网格时一样。不幸的是，这最后一步涉及很多代码行。

最终的代码可能如下所示：

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

def get_data(i=0):
    x,y = np.random.normal(loc=i,scale=3,size=(2, 260))
    return x,y

x,y = get_data()
g = sns.JointGrid(x=x, y=y, size=4)
lim = (-10,10)

def prep_axes(g, xlim, ylim):
    g.ax_joint.clear()
    g.ax_joint.set_xlim(xlim)
    g.ax_joint.set_ylim(ylim)
    g.ax_marg_x.clear()
    g.ax_marg_x.set_xlim(xlim)
    g.ax_marg_y.clear()
    g.ax_marg_y.set_ylim(ylim)
    plt.setp(g.ax_marg_x.get_xticklabels(), visible=False)
    plt.setp(g.ax_marg_y.get_yticklabels(), visible=False)
    plt.setp(g.ax_marg_x.yaxis.get_majorticklines(), visible=False)
    plt.setp(g.ax_marg_x.yaxis.get_minorticklines(), visible=False)
    plt.setp(g.ax_marg_y.xaxis.get_majorticklines(), visible=False)
    plt.setp(g.ax_marg_y.xaxis.get_minorticklines(), visible=False)
    plt.setp(g.ax_marg_x.get_yticklabels(), visible=False)
    plt.setp(g.ax_marg_y.get_xticklabels(), visible=False)


def animate(i):
    g.x, g.y = get_data(i)
    prep_axes(g, lim, lim)
    g.plot_joint(sns.kdeplot, cmap="Purples_d")
    g.plot_marginals(sns.kdeplot, color="m", shade=True)

frames=np.sin(np.linspace(0,2*np.pi,17))*5
ani = matplotlib.animation.FuncAnimation(g.fig, animate, frames=frames, repeat=True)

plt.show()

Answer 2

使用 celluloid 包 (https://github.com/jwkvam/celluloid) 我能够轻松地为 seaborn 地块制作动画：

import numpy as np
from celluloid import Camera
import pandas as pd
import seaborn as sns
import matplotlib.pyplot as plt

fig = plt.figure()
camera = Camera(fig)

# animation draws one data point at a time
for i in range(0, data.shape[0]):
    plot = sns.scatterplot(x=data.x[:i], y=data.y[:i])
    camera.snap()

anim = camera.animate(blit=False)
anim.save('animation.mp4')

我确信可以为关节图编写类似的代码