具有时间相关参数的 Matplotlib 动画

Question

我是使用 python 的新手，我想问你一个我当前代码的问题。我正在求解一个偏微分方程（空间中的一维），我想每次都为给定的数值解制作动画，但我不想每次都保存解的所有数组（因为这不是高效的）

为简单起见，我只向您展示微分方程的解析解。

我试图为每次尝试制作动画都制作一个陡峭的情节，但正如我在其他地方读到的那样，由于 plt.pause()

import numpy as np

import math

from matplotlib import pyplot as plt

pi = math.pi

xmin = 0.0

xmax = 10.0

N = 100 #number of points

x = np.arange(xmin, xmax , (xmax-xmin)/N)

def solution(t):

    p = np.exp(-x**2/(4*k*t))/(np.sqrt(4.*pi*k*t))

    return p

t_final = 10.0

t_initial = 0.0

t = t_initial

dt = 0.1

k = 1.0

while t<t_final:

    t +=dt

    pp = solution(t)

    plt.ion()

    plt.xlabel('x')

    plt.ylabel('P')

    plt.plot(x, pp, 'r-',label = "t=%s" % t)

    plt.legend(loc='upper right')

    plt.draw()

    plt.pause(10**(-20))

    plt.show()

    plt.clf()

您知道如何在不保存数据的情况下重新实现我的代码以制作动画（并保存）吗？

Answer 1

这种方式使用了我写的一个库，celluloid。有了它，我只需要从您的代码中更改几行：主要是使用赛璐珞的少数调用并稍微更改图例创建。import numpy as npimport mathfrom matplotlib import pyplot as pltfrom celluloid import Camerapi = math.pixmin = 0.0xmax = 10.0N = 100 #number of pointsx = np.arange(xmin, xmax , (xmax-xmin)/N)def solution(t):&nbsp; &nbsp; p = np.exp(-x**2/(4*k*t))/(np.sqrt(4.*pi*k*t))&nbsp; &nbsp; return pt_final = 10.0t_initial = 0.0t = t_initialdt = 0.1k = 1.0fig = plt.figure()camera = Camera(fig)plt.xlabel('x')plt.ylabel('P')while t<t_final:&nbsp; &nbsp; t +=dt&nbsp; &nbsp; pp = solution(t)&nbsp; &nbsp; line = plt.plot(x, pp, 'r-')&nbsp; &nbsp; plt.legend(line, ['t={:.1f}'.format(t)], loc='upper right')&nbsp; &nbsp; camera.snap()animation = camera.animate()animation.save('animation.mp4')