绘制matplotlib中的最后100个点

时间:2018-02-27 19:45:38

标签: python animation matplotlib gif

使用gnuplot和Fortran创建以下gif。但是,现在我想只使用Python(主要是动画形式matplotlib)来做同样的事情。

我可以用Python生成一个gif,但我不知道如何生成类似右手边的gif(你也绘制了最后100个点),相空间的演变。

任何帮助都会被贬低,

由于

enter image description here

gif的Gnuplot代码:

set term gif size 1000,600 animate  delay 1000 loop 0 
set output "animacio.gif"
cd 'C:\Users\Usuario\Desktop'
datafile ="P7-1718-b-res.dat" 


do for[i=1:5000:10]{

set multiplot

set size 0.5,0.8
set origin 0.0,0.0
set title "Evolució de l'angle girat i velocitat angular (t)"
set xrange[0:50]
set yrange[-pi:pi]
set xlabel "t (s)"
set ylabel "Angle girat, v_{ang}"
set key below
plot datafile index 9 every ::1::i with line linewidth 4 t"Posició angular" ,datafile index 9 every ::1::i u 1:3 with line linewidth 4 t"V_{ang}"

set origin 0.5,0
set size 0.5,0.8
set title "Evolució en l'espai de fases"
set yrange[-pi:pi]
set xrange[-pi:pi]
set xlabel "Angle girat(rad)"
set ylabel "Velocitat angular(rad/s)"
set key below
if (i>101) { 
plot datafile index 9 every::i::i u 2:3 t"" ps 3,datafile index 9 every::i-100::i u 2:3 w l t"" } 
else {
plot datafile index 9 every::i::i u 2:3 t"" ps 3}
unset multiplot
}

并假设您的数据在索引9处有三行(时间,位置,角速度)。

1 个答案:

答案 0 :(得分:1)

正如您所提到的,您可以使用Matplotlib动画使其正常工作。

我做了一些事情"类似的"对你的数据,但当然,因为我没有得到数据文件,它是不相等的。代码如下:

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

w = 1.
t = np.linspace(0,10,200)
x = np.cos(w*t)
v = -w*np.cos(w*t)

fig, ax = plt.subplots(1,2)

line_1, = ax[0].plot([], [], 'b-', lw=2)
line_2, = ax[0].plot([], [], 'r-', lw=2)
ax[0].set_xlim([0,50])
ax[0].set_ylim([-np.pi,np.pi])

line_3, = ax[1].plot([], [], 'g-', lw=2)
star_3, = ax[1].plot([], [], 'g*')
ax[1].set_xlim([-np.pi,np.pi])
ax[1].set_ylim([-np.pi,np.pi])

def animate(i):
    line_1.set_data(t[:i],x[:i])  # update the data
    line_2.set_data(t[:i],v[:i])
    nLast = 20
    idFrom = i-nLast if(i-nLast >= 0) else 0
    line_3.set_data(np.cos(t[idFrom:i+1]),np.sin(t[idFrom:i+1]))
    star_3.set_data(np.cos(t[i]),np.sin(t[i]))
    return line_1,line_2,line_3,star_3

# Init only required for blitting to give a clean slate.
def init():
    line_1.set_data([], [])
    line_2.set_data([], [])
    line_3.set_data([], [])
    star_3.set_data([], [])
    return line_1,line_2,line_3,star_3

anim = animation.FuncAnimation(fig, animate, np.arange(1, len(t)), init_func=init, interval=100, blit=True)
#anim.save('Plot_last_nLast.mp4', fps=15)
#anim.save('Plot_last_nLast.gif', dpi=80, writer='imagemagick')
plt.show()

enter image description here

如果您有ffmpeg将其用作动画的编写者,则可以将动画保存为GIF(必需的Imagemagick)或MP4电影。 但这是另一个问题