Dynamisches Diagramm in Python

Question

Ich versuche, ein Diagramm in Python zu erstellen, in dem die Daten, die geplottet werden, aktualisiert werden, während meine Simulation fortschreitet. In MATLAB ich mit dem folgenden Code tun könnte:

t = linspace(0, 1, 100); 
figure 
for i = 1:100 
x = cos(2*pi*i*t); 
plot(x) 
drawnow 
end 

Ich versuche matplotlib ‚s FuncAnimation Funktion im animation Modul zu tun dies in einer Klasse zu verwenden. Es ruft eine Funktion plot_voltage auf, die die Spannung nach jedem Zeitschritt in meiner Simulation neu berechnet. Ich habe es wie folgt eingerichtet:

import matplotlib.pyplot as plt 
import matplotlib.animation as animation 

def __init__(self): 
    ani = animation.FuncAnimation(plt.figure(2), self.plot_voltage) 
    plt.draw() 

def plot_voltage(self, *args): 
    voltages = np.zeros(100) 
    voltages[:] = np.nan 

    # some code to calculate voltage 

    ax1 = plt.figure(2).gca() 
    ax1.clear() 
    ax1.plot(np.arange(0, len(voltages), 1), voltages, 'ko-')` 

Wenn meine Simulation ausgeführt wird, erscheinen die Zahlen, aber nur einfrieren. Der Code läuft jedoch ohne Fehler. Kann mir bitte jemand sagen, was ich vermisse?

Answer 1

Dies ist eine Übersetzung des Matlab-Code in matplotlib FuncAnimation mit:

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

t = np.linspace(0, 1, 100) 
fig = plt.figure() 
line, = plt.plot([],[]) 

def update(i): 
    x = np.cos(2*np.pi*i*t) 
    line.set_data(t,x) 

ani = animation.FuncAnimation(fig, update, 
       frames=np.linspace(1,100,100), interval=100) 
plt.xlim(0,1) 
plt.ylim(-1,1) 
plt.show()