Animación de una onda seno como un generador de señales en python

Animación de una onda seno como un generador de señales en python

#s9_animacion
import numpy as np
import matplotlib.pyplot as plt
from matplotlib import animation
# First set up the figure, the axis, and the plot element we want to animate

fig, ax = plt.subplots()
ax.set_xlim(0, 2)
ax.set_ylim(-2, 2)

x = np.linspace(0, 2, 1000)
y = np.zeros_like(x)

line, = ax.plot(x, y, lw=2)

# initialization function: plot the background of each frame

def init():

    line.set_data([], [])
    return line,
# animation function.  This is called sequentially

def animate(i):

    y = np.sin(2 * np.pi * (x - 0.01 * i))
    line.set_data(x, y)
    return line,
# call the animator.  blit=True means only re-draw the parts that have changed.

anim = animation.FuncAnimation(fig, animate, init_func=init, 

                              frames=200, interval=20, blit=True)
plt.show()

# save the animation as an mp4.  This requires ffmpeg or mencoder to be# installed.  The codec argument ensure that the x264 codec is used, so that# the video can be embedded in html5.  You may need to adjust this for# your system: for more information, see# http://matplotlib.sourceforge.net/api/animation_api.html#anim.save('basic_animation.mp4', fps=30, codec='libx264')