Python, Matplotlib, plot multi-lines (array) e animação
Estou iniciando a programação em Python (e OOP), mas tenho uma sólida experiência em programação Fortran (90/95) e Matlab.
Estou desenvolvendo uma pequena ferramenta usando animação no ambiente do tkinter. O objetivo dessa ferramenta é animar várias linhas (uma matriz e não um vetor de dados). Abaixo, um exemplo simples do meu problema. Eu não entendo porque o resultado dessas duas maneiras de plotar dados é tão 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()
Se você traçar apenas "dline", cada linha é plotada separadamente e com uma cor diferente. Se você traçar apenas "mline", todas as linhas serão vinculadas e com apenas uma cor.
Meu objetivo é fazer uma animação com "mline" alterando os dados em cada loop. Aqui um código fonte simples ilustrando meus 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 ter exposto claramente o meu problema.
Obrigado Nicolas.