使用PyPlot为ScatterPlot设置动画

时间:2015-08-28 10:10:59

标签: python matplotlib ipython jupyter

我正试图用pyplot绘制粒子的运动。 问题是我无法弄清楚如何创建动画。

这是笔记本:http://nbviewer.ipython.org/gist/lhk/949c7bf7007445033fd9

显然更新功能无法正常工作,但错误信息对我来说太神秘了。我需要改变什么?

你有关于pyplot的动画教程吗?

1 个答案:

答案 0 :(得分:4)

作为@ s0upa1t注释,您应该将数字句柄作为动画的第一个参数。由于动画需要fig对象,该对象具有属性canvas,而是获得scatterPathCollection对象,而不是import matplotlib.pyplot as plt from matplotlib.animation import FuncAnimation import numpy as np dt = 0.005 n=20 L = 1 particles=np.zeros(n,dtype=[("position", float , 2), ("velocity", float ,2), ("force", float ,2), ("size", float , 1)]) particles["position"]=np.random.uniform(0,L,(n,2)); particles["velocity"]=np.zeros((n,2)); particles["size"]=0.5*np.ones(n); fig = plt.figure(figsize=(7,7)) ax = plt.axes(xlim=(0,L),ylim=(0,L)) scatter=ax.scatter(particles["position"][:,0], particles["position"][:,1]) def update(frame_number): particles["force"]=np.random.uniform(-2,2.,(n,2)); particles["velocity"] = particles["velocity"] + particles["force"]*dt particles["position"] = particles["position"] + particles["velocity"]*dt particles["position"] = particles["position"]%L scatter.set_offsets(particles["position"]) return scatter, anim = FuncAnimation(fig, update, interval=10) plt.show() 对象。作为您想要的形式的动画的最小示例,请考虑,

True

many个好动画教程,但答案here特别好。