matplotlibのplt.draw()を使ってリアルタイムで画像の描画をしたいのですがディスプレイに何も表示されません

matplotlibで白黒反転するアニメーションを作りたいのですが、下のプログラムを実行してもディスプレイに何も表示されません
下のプログラムをどのように変えたら良いのでしょうか？教えてくださると助かります。

python
1import numpy as np
2import matplotlib.animation as animation
3import matplotlib.pyplot as plt
4
5if __name__ == "__main__":
6    img = plt.imshow(np.zeros((8, 8)), interpolation="none", cmap="gray")
7
8    for e in range(10000):
9        if e % 2 == 0:
10            img.set_data(np.zeros((8, 8)))
11        else:
12            img.set_data(np.zeros((8, 8))+1)
13        plt.draw()
14        
15

行動規範の内容に同意します

回答1件

ベストアンサー

やりたい事はこんな感じでしょうか？

Python
1import numpy as np
2import matplotlib.animation as animation
3import matplotlib.pyplot as plt
4
5
6if __name__ == "__main__":
7    fig = plt.figure()
8    imgs = []
9    for e in range(100):
10        if e % 2 == 0:
11            im = plt.imshow(np.zeros((8, 8)), interpolation="none", cmap="gray", vmin=0, vmax=1)
12        else:
13            im = plt.imshow(np.zeros((8, 8))+1, interpolation="none", cmap="gray", vmin=0, vmax=1)
14        imgs.append([im])
15
16    ani = animation.ArtistAnimation(fig, imgs, interval=100, blit=True)
17    plt.show()