Python, Matplotlib, plot multi-lines (array) i animacja
Zaczynam programowanie w Pythonie (i OOP), ale mam solidne doświadczenie w programowaniu Fortran (90/95) i Matlab.
Tworzę małe narzędzie wykorzystujące animację w środowisku tkinter. Celem tego narzędzia jest animacja wielu linii (tablica, a nie wektor danych). Poniżej prosty przykład mojego problemu. Nie rozumiem, dlaczego wyniki tych dwóch sposobów kreślenia danych są tak różne?
from pylab import *
Nx=10
Ny=20
xx = zeros( ( Nx,Ny) )
data = zeros( ( Nx,Ny) )
for ii in range(0,Nx):
for jj in range(0,Ny):
xx[ii,jj] = ii
data[ii,jj] = jj
dline = plot(xx,data)
mline, = plot([],[])
mline.set_data(xx.T,data.T)
show()
Jeśli drukujesz tylko „dline”, każda linia jest wykreślana oddzielnie iz innym kolorem. Jeśli wykreślasz tylko „linię”, wszystkie linie są połączone i mają tylko jeden kolor.
Moim celem jest stworzenie animacji z „mline” zmieniającym dane w każdej pętli. Oto prosty kod źródłowy ilustrujący moje cele:
from pylab import *
from matplotlib import animation
Nx=10
Ny=20
fig = plt.figure()
fig.set_dpi(100)
fig.set_size_inches(7, 6.5)
ax = plt.axes(xlim=(0, Nx), ylim=(0, Ny))
xx = zeros( ( Nx,Ny) )
data = zeros( ( Nx,Ny) )
odata = zeros( ( Nx,Ny) )
for ii in range(0,Nx):
for jj in range(0,Ny):
xx[ii,jj] = ii
odata[ii,jj] = jj
data[ii,jj] = 0.
#dline = plot(xx,odata)
mline, = plot([],[])
def init():
mline.set_data([],[])
return mline,
def animate(coef):
for ii in range(0,Nx):
for jj in range(0,Ny):
data[ii,jj] = odata[ii,jj] * (1.-float(coef)/360.)
mline.set_data(xx.T,data.T)
return mline,
anim = animation.FuncAnimation(fig, animate,
init_func=init,
frames=360,
interval=5,
blit=True)
plt.show()
Mam nadzieję, że wyraźnie ujawniłem mój problem.
Dzięki, Nicolas.