In [1]:
#http://louistiao.me/posts/notebooks/embedding-matplotlib-animations-in-jupyter-notebooks/
%matplotlib inline
In [17]:
import numpy as np
import matplotlib
import matplotlib.pyplot as plt
from matplotlib import animation, rc
from IPython.display import HTML, Image
import matplotlib.animation as animation
print(animation.writers.list())
matplotlib.rcParams['animation.writer'] = 'avconv'
In [4]:
# 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))
line, = ax.plot([], [], lw=2)
In [5]:
# initialization function: plot the background of each frame
def init():
line.set_data([], [])
return (line,)
In [6]:
# animation function. This is called sequentially
def animate(i):
x = np.linspace(0, 2, 1000)
y = np.sin(2 * np.pi * (x - 0.01 * i))
line.set_data(x, y)
return (line,)
In [7]:
# call the animator. blit=True means only re-draw the parts that have changed.
anim = animation.FuncAnimation(fig, animate, init_func=init,
frames=100, interval=20, blit=True)
In [8]:
HTML(anim.to_html5_video())
Out[8]:
In [9]:
anim._repr_html_() is None
Out[9]:
In [10]:
anim
Out[10]:
In [11]:
# equivalent to rcParams['animation.html'] = 'html5'
rc('animation', html='html5')
In [12]:
anim
Out[12]:
In [13]:
anim.save('animation.gif', writer='imagemagick', fps=60)
In [18]:
Image(url='animation.gif')
Out[18]:
In [ ]: