使用Python的Matplotlib 3D散点图动画不会出错,但它也不起作用

时间:2015-04-12 10:37:47

标签: python matplotlib physics simulator graphing

我正在尝试写一个N-body simulation,而且我遇到了Matplotlib动画功能的障碍。我不完全确定如何更新散点图中的数据,所以我认为这可能是问题所在。

import numpy as np
import matplotlib.pyplot as plt
from mpl_toolkits.mplot3d import Axes3D
import matplotlib.animation as animation
system = []
xlist = []
ylist = []
zlist = []
poslist = []
nbodies = 3
scale = 100
mass = 0
timestep = 0.1
step = 0.01
fig = plt.figure()
ax = fig.add_subplot(111, projection='3d')
i = 1
plt.ion()
scat = ax.scatter(xlist, ylist, zlist, c='r', marker='o', animated=True)
class Body:
    def __init__(self, posvel, mass):
        self.state = np.zeros((3, 3))
        self.state[0:2] = posvel * scale
        self.state[2] = np.array([0.0, 0.0, 0.0])
        self.mass = mass
    def setPos(self, pos):
        self.state[0] = pos
    def setVel(self, vel):
        self.state[1] = vel
    def setAcc(self, acc):
        self.state[2] = acc
    def getPos(self):
        return self.state[0]
    def getVel(self):
        return self.state[1]
    def getAcc(self):
        return self.state[2]
def startSystem():
    for i in range(0,nbodies):
        system.append(Body(np.random.rand(2,3), np.random.random_sample()))
        curpos = system[i].getPos()
        xlist.append(curpos[0])
        ylist.append(curpos[1])
        zlist.append(curpos[2])
def stepSystem():
    xlist = []
    ylist = []
    zlist = []
    poslist = []
    for i in range(0,nbodies):
        curpos = system[i].getPos()
        system[i].setPos(curpos + 1*step)
        poslist.append(curpos + 1*step)
        xlist.append(curpos[0] + 1*step)
        ylist.append(curpos[1] + 1*step)
        zlist.append(curpos[2] + 1*step)
    yield xlist, ylist, zlist
def simPoints(stepSystem):
    xlist, ylist,zlist = stepSystem[0], stepSystem[1], stepSystem[2]
    scat.set_offsets(poslist)
    return scat
def Plot():
    ani = animation.FuncAnimation(fig, simPoints, stepSystem, blit=False,\
     interval=100, repeat=True)
    plt.show()
def main():
    startSystem()
    Plot()

0 个答案:

没有答案