Adding other ways to do animations in MPL.

Author: Jonathan Rocher

matplotlib/animation_demo4.py

@@ -0,0 +1,34 @@
+import numpy as np
+import matplotlib.pyplot as plt
+import matplotlib.animation as animation
+
+def update_line(num, data, line):
+    line.set_data(data[...,:num])
+    return line,
+
+fig1 = plt.figure()
+
+data = np.random.rand(2, 25)
+l, = plt.plot([], [], 'r-')
+plt.xlim(0, 1)
+plt.ylim(0, 1)
+plt.xlabel('x')
+plt.title('test')
+line_ani = animation.FuncAnimation(fig1, update_line, 25, fargs=(data, l),
+    interval=50, blit=True)
+#line_ani.save('lines.mp4')
+
+fig2 = plt.figure()
+
+x = np.arange(-9, 10)
+y = np.arange(-9, 10).reshape(-1, 1)
+base = np.hypot(x, y)
+ims = []
+for add in np.arange(15):
+    ims.append((plt.pcolor(x, y, base + add, norm=plt.Normalize(0, 30)),))
+
+im_ani = animation.ArtistAnimation(fig2, ims, interval=50, repeat_delay=3000,
+    blit=True)
+#im_ani.save('im.mp4', metadata={'artist':'Guido'})
+
+plt.show()

matplotlib/animation_demo5.py

@@ -0,0 +1,58 @@
+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()
+    # Saving to a movie file requires for ffmpeg or mencoder or something like 
+    # that to be present on the machine/passed to the save function.
+    #a.ani.save("movie.avi", codec='avi')
+    a.show()