matplotlibでアニメーション中に値によってプロットの色を変える

Question

FuncAnimation()を用いてアニメーションを作成しています。
FuncAnimation()が毎回呼び出す関数の中でset_data()を使って描画を行っているのですが、この線の色を、プロットする値に応じてグラデーションのようにリアルタイムで変更したいのですが、どなたかご教授お願いできないでしょうか。
宜しくお願いします。

Answer 1

こんな感じでしょうかね

import matplotlib.animation as anm
import matplotlib.pyplot as plt
import matplotlib.cm as cm
import numpy as np

fig = plt.figure()

def update(i):
    if i != 0:
        plt.cla()
    z = np.sin(i * 2 * np.pi / 100)
    plt.scatter(0, 0, c=z, vmin=-1, vmax=1, cmap=cm.bwr)

ani = anm.FuncAnimation(fig, update, frames=100, interval=50)
plt.show()

---

【追記】

set_color() を使う例

import matplotlib.animation as anm
import matplotlib.pyplot as plt
import matplotlib.cm as cm
import numpy as np

fig = plt.figure()
ax = fig.add_subplot(1,1,1)
line, = ax.plot([], [])
ax.set_ylim(-1, 1)

def update(i, ax, line):
    x = np.arange(i)
    y = np.sin(x * 2 * np.pi / 100)
    line.set_data(x, y)
    line.set_color(cm.bwr(plt.Normalize(vmin=0, vmax=100)(i)))
    ax.set_xlim(0, i)

ani = anm.FuncAnimation(fig, update, fargs=(ax, line), frames=100, interval=50)
plt.show()

Comments

[KriegReisfeld]

プロットの値自体はset_data()で更新できるようにしたいのです。
fig = plt.figure(figsize=(5, 5))
plt.axes([0.1, 0.1, 0.1, 0.1])
line, = plt.plot(0,0)
などとしておき、FuncAnimationのupdate関数の中でset_data()とlineの色を変更する操作ができればいいのですが......

[magichan]

であれば set_color() を使うとよいのではないでしょうか

[magichan]

set_color() を使うサンプルを追記しました

[KriegReisfeld]

まさにこのような関数を探していました。
本当にありがとうございます。