matplotlibを使ったアニメーションでグラフの間を塗りつぶす方法

Question

基本的な質問になってしまい申し訳ありません。
python3とipython notebookを使っています。
例えば、matplotlibで下のようなsin波のアニメーションを作成したときに、
sin波とx軸の間を塗りつぶしたい場合はどのようにすればよろしいのでしょうか。

import numpy as np
import matplotlib.pyplot as plt
import matplotlib.animation as animation
%matplotlib nbagg

fig = plt.figure()
x = np.arange(0, 10, 0.1)

ims = []
for a in range(10):
    y = np.sin(x - a)
    line, = plt.plot(x, y,"r")
    ims.append([line])

ani = animation.ArtistAnimation(fig, ims)
ani.save('sin.gif', writer="imagemagick")

目標としては下の画像がアニメーションとして流れるようにしたいです。

よろしくお願いします。

Answer 1

import numpy as np
import matplotlib.pyplot as plt
import matplotlib.animation as animation
#%matplotlib nbagg

fig = plt.figure()
x = np.arange(0, 10, 0.1)
zeros = np.zeros(x.shape)
ims = []
for a in range(10):
    y = np.sin(x - a)
    #line, = plt.plot(x, y,"r")
    line = plt.fill_between(x,y,facecolor="r")
    ims.append([line])

ani = animation.ArtistAnimation(fig, ims)
ani.save('sin.gif', writer="imagemagick")
#ani.save('sin.html', writer="imagemagick")
plt.show()

上記のコードでアニメーションとして動きました．
ただ保存時，拡張子が.gifだとエラーが出たため，コメントアウトしている拡張子が.htmlの方で保存しました．

Answer 2

この辺が役に立つでしょうか。

matplotlibで領域の図示 - 気ままに宙をみる

アニメは確認するのが手間なので、静止画ですが、こんな感じ。質問文のコードで、forの中をこんな感じにしていけば良さそうです。

import numpy as np
import matplotlib.pyplot as plt

x = np.arange(0,10,0.01)
y = np.sin(x)
zeros = np.zeros(x.shape)
plt.plot(x, y, c="r")  # 線のぶんだけ微妙に見た目が変わる。なくても基本的には出るが、色を変えてエッジを強調したりするのには使える
plt.fill_between(x,y,zeros,facecolor="r")
plt.show()