Python, Matplotlib, trazar líneas múltiples (matriz) y animación
Estoy empezando a programar en Python (y OOP), pero tengo una sólida experiencia en programación en Fortran (90/95) y Matlab.
Estoy desarrollando una pequeña herramienta que usa animación en el entorno tkinter. El objetivo de esta herramienta es animar varias líneas (una matriz y no un vector de datos). A continuación, un simple ejemplo de mi problema. No entiendo por qué el resultado de estas dos formas de trazar datos es tan diferente.
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()
Si solo traza "dline", cada línea se traza por separado y con un color diferente. Si grafica solo "mline" todas las líneas están vinculadas y con un solo color.
Mi objetivo es hacer una animación con "mline" cambiando los datos en cada bucle. Aquí un código fuente simple que ilustra mis propósitos:
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()
Espero haber expuesto claramente mi problema.
Gracias nicolas