python - 如何使用 matplotlib 为 Pandas 数据的散点图制作动画

Question

我目前正在尝试为保存在 pandas 数据框中的每月数据的散点图制作动画。到目前为止，我制作了一个循环，一个接一个地生成一个图。现在我想将它们加入一个 gif(或我不在乎的 mp4)中。有没有简单的方法来使用 mathplotlibs 动画功能？我无法理解如何通过 FuncAnimation 循环切片数据。到目前为止，我是这样做的:

time = df.monat.unique()
for i in time:
    dft = df[(df.monat == i) & (df.xcol < 4000)]
    plt.scatter(x=dft['xcol'],
            y=dft['ycol'],
            s=dft['scol'] / 25,
            c=dft['clr'],
            linewidth=0,
            alpha=0.8)
    plt.title('Title ' + str(i), fontsize=10)
    plt.xlabel('x label', fontsize=9)
    plt.ylabel('y label', fontsize=9)
    legend1_line2d = list()
    for val in clrdict.values():
        legend1_line2d.append(mlines.Line2D([0], [0],
            linestyle='none',
            marker='o',
            alpha=0.6,
            markersize=6,
            markeredgecolor=None,
            markeredgewidth=0,
            markerfacecolor=val))
    legend1 = plt.legend(legend1_line2d,
             names,
             frameon=False,
             numpoints=1,
             fontsize=8,
             loc='upper right') 
    plt.show()

Answer 1

我自己想出来的:

生成一个空图(图)。像以前一样，所有唯一的时间值都存储在一个系列(时间)中。一个简单的计数器 (i) 有助于在更新函数中为每个月生成正确的数据片段 (dft)(df.monat == 来自系列“时间”的值)。调用更新函数乘以 anim.FuncAnimation (frames=len(time)) 中帧参数的值。

希望，这会对其他人有所帮助(我发现对 matplotlib FuncAnimation 的大多数解释都使用随机数 - 而不是特定的 pandas 列):

import pandas as pd
import matplotlib.pyplot as plt
import matplotlib.lines as mlines
import matplotlib.animation as anim

...

time = df.monat.unique()    
fig = plt.figure()
i = 1

def update(i):
        plt.clf()
        dft = df[(df.monat == time[i]) & (df.xcol < 4000)]
        plt.scatter(x=dft['xcol'],
                    y=dft['ycol'],
                    s=dft['scol'] / 25,
                    c=dft['clr'],
                    linewidth=0,
                    alpha=0.8)
        plt.title('Title ' + str(time[i]), fontsize=10)
        plt.xlabel('x label', fontsize=9)
        plt.ylabel('y label', fontsize=9)
        plt.xlim(0, 900)  # fixed dimensions x
        plt.ylim(-5, 100) # fixed dimensions y 
        legend1_line2d = list()
        for val in clrdict.values():
            legend1_line2d.append(mlines.Line2D([0], [0],
                    linestyle='none',
                    marker='o',
                    alpha=0.6,
                    markersize=6,
                    markeredgecolor=None,
                    markeredgewidth=0,
                    markerfacecolor=val))
        legend1 = plt.legend(legend1_line2d,
                    names,
                    frameon=False,
                    numpoints=1,
                    fontsize=8,
                    loc='upper right')
        i += 1

ani = anim.FuncAnimation(fig, update, frames=len(time), interval=500)
# plt.show() # this will show the ani over and over
ani.save("test.mp4", dpi=200, fps=1, codec="libx264", bitrate=5000, extra_args=['-pix_fmt', 'yuv420p'])