python用animation绘制动画

代码

import matplotlib.pyplot as plt
import numpy as np
from matplotlib.animation import FuncAnimation
import os
fig, ax = plt.subplots(1,1)
 
x = np.linspace(0, 2 * np.pi, 5000)
y=np.exp(-x)*np.cos(2*np.pi*x)
line, = ax.plot(x,y,color="cornflowerblue",lw=3)
ax.set_ylim(-1.0,1.0)
 
def animate(data):
    line.set_ydata(np.exp(-x)*np.cos(2*np.pi*x+float(data)/100))
    return line,


def init():
    line.set_ydata([np.nan]*len(x))
    return line,
 
 
ani = FuncAnimation(fig,
                    animate,
                    frames=1000,
                    init_func=init,
                    interval=2,
                    blit=True)

 
plt.show()