Using matplotlib's FuncAnimation to do a basic animation of a sine wave moving across the screen:

Moving Sine Wave.py

import numpy as np
from matplotlib import pyplot as plt
from matplotlib.animation import FuncAnimation
plt.style.use('seaborn-pastel')


fig = plt.figure()
ax = plt.axes(xlim=(0, 4), ylim=(-2, 2))
line, = ax.plot([], [], lw=3)

def init():
    line.set_data([], [])
    return line,
def animate(i):
    x = np.linspace(0, 4, 1000)
    y = np.sin(2 * np.pi * (x - 0.01 * i))
    line.set_data(x, y)
    return line,

anim = FuncAnimation(fig, animate, init_func=init,
                               frames=200, interval=20, blit=True)


anim.save('sine_wave.gif', writer='imagemagick')

Nice..

when I run this code I am getting "TypeError: 'MovieWriterRegistry' object is not an iterator". How do I fix this?

Hello.

Is it possible to show the animation just in the notebook without saving it to file? If I remove the anim.save line I just get an empty figure.

Thanks