animate.py

animate.py

#! /usr/bin/env python3
import matplotlib.pyplot as plt
import matplotlib.animation as anim
import numpy as np
x = np.arange(0,2*(np.pi),.01)
y, z = 0 * x, 0 * x
fig = plt.figure()
ax = plt.axes(xlim=(0, 2*np.pi), ylim=(-2, 2))
yline, = ax.plot([], [])
zline, = ax.plot([], [])
def init():
    y.fill(0)
    z.fill(0)
    yline.set_data(x, y)
    zline.set_data(x, z)
    return yline, zline
def animate(i):
    global y, z
    y += np.sin((2*i+1)*x)/(2*i+1)
    z += np.sin((i+1)*x)/(i+1)
    yline.set_data(x, y)
    zline.set_data(x, z)
    return yline, zline
a = anim.FuncAnimation(fig, animate, init_func=init, frames=40, interval=250)
a.save("foo.gif")
plt.show()