Matplotlib en Python - Dibujando formas y animándolas

Question

Así que estoy representando una red Token Ring (haciendo la simulación en SimPy), soy totalmente novato en matplotlib, pero me dijeron que sería realmente bueno para representando mi simulación visualmente

Así que busqué en Google y descubrí cómo dibujar formas y líneas, usando add_patch y add_line, respectivamente, a los ejes (creo). Así que ahora tengo esta salida que es absolutamente bien:

(No puede enviar imágenes hasta ahora !!) http://img137.imageshack.us/img137/7822/screenshot20100121at120.png

Pero me estoy haciendo esto usando la función pylab.show(), y lo que pienso Lo que quiero es lograr esto usando la función pylab.plot() para poder luego actualizarla a medida que progresa mi simulación usando pylab.draw() luego.

Mi código es el siguiente:

plab.ion() 

plab.axes() 

for circ in self.circleList: 
    plab.gca().add_patch(circ) 

for line in self.lineList: 
    plab.gca().add_line(line) 

plab.axis('scaled') 
plab.show() 

Dónde circleList y LineList son listas con los círculos y las líneas en el diagrama

probablemente estoy algo simple malentendido aquí, pero en realidad no puede encuentre ejemplos que no estén basados ​​abiertamente en gráficos que usen la función plot().

---

Aclaración:

¿Cómo puedo conseguir que la misma salida, utilizando pylab.plot() en lugar de pylab.show()?

Answer 1

Replicar su imagen utilizando el método de la gráfica:

from pylab import * 

points = [] 
points.append((-0.25, -1.0)) 
points.append((0.7, -0.7)) 
points.append((1,0)) 
points.append((0.7,1)) 
points.append((-0.25,1.2)) 
points.append((-1,0.5)) 
points.append((-1,-0.5)) 
points.append((-0.25, -1.0)) 

a_line = plot(*zip(*points))[0] 
a_line.set_color('g') 
a_line.set_marker('o') 
a_line.set_markerfacecolor('b') 
a_line.set_markersize(30) 
axis([-1.5,1.5,-1.5,1.5]) 

show() 

EDITAR basándose en las observaciones

Esto utiliza pitón multiprocesamiento biblioteca para ejecutar la animación matplotlib en un proceso separado. El proceso principal utiliza una cola para pasarle datos que luego actualiza la imagen de la gráfica.

# general imports 
import random, time 
from multiprocessing import Process, Queue 

# for matplotlib 
import random 
import numpy as np 
import matplotlib 
matplotlib.use('GTKAgg') # do this before importing pylab 

import matplotlib.pyplot as plt 
from matplotlib.patches import Circle 


def matplotLibAnimate(q,points): 

    # set up initial plot 
    fig = plt.figure() 
    ax = fig.add_subplot(111) 


    circles = [] 
    for point in points: 
     ax.add_patch(Circle(point,0.1)) 

    a_line, = ax.plot(*zip(*points)) 
    a_line.set_color('g') 
    a_line.set_lw(2) 

    currentNode = None 
    def animate(currentNode = currentNode): 
     while 1: 
      newNode = q.get() 
      if currentNode: currentNode.remove() 
      circle = Circle(newNode,0.1) 
      currentNode = ax.add_patch(circle) 
      circle.set_fc('r') 
      fig.canvas.draw() 

    # start the animation 
    import gobject 
    gobject.idle_add(animate) 
    plt.show() 

#initial points 
points = ((-0.25, -1.0),(0.7, -0.7),(1,0),(0.7,1),(-0.25,1.2),(-1,0.5),(-1,-0.5),(-0.25, -1.0)) 
q = Queue() 
p = Process(target = matplotLibAnimate, args=(q,points,)) 
p.start() 

# feed animation data 
while 1: 
    time.sleep(random.randrange(4)) 
    q.put(random.sample(points,1)[0]) 

Por supuesto, después de hacer esto, creo que estará mejor servido con la solución de imagen de whatnick. Crearía mi propia GUI y no usaría matplotlibs en el widget. Entonces "animé" mi GUI generando PNG e intercambiándolos.

Answer 2

La técnica básica consiste en actualizar los datos de los elementos que se procesan utilizando set_data. Luego llama a draw(). Vea si sus elementos de círculo y línea tienen funciones set_data. De lo contrario, puede usar pyvtk. La otra opción es renderizar y guardar los trazados en archivos png y luego crear una animación a partir de ellos.

Answer 3

Suena como Mark tiene la respuesta que estabas buscando, pero si usted decide ir con el enfoque de whatnick y construir una animación a partir PNG individuales, aquí está el código para implementar la sugerencia de Amit usar mencoder (de http://en.wikibooks.org/wiki/Mplayer):

mencoder mf://*.png -mf w=400:h=400 -ovc lavc -lavcopts vcodec=xvid -of avi -o output.avi