How to generate animation using numpy arrays.

generate_gif.py

from matplotlib.animation import FuncAnimation
import matplotlib.pyplot as plt
import numpy as np

fig, ax = plt.subplots(figsize=(5, 8))


def update(i):
    im_normed = np.random.random((64, 64))
    ax.imshow(im_normed)
    ax.set_title("Angle: {}*pi/10".format(i), fontsize=20)
    ax.set_axis_off()


anim = FuncAnimation(fig, update, frames=np.arange(0, 20), interval=50)
anim.save('colour_rotation.gif', dpi=80, writer='imagemagick')
plt.close()

Is there a way to embed this in notebook?

I got this code working in a notebook using a markdown cell with the following code,

![SegmentLocal](colour_rotation.gif "colour_rotation")