从熊猫数据框列表生成动画

Question

我有一个 pandas 数据帧列表，其中的数据帧代表进程中的后续帧。因此，数据帧的大小和结构都相同（相同的列和索引）。我想知道是否有办法为这些数据帧设置动画并将动画保存为 mpeg 电影。

我尝试了以下操作，但是没有任何运气：

#X: a list of 1000 dataframe, all of same size 
fig = plt.figure()
ax = fig.add_subplot(111)

im = ax.imshow(X,interpolation='nearest')


def update_img(n):

    im.set_data(n.values)
    return im

animation = animation.FuncAnimation(fig,update_img, frames = X, interval=30)

上面卡在第一帧。

Answer 1

要将动画保存为mp4，可以使用

import matplotlib.pyplot as plt
import matplotlib.animation
import numpy as np; np.random.seed(1)
import pandas as pd

X = [pd.DataFrame(np.random.rand(10,10)) for i in range(100)] 
fig = plt.figure()
ax = fig.add_subplot(111)

im = ax.imshow(X[0],interpolation='nearest')

def update_img(n):
    im.set_data(n.values)

ani = matplotlib.animation.FuncAnimation(fig,update_img, frames = X, interval=30)

FFwriter = matplotlib.animation.FFMpegWriter(fps=30)     
ani.save(__file__+".mp4",writer = FFwriter)

plt.show()