matplotlib contourマップの値の制限とanimation

Question

import numpy as np
import matplotlib.pyplot as plt

E=np.zeros((10,10))
for i in range(1,10,1):
    for j in range(1,10,1):
        E[i][j]=i/j

fig = plt.figure()
ax = fig.add_subplot(111)
cax = ax.imshow(E, interpolation='none')
fig.colorbar(cax)
plt.show()

コンターマップを連続して表示させ、波が減衰していく様子を作りたいのですがコンターマップ上限下限のせいげんの仕方がわかりません。
コンターマップの数字を変更して表示させたいです。この場合だとカラーバーのmaxを例えば―1から１５で表示させたいです。どうしたらいいでしょうか。

また、コンターマップでアニメーションを作るにはどうすればいいでしょうか。

Answer 1

こんな感じでしょうか。
質問に挙げておられますコードを少し改変してみました。

import numpy as np
import matplotlib.pyplot as plt
from matplotlib.animation import FuncAnimation

def data(count):
    E = np.zeros((10,10))
    for i in range(1,10,1):
        for j in range(1,10,1):
            E[i][j]=i/j + count*0.5
    return E


fig = plt.figure()
ax = fig.add_subplot(111)
cax = ax.imshow(data(0), interpolation='none', vmin=0, vmax=10, animated=True)
fig.colorbar(cax)

def animate(i):
    cax.set_array(data(i))
    return cax,

ani = FuncAnimation(fig, animate, frames=20, interval=100, blit=True)
plt.show()

変更点としては

• データ生成部を関数化 (引数countによりアニメーション対応)
• 上限・下限の指定は imshow() の引数にて vmin,vmax を指定
• アニメーション対応
  1. imshow() の引数に animated=True 追加
  2. animate()関数を追加しカウンター値でimageを書き換える処理を記述
  3. FuncAnimation()関数を呼び出し、animate関数を指定

です。

現状では frames=20 を指定しているのでカウンター値は 0～19 の範囲でアニメーションします。

Comments

[simasimaa]

非常に助かりました。
matplolibについて勉強するにはなにがいいんでしょうか。英語のチュートリアルでしょうか。

[magichan]

個人的には gallery
https://matplotlib.org/gallery.html
を眺めて、気になるものがあったらソースコード読む＆テストコード書く
くらいですかね。あとは必用に応じてドキュメント読むくらいです。

[simasimaa]

ありがとうございます。勉強します。