动画在matplotlib中没有动画

时间:2015-01-04 16:38:05

标签: python animation matplotlib

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

这是我到目前为止(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()

1 个答案:

答案 0 :(得分:2)

您必须保存在FuncAnimation中创建的animate(self)对象。否则在调用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)