如何更改帧之间的间隔(python)?

时间:2014-09-14 01:43:24

标签: python matplotlib

所以我使用matplotlib.animation在python中制作动画,我希望每帧之间的时间在每一帧上都有所改变。根据我到目前为止发现的所有内容,'间隔' arg只能是一个int,我希望它是一个数组。 代码到目前为止:

import numpy as np
import matplotlib.pyplot as plt
from matplotlib.lines import Line2D
import matplotlib.animation as animation
import time

data=np.loadtxt("coord_dump.dat")
r=(data[:,0]**2+data[:,2]**2)**0.5
distance=np.diff(r)
ti=np.abs(distance)*1e-9
tt=0

fig = plt.figure()
ax=plt.axes(xlim=(-2e16,2e16),ylim=(-1.5e16,1.5e16))
ax.set_xlabel('x(cm)')
ax.set_ylabel('y(cm)')

x1 = data[:,0]
y1 = data[:,2]
x2 = data[:,1]
y2 = data[:,3]
line1 = Line2D([], [], color='black')
line1a = Line2D([], [], color='black', marker='o', markersize=6)
line2= Line2D([], [], color='red')
line2a = Line2D([], [], color='red', marker='o', markersize=6)
ax.add_line(line1)
ax.add_line(line2)
ax.add_line(line1a)
ax.add_line(line2a)

def animate(i):

    line1.set_data(x1[:i], y1[:i])
    line1a.set_data(x1[i], y1[i])
    line2.set_data(x2[:i], y2[:i])
    line2a.set_data(x2[i], y2[i])
    lines =  [line1, line1a, line2, line2a]
    tt=ti[i]
    return lines,tt

def init():
    lines =  [line1, line1a, line2, line2a]
    for l in lines:
        l.set_data([], [])
    return lines,

ani = animation.FuncAnimation(fig,animate,frames=15000, interval=5+tt,init_func=init, blit=True)
print tt
ani.save('xy2.mp4')
plt.show()          

基本上我想要帧i的interval = ti [i]。 (是的,我知道我的代码效率低下,我是python的新手,我只是从一堆网站上捣乱了) 如果这是不可能的,那么之后是否有一种简单的方法来编辑电影以达到同样的效果?

1 个答案:

答案 0 :(得分:1)

您可以更改ani.event_source.interval,如下面的代码

def animate(i):

    line1.set_data(x1[:i], y1[:i])
    line1a.set_data(x1[i], y1[i])
    line2.set_data(x2[:i], y2[:i])
    line2a.set_data(x2[i], y2[i])
    lines =  [line1, line1a, line2, line2a]
    tt=ti[i]
    ani.event_source.interval = tt
    return lines,

警告:它似乎与GIF文件不兼容。