matplotlibでアニメーションを表示

matplotlibでアニメーションを表示する。K-meansや混合ガウスモデルを使ったクラスタリングを説明するのに使いたいので、その準備としてメモ。
残念ながらipython notebookでは最終結果しか表示されない。

import numpy as np
import matplotlib.pyplot as plt
import time

x = np.linspace(-np.pi, np.pi, 100)
y = np.sin(x)

plt.ion()

fig = plt.figure()
ax = fig.add_subplot(111)
line, = ax.plot(x, y)

N=50
for i in xrange(N):
    line.set_ydata(np.sin(x+i*np.pi/N))
    fig.canvas.draw()
    #time.sleep(.05)