anim = animation.FuncAnimation(fig, animate, 
            np.arange(0, 200, 10), interval=500, repeat=True)
html = HTML(anim.to_html5_video())
display(html)