Matplotlib动画更新功能不是设置数据

时间:2017-09-05 14:32:28

标签: python animation matplotlib

在生成几张图之后,我正在阅读fortran的一些模拟中的输出数据,以制作一部轨道电影。起初,我没有使用blitting来制作动画,所以虽然它有效,但它非常非常慢。

我最初认为我想要的动画可以分散,因为我有五个系列的数据减少alphas以创建一个尾随效果。这是我的原始(非blit)更新功能:

def animate(frame):
    jptx, jpty = jx[frame-3:frame], jy[frame-3:frame]
    cptx, cpty = cx[frame-3:frame], cy[frame-3:frame]
    eptx, epty = ex[frame-3:frame], ey[frame-3:frame]
    gptx, gpty = gx[frame-3:frame], gy[frame-3:frame]
    iptx, ipty = ix[frame-3:frame], iy[frame-3:frame]
    ax2.clear()
    ax2.scatter(jptx, jpty, s=32, c=ablue, marker="s", label='Jupiter')
    ax2.scatter(cptx, cpty, s=8, c=ared, marker="o", label='Callisto')
    ax2.scatter(eptx, epty, s=8, c=agreen, marker="o", label='Europa')
    ax2.scatter(gptx, gpty, s=8, c=ablack, marker="o", label='Ganymede')
    ax2.scatter(iptx, ipty, s=8, c=ayellow, marker="o", label='Io')
    ax2.set_xlim(-3, 7)
    ax2.set_ylim(-3, 4)
animation = animation.FuncAnimation(fig2, animate, interval=0.5, frames=jt.size)
print('Begin saving animation')
animation.save('Tabbys Star.mp4', writer='ffmpeg', fps=60)
print('Animation saved')
plt.show()

现在,当我运行脚本时,窗口会出现一小段时间,并且屏幕上有一个非常清晰的黄色圆圈,表示正在绘制背景。但是,窗口立即关闭。这是第二次尝试的相关代码。在这次尝试中添加了黄色圆圈。

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

# j_file = location + 'JUPITER.aei'
# jt, jx, jy, jz = read_data(j_file)
jt, jx, jy, jz = np.random.random([100,4]), np.random.random([100,4]), np.random.random([100,4]), np.random.random([100,4])

# c_file = location + 'CALLISTO.aei'
# ct, cx, cy, cz = read_data(c_file)
ct, cx, cy, cz = np.random.random([100,4]), np.random.random([100,4]), np.random.random([100,4]), np.random.random([100,4])

alphas = [0.25, 0.5, 0.75, 1]

ablue = np.zeros((4, 4))
ablue[:, 2] = 1.0
ablue[:, 3] = alphas

ared = np.zeros((4, 4))
ared[:, 0] = 1.0
ared[:, 3] = alphas

fig2 = plt.figure()
ax2 = fig2.add_subplot(111, aspect='equal')
xdata, ydata = np.zeros((4,)), np.zeros((4,))
jpt, = plt.plot(xdata, ydata, marker='.', ms=32, c=ablue, label='Jupiter')
cpt, = plt.plot(xdata, ydata, marker='.', ms=8, c=ared, label='Callisto')


def init():
    ax2.set_xlim(-3, 7)
    ax2.set_ylim(-3, 4)
    circle = plt.Circle((0, 0), 0.1, color='y')
    ax2.add_patch(circle)
    for pt in [jpt, cpt]:
        pt.set_data(np.zeros((4,)), np.zeros((4,)))
    return jpt, cpt


def animate(frame, j, c):
    jptx, jpty = jx[frame-3:frame], jy[frame-3:frame]
    cptx, cpty = cx[frame-3:frame], cy[frame-3:frame]
    j.set_data(jptx, jpty)
    c.set_data(cptx, cpty)
    return j, c


animation = animation.FuncAnimation(fig2, animate, fargs=(jpt, cpt), interval=0.5, frames=jt.size, init_func=init, blit=True)
print('Begin saving animation')
# animation.save('Tabbys Star.mp4', writer='ffmpeg', fps=60)
print('Animation saved')
plt.show()

我最终还想添加一个图例和一些轴标签,但我相信这可以正常完成。

那么第二个代码段中动画的问题是什么?

由于

为了清晰而编辑(再次)

2 个答案:

答案 0 :(得分:0)

请确保您将setting struct tab { struct node *vet[64]; }; struct node { char nome[255]; int num; int tipo; char *dati; struct tab *array; struct node *chain; }; 提取的值超过1帧,并将其提高到较高值。在您发布的代码中,帧数没有明确定义,这可能会导致此问题。

答案 1 :(得分:0)

您在此处混淆了plt.plotplt.scatter。你得到的错误甚至可以在没有任何动画的情况下产生。

虽然plt.plot有参数colorms分别设置颜色和标记,但它们不允许对不同的点使用不同的值。这就是存在scatter图的原因。 plt.scatter有参数cs分别设置颜色和标记。

所以你需要使用散射来获得不同颜色的点。

jpt = plt.scatter(xdata, ydata, marker='.', s=32, c=ablue, label='Jupiter')

然后,对于动画,您需要调整代码以便与scatter一起使用,因为它没有.set_data方法,而是.set_offsets方法,需要2列数组输入。

j.set_offsets(np.c_[jptx, jpty])

总的来说,脚本看起来像

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


jt, jx, jy, jz = [np.random.random([100,4]) for _ in range(4)]
ct, cx, cy, cz = [np.random.random([100,4]) for _ in range(4)]

alphas = [0.25, 0.5, 0.75, 1]

ablue = np.zeros((4, 4))
ablue[:, 2] = 1.0
ablue[:, 3] = alphas

ared = np.zeros((4, 4))
ared[:, 0] = 1.0
ared[:, 3] = alphas

fig2 = plt.figure()
ax2 = fig2.add_subplot(111, aspect='equal')
xdata, ydata = np.zeros((4,)), np.zeros((4,))

jpt = plt.scatter(xdata, ydata, marker='.', s=32, c=ablue, label='Jupiter')
cpt = plt.scatter(xdata, ydata, marker='.', s=8, c=ared, label='Callisto')


def init():
    ax2.axis([0,1,0,1])
    circle = plt.Circle((0, 0), 0.1, color='y')
    ax2.add_patch(circle)
    for pt in [jpt, cpt]:
        pt.set_offsets(np.c_[np.zeros((4,)), np.zeros((4,))])
    return jpt, cpt


def animate(frame, j, c):
    jptx, jpty = jx[frame-3:frame], jy[frame-3:frame]
    cptx, cpty = cx[frame-3:frame], cy[frame-3:frame]
    j.set_offsets(np.c_[jptx, jpty])
    c.set_offsets(np.c_[cptx, cpty])
    return j, c


animation = animation.FuncAnimation(fig2, animate, fargs=(jpt, cpt), 
                                    interval=50, frames=jt.size, init_func=init, blit=True)

plt.show()