Matplotlib ile scatterplot animasyonlarını kaydetme

Question

Animasyonlu bir dağılım grafiğini matplotlib ile kaydetmeye çalışıyorum ve animasyonlu bir şekil olarak görüntülemek ve bir kopyasını kaydetmek için tamamen farklı bir kod gerektirmemesini tercih ederim. Şekil, kayıt tamamlandıktan sonra tüm veri noktalarını mükemmel bir şekilde gösterir.

Bu kod Matplotlib 3D scatter color lost after redraw üzerinde Yann's answer gelen renkler için bir düzeltme ile Animating 3d scatterplot in matplotlib üzerinde Giggi's değiştirilmiş bir versiyonudur (renk Videomda önemli olacaktır çünkü, bu yüzden çalıştıklarından emin olmak istiyoruz).

import matplotlib.pyplot as plt 
import matplotlib.animation as animation 
import numpy as np 
from mpl_toolkits.mplot3d import Axes3D 

FLOOR = -10 
CEILING = 10 

class AnimatedScatter(object): 
    def __init__(self, numpoints=5): 
     self.numpoints = numpoints 
     self.stream = self.data_stream() 
     self.angle = 0 

     self.fig = plt.figure() 
     self.fig.canvas.mpl_connect('draw_event',self.forceUpdate) 
     self.ax = self.fig.add_subplot(111,projection = '3d') 
     self.ani = animation.FuncAnimation(self.fig, self.update, interval=100, 
             init_func=self.setup_plot, blit=True,frames=20) 

    def change_angle(self): 
     self.angle = (self.angle + 1)%360 

    def forceUpdate(self, event): 
     self.scat.changed() 

    def setup_plot(self): 
     X = next(self.stream) 
     c = ['b', 'r', 'g', 'y', 'm'] 
     self.scat = self.ax.scatter(X[:,0], X[:,1], X[:,2] , c=c, s=200, animated=True) 

     self.ax.set_xlim3d(FLOOR, CEILING) 
     self.ax.set_ylim3d(FLOOR, CEILING) 
     self.ax.set_zlim3d(FLOOR, CEILING) 

     return self.scat, 

    def data_stream(self): 
     data = np.zeros((self.numpoints , 3)) 
     xyz = data[:,:3] 
     while True: 
      xyz += 2 * (np.random.random((self.numpoints,3)) - 0.5) 
      yield data 

    def update(self, i): 
     data = next(self.stream) 
     #data = np.transpose(data) 

     self.scat._offsets3d = (np.ma.ravel(data[:,0]) , np.ma.ravel(data[:,1]) , np.ma.ravel(data[:,2])) 

     plt.draw() 
     return self.scat, 

    def show(self): 
     plt.show() 

if __name__ == '__main__': 
    a = AnimatedScatter() 
    a.ani.save("movie.avi", codec='avi') 
    a.show() 

Mükemmel geçerli .avi bu tarafından oluşturulan, ancak eksenler haricinde dört saniye tümü için boş. Asıl rakam her zaman tam olarak görmek istediğimi gösterir. Kaydetme işlevinin grafiklerini, normal olarak çalışan bir animasyonu doldurduğum şekilde nasıl doldurabilirim veya matplotlib'de mümkün mü?

DÜZENLEME: (in başlatıcısı olarak sınırları ayarlamadan) güncellemede bir dağılım çağrısını kullanma sadece gösterilmediğinden, eksenler veriler her zaman yürütülüyor gittiğine işaret büyüyen göstermek için avi neden olur videonun kendisi. Python 2.7.3 ile matplotlib 1.1.1rc kullanıyorum.

Answer 1

FuncAnimation ve animated=True'dan gelen blit=True'u scatter'dan kaldırın ve çalışır. Mantıkla yanlış giden bir şey olduğunu sanıyorum, sadece güncellenmesi gereken sanatçıların çerçeveler arasında (her şeyi yeniden çizmek yerine) güncellendiğinden/yeniden çizildiğinden emin olur.

import matplotlib.pyplot as plt 
import matplotlib.animation as animation 
import numpy as np 
from mpl_toolkits.mplot3d import Axes3D 

FLOOR = -10 
CEILING = 10 

class AnimatedScatter(object): 
    def __init__(self, numpoints=5): 
     self.numpoints = numpoints 
     self.stream = self.data_stream() 
     self.angle = 0 

     self.fig = plt.figure() 
     self.fig.canvas.mpl_connect('draw_event',self.forceUpdate) 
     self.ax = self.fig.add_subplot(111,projection = '3d') 
     self.ani = animation.FuncAnimation(self.fig, self.update, interval=100, 
             init_func=self.setup_plot, frames=20) 

    def change_angle(self): 
     self.angle = (self.angle + 1)%360 

    def forceUpdate(self, event): 
     self.scat.changed() 

    def setup_plot(self): 
     X = next(self.stream) 
     c = ['b', 'r', 'g', 'y', 'm'] 
     self.scat = self.ax.scatter(X[:,0], X[:,1], X[:,2] , c=c, s=200) 

     self.ax.set_xlim3d(FLOOR, CEILING) 
     self.ax.set_ylim3d(FLOOR, CEILING) 
     self.ax.set_zlim3d(FLOOR, CEILING) 

     return self.scat, 

    def data_stream(self): 
     data = np.zeros((self.numpoints , 3)) 
     xyz = data[:,:3] 
     while True: 
      xyz += 2 * (np.random.random((self.numpoints,3)) - 0.5) 
      yield data 

    def update(self, i): 
     data = next(self.stream) 
     self.scat._offsets3d = (np.ma.ravel(data[:,0]) , np.ma.ravel(data[:,1]) , np.ma.ravel(data[:,2])) 
     return self.scat, 

    def show(self): 
     plt.show() 

if __name__ == '__main__': 
    a = AnimatedScatter() 
    a.ani.save("movie.avi", codec='avi') 
    a.show()