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()