Анимация Matplotlib с зависящим от времени параметром

Question

Я новичок в использовании python, и я хотел бы задать вам проблему с моим текущим кодом.Я решаю уравнение в частных производных (1D в пространстве) и хочу каждый раз создавать анимацию для данного численного решения, но я не хочу сохранять все массивы решения одновременно (потому что это неэффективный)

Для простоты я просто покажу вам аналитическое решение дифференциального уравнения.

Я пытался составить сюжет для каждого крутого поворота, пытаясь создать анимацию, но, как я читал,в других местах это не слишком эффективно из-за функции 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 np
import math
from matplotlib import pyplot as plt
from celluloid import Camera

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
fig = plt.figure()
camera = Camera(fig)
plt.xlabel('x')
plt.ylabel('P')
while t<t_final:
    t +=dt
    pp = solution(t)

    line = plt.plot(x, pp, 'r-')
    plt.legend(line, ['t={:.1f}'.format(t)], loc='upper right')
    camera.snap()
animation = camera.animate()
animation.save('animation.mp4')

Answer 2

Вот как использовать FuncAnimation для создания желаемой анимации

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)

t_initial = 0.0
t_final = 10.0
dt = 0.1
k = 1.0

fig, ax = plt.subplots()
ax.set_xlabel('x')
ax.set_ylabel('P')
plotLine, = ax.plot(x, np.zeros(len(x))*np.NaN, 'r-')
plotTitle = ax.set_title("t=0")
ax.set_ylim(0,1.)
ax.set_xlim(xmin,xmax)


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


def animate(t):
    pp = solution(t)
    plotLine.set_ydata(pp)
    plotTitle.set_text(f"t = {t:.1f}")
    #ax.relim() # use if autoscale desired
    #ax.autoscale()
    return [plotLine,plotTitle]



ani = animation.FuncAnimation(fig, func=animate, frames=np.arange(t_initial, t_final+dt, dt), blit=True)
plt.show()