如何使用 matplotlib.animation 在 python 中为数据图制作动画

Question

我有两个数组 x 和 y，每个都有超过 365000 个元素。我想使用这些数组元素绘制一条动画线。我正在使用 matplotlib.animation 。问题是当我执行下面的代码时，我看不到平滑（动画）绘制的图形。相反，我看到它是最终绘制的版本。

这是我的代码：

#libs
# Movement instance creation-----------------------------
movement1=Movement(train1, track1)
# # Move the train on the track

movement1.move()
y = movement1.speed
x = movement1.pos

Writer = animation.writers['ffmpeg']
writer = Writer(fps=20, metadata=dict(artist='Me'), bitrate=1800)

fig = plt.figure()
ax = plt.axes(xlim=(0, 25), ylim=(0, 300))
line, = ax.plot([], [], lw=2)
# initialization function: plot the background of each frame
def init():
    line.set_data([], [])
    return line,

# animation function.  This is called sequentially
def animate(i):
    line.set_data(x, y)
    return line,

anim = animation.FuncAnimation(fig, animate, init_func=init,
                                frames=200, interval=200, blit=True)


anim.save('basic_animation.mp4', writer=writer)

这是我期望的类似结果：

当然，我的图表将是另一条曲线。

Answer 1

您的代码基本没问题；你只需要做三件事。

1. 在anim_func 的每次迭代中将行的xdata 和ydata 设置为不同的值（否则，就没有动画了，会不会？）

2. 设置恒定的轴限制，这样您的绘图就不会改变形状

3. 删除 save 用于显示目的的调用（就我个人而言，我发现它会影响动画）

所以：

ax.axis((x.min(), x.max(), y.min(), y.max())

def animate(i):
    line.set_data(x[:i], y[:i])
    return line,

Answer 2

您需要定义一组数据，这些数据会发生变化以使动画发生。在您提供的示例站点中，作者通过使用 overdose.iloc[:int(i+1] 对数据进行切片来实现这一点（有关使用的实际代码，请参见下文）。这是在 matplotlib 绘制任何数据时创建动画的部分在 animate 函数中。在您的代码中，您输入 line.set_data(x, y) 我想这是您的整个数据集。这就是它没有移动的原因。

def animate(i):
    data = overdose.iloc[:int(i+1)] #select data range
    p = sns.lineplot(x=data.index, y=data[title], data=data, color="r")
    p.tick_params(labelsize=17)
    plt.setp(p.lines,linewidth=7)

要注意的第二件事是，您的情节在顶部被切断了。这可能是因为您的初始化已经错误地设置了轴。我要做的是添加 plt.axis([0, 25, 0, 'upper limit']) 以帮助正确设置轴。