[matplotlib animation] 64. 1次元ランダムウォークアニメーション

はじめに

matplotlibのFuncAnimationで1次元のランダムウォークアニメーションを表示する。

Wは動く回数であり、ここでは500とした。
stepはnp.random.choice([-1,1],W) により-1 か1を500回ランダムに選択した配列となる。
positionはnp.cumsum(step)により、累積和とすることで順次変化するランダムウォークの位置となる。

上記で作成したデータを表示すると以下のようになる。

p1,で空のプロットを作成して、データを流し込んでいく。
データのセットには、set_dataを用いる。
データの更新に伴ってx軸の表示範囲も変化させる。ax.set_xlim(0,x[i+1])とすることでx軸の表示範囲を徐々に大きくできる。

ani = animation.FuncAnimationとし、このaniをHTML(ani.to_html5_video())とすることでjupyter lab(notebookも)上でアニメーションを表示できる。