animationがうまくいかないです。

Question

animationとして、以下のコード結果を動画で保存したいのですが、以下のようなエラーが出てしまいました。
いくつかネットで調べたのですが、おそらく
def plot(data):
plt cla()
のやり方の方が今回は良いかもしれませんが、以下は違う書き方を自分がしていると思います。

import numpy as np
import matplotlib.pyplot as plt
import math
#from mpl_toolkits.mplot3d import Axes3D
import matplotlib.animation as animation

fig = plt.figure()
ims=[]

#環境設定
n=100
m=100
theta = 3*(math.pi)/12

P = [[np.cos(theta),np.sin(theta)],[0,0]]
Q = [[0,0],[np.sin(theta),-np.cos(theta)]]
x_list=[]#xline
t_list=[]#time
p_list=[]#probability
s_list=[]#state
a = 1/math.sqrt(2)
b = 1j/math.sqrt(2)

for x in range(0,2*m+1):
    if x == m:
        phi = [a ,b]
    else:
        phi = [0,0]
    p = np.dot(phi,np.conj(phi))

    x_list.append(x)
    s_list.append(phi)
    p_list.append(p)


for t in range(0,n+1):
    t_list.append(t)
    if t ==0:
        s_list
        p_list
    else:
        next_s_list = [0]*len(s_list)  #s_listと同じ要素の数ですべて0を用意（初期化）
        for x in range(0,2*m+1):
            if x == 0:
                next_s_list[0] = np.inner(P, s_list[1])

            elif x == 2*n:
                next_s_list[2*m] = np.inner(Q, s_list[2*m-1])

            else:
                next_s_list[x] = np.inner(P, s_list[x+1]) + np.inner(Q, s_list[x-1])
            p_list[x] = np.dot(next_s_list[x],np.conj(next_s_list[x]))
        s_list = next_s_list


    #確率の挙動をみたいなら
    print(t,s_list)
    plt.xlabel("x")
    plt.ylabel("probability")
    plt.ylim([0,0.1])
    plt.xlim([-m,3*m])
    line = plt.plot(x_list,np.real(p_list),color="red",linewidth=0.3)
    ims.append([line])

ani = animation.ArtistAnimation(fig,ims, interval=100)
ani.save("quantumwalk.gif")

ちなみに、コード自体は計算されており、エラーが以下のように出ました。

ValueError: outfile must be *.htm or *.html

結構、調べたんですけど、力足らずで申し訳ないです。。
ご指導の程、宜しくお願いします。。

Answer 1

動画の生成に必要なMovieWriterが存在しないためだと思われます。
環境にもよりますがmatplotlib のanimation を保存などに記載されているように別途ツールのインストールと設定が必要なようです。

詳細未未確認ですが.to_jshtmlにてインタラクティブなHTML形式で出力する手法だとMovieWrite不要なようです。

ちなみに当方Win10+Anaconda環境での検証結果は以下のとおりとなりました。

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

# 単純なアニメーション
fig = plt.figure()
x = np.arange(0, 10, 0.1)
ims = []
for a in range(50):
    y = np.sin(x - a)
    line, = plt.plot(x, y, "r")
    ims.append([line])
ani = animation.ArtistAnimation(fig, ims)

ani.save('anim.gif')
# MovieWriter ffmpeg unavailable. と表示されるがOK

# ani.save('anim.gif', writer="imagemagick")
# ani.save('anim.mp4')
# ani.save('anim.mp4', writer="ffmpeg")
#
# 以下のようなエラーにてNG
#
# MovieWriter imagemagick(またはffmgpeg)  unavailable.
# Traceback (most recent call last):
# 略
# ValueError: outfile must be *.htm or *.html
#

#ani.save('anim.htm')
#
# 以下のようなエラーにてNG
#
# MovieWriter ffmgpeg unavailable.
# Traceback (most recent call last):
# 略
# AttributeError: module 'PIL._webp' has no attribute 'HAVE_WEBPANIM'

# OK
s = ani.to_jshtml()
with open( 'anim.html', 'w') as f:
    f.write(s)

Comments

[Fallout_18]

ありがとうございます！！！
確認してみます