2015-01-04 243 views
2

我正在设计一个模拟查看器,其中点使用matplotlib中的FuncAnimation动画。动画未在matplotlib动画

这是我到目前为止(VX,VY,M,t_lim将在以后使用)。它只产生一个空白的情节,没有任何动作。

我从here的第一个例子中复制了一些。

这真的很简单(字面意思是两个点在一个时间步聚在一起),为什么这不工作?

import numpy as np 
import matplotlib.pyplot as plt 
from matplotlib import animation 

class visualisation(object): 
    def __init__(self, X, Y, VX, VY, M, t_lim=None): 
     self.X = X 
     self.Y = Y 
     self.VX = VX 
     self.VY = VY 
     self.M = M 
     self.t_lim = t_lim 

     self.fig = plt.figure() 
     addx = (X.max() - X.min()) * 0.05 
     addy = (Y.max() - Y.min()) * 0.05 
     self.ax = plt.axes(xlim=(X.min()-addx, X.max()+addx), ylim=(Y.min()-addy, Y.max()+addy)) 
     self.points, = self.ax.plot([], [], 'b.', ms=10) 

    def init(self): 
     self.points.set_data([], []) 
     return self.points, 

    def animator(self, i): 
     print self.X[:,i] 
     self.points.set_data(self.X[:,i], self.Y[:,i]) 
     return self.points, 

    def animate(self): 
     return animation.FuncAnimation(self.fig, self.animator, init_func=self.init, frames=200, interval=20, blit=True) 


M_sun = 2.99e30 
N = 1000 

ms = np.array([1., 1.]) * M_sun 

#initial conditions 
xs = np.zeros([len(ms), N]) #[n, t] 
xs[:, 0] = [0, 1] 

ys = np.zeros([len(ms), N]) #[n, t] 
ys[:, 0] = [0, 1] 

vxs, vys = (np.zeros_like(xs))*2 

visual = visualisation(xs, ys, vxs, vys, ms) 
visual.animate() 

plt.show() 

回答

2

你必须保存您在animate(self)创建FuncAnimation对象。否则,它plt.show()之前收集的垃圾被称为:

amin = visual.animate() 

或:

def animate(self): 
    self.anim = animation.FuncAnimation(self.fig, self.animator, 
      init_func=self.init, frames=200, interval=20, blit=True)