Animowanie "rosnącego" wykresu liniowego w Pythonie/Matplotlib

Question

Chcę stworzyć zestaw ramek, które można wykorzystać do animowania fabuły rosnącej linii. W przeszłości zawsze używałam plt.draw() i set_ydata() do przerysowywania danych y, które zmieniały się w czasie. Tym razem chciałbym narysować "rosnącą" linię, przesuwając się po wykresie z czasem. Z tego powodu set_ydata nie działa (xdata zmienia długość). Na przykład:

import numpy as np 
import matplotlib.pyplot as plt 

x = np.linspace(0, 10, 100) 
y = np.sin(x) 

plt.figure() 
for n in range(len(x)): 
    plt.plot(x[:n], y[:n], color='k') 
    plt.axis([0, 10, 0, 1]) 
    plt.savefig('Frame%03d.png' %n) 

Podczas gdy to działa, staje się bardzo powolne w miarę skalowania. Czy jest to szybszy sposób?

Answer 1

Kilka uwag:

Po pierwsze, dlatego, że rzeczy stają się coraz wolniej, jest to, że jesteś rysunek coraz więcej i więcej nakładających się linii w tej samej pozycji.

Quick Fix jest oczyszczenie działki za każdym razem:

import numpy as np 
import matplotlib.pyplot as plt 

x = np.linspace(0, 10, 100) 
y = np.sin(x) 

plt.figure() 
for n in range(len(x)): 
    plt.cla() 
    plt.plot(x[:n], y[:n], color='k') 
    plt.axis([0, 10, 0, 1]) 
    plt.savefig('Frame%03d.png' %n) 

Jeszcze lepiej jednak aktualizować zarówno xiy dane w tym samym czasie:

import numpy as np 
import matplotlib.pyplot as plt 

x = np.linspace(0, 10, 100) 
y = np.sin(x) 

fig, ax = plt.subplots() 
line, = ax.plot(x, y, color='k') 

for n in range(len(x)): 
    line.set_data(x[:n], y[:n]) 
    ax.axis([0, 10, 0, 1]) 
    fig.canvas.draw() 
    fig.savefig('Frame%03d.png' %n) 

a jeśli” lubię używać modułu animacji (notatka boczna: blit=True może nie działać poprawnie w niektórych serwerach (np. OSX), więc spróbuj blit=False, jeśli masz problemy):

import numpy as np 
import matplotlib.pyplot as plt 
import matplotlib.animation as animation 

x = np.linspace(0, 10, 100) 
y = np.sin(x) 

fig, ax = plt.subplots() 
line, = ax.plot(x, y, color='k') 

def update(num, x, y, line): 
    line.set_data(x[:num], y[:num]) 
    line.axes.axis([0, 10, 0, 1]) 
    return line, 

ani = animation.FuncAnimation(fig, update, len(x), fargs=[x, y, line], 
           interval=25, blit=True) 
ani.save('test.gif') 
plt.show()