动态显示图
import numpy as np
import matplotlib.pyplot as plt
import matplotlib.animation as animation
fig, ax = plt.subplots(2, 2)
lines = []
x = np.arange(0, 2*np.pi, 0.01)
lines = ax[0][0].plot(x, np.sin(x))
lines += ax[0][0].plot(x, np.cos(x))
line1, = ax[0][1].plot(x, np.tan(x))
line2, = ax[1][0].plot(x, np.zeros(len(x)))
line3, = ax[1][1].plot(x, np.zeros(len(x)))
def animate(i):
lines[0].set_ydata(np.sin(x + i / 50)) # update the data.
lines[1].set_ydata(np.cos(x + i / 50)) # update the data.
line1.set_ydata(np.tan(x + i / 50)) # update the data.
return lines + [line1]
ani = animation.FuncAnimation(
fig, animate, interval=20, blit=True, save_count=50)
# To save the animation, use e.g.
#
# ani.save("movie.mp4")
#
# or
#
# writer = animation.FFMpegWriter(
# fps=15, metadata=dict(artist='Me'), bitrate=1800)
# ani.save("movie.mp4", writer=writer)
plt.show()