ArtistAnimationの出力がフレーム毎に分割されて出力される

Question

Axes3Dで作成したグラフをArtistAnimationで動かしたいのですが、現在のコードではうまくアニメーションが出力されません。
コードを実行すると、画像のような各フレームのグラフが100個のウィンドウとして個別に出力されてしまいます。

ソースコード

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

N=100
frames = []
for i in range(N):
    # (x,y)データを作成
    a1= np.loadtxt("4000rpm_wUV_6,52-heimen.csv", delimiter=",", usecols=(0) )
    a= a1[i]
    b1= np.loadtxt("4000rpm_wUV_6,52-heimen.csv", delimiter=",", usecols=(1) )
    b= b1[i]
    c1= np.loadtxt("4000rpm_wUV_6,52-heimen.csv", delimiter=",", usecols=(2) )
    c= c1[i]
    h1= np.loadtxt("4000rpm_wUV_6,52-heimen.csv", delimiter=",", usecols=(3) )
    h= h1[i]

    #x,yの範囲
    x = np.linspace(-45, 45, 10)
    y = np.linspace(-45, 45, 10)

    # 格子点を作成
    X, Y = np.meshgrid(x, y)
    # Zの作成
    Z = (h-a*X-b*Y)/c

    # 3Dでプロット
    fig = plt.figure()
    ax = Axes3D(fig)
    
    ax.set_xlim([-45,45])
    ax.set_ylim([-45,45])
    ax.set_zlim([-45,45])

    # 軸ラベル
    ax.set_xlabel("X軸: 中心からの距離[mm]", fontsize = 16, fontname="MS Gothic")
    ax.set_ylabel("Y軸: 中心からの距離[mm]", fontsize = 16, fontname="MS Gothic")
    ax.set_zlabel("Z軸: 変位[μm]", fontsize = 16, fontname="MS Gothic")
    
    frame = ax.plot_surface(X, Y, Z, color="tomato",antialiased=False, alpha=0.3)
    frames.append(frame)

ani = ArtistAnimation(fig, frames, interval=100)

# 描画
plt.show()

Comments

[jbpb0]

> # 3Dでプロット
fig = plt.figure()
ax = Axes3D(fig)

を「for i in range(N):」よりも前に移して、

frames.append(frame)
↓ 修正
frames.append([frame])

で、どうでしょうか？

[jbpb0]

質問内容とは関係ありませんが、下記の行は変数「i」と「Z」に関係しないので、「for i in range(N):」のループ内で毎回同じことをやってますよね
同じことを何回もやるのは時間のムダなので、それらは「for i in range(N):」よりも前に移した方がいいですよ

・「np.loadtxt」の行
・「#x,yの範囲」の下
・「# 格子点を作成」の下
・「ax.set_*lim」の行
・「# 軸ラベル」の下

[kbox]

コメントしていただきありがとうございます。
FuncAnimationを使用して目的のアニメーションが得られましたので、今後ArtistAnimationを使用する際の参考にさせていただきます。

Answer 1

FuncAnimationを使用したプログラムに変更したところ、無事動作しました。

ソースコード

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

fig = plt.figure(dpi=150)
ax = fig.add_subplot(projection='3d')

ax.set_xlim([-45,45])
ax.set_ylim([-45,45])
ax.set_zlim([-45,45])

# 軸ラベル
ax.set_xlabel("X軸: 中心からの距離[mm]", fontsize = 16, fontname="MS Gothic")
ax.set_ylabel("Y軸: 中心からの距離[mm]", fontsize = 16, fontname="MS Gothic")
ax.set_zlabel("Z軸: 変位[μm]", fontsize = 16, fontname="MS Gothic")

a1= np.loadtxt("gifyou.csv", delimiter=",", usecols=(0) )
a= a1[0]
b1= np.loadtxt("gifyou.csv", delimiter=",", usecols=(1) )
b= b1[0]
c1= np.loadtxt("gifyou.csv", delimiter=",", usecols=(2) )
c= c1[0]
h1= np.loadtxt("gifyou.csv", delimiter=",", usecols=(3) )
h= h1[0]

#x,y面の範囲
x = np.linspace(-45, 45, 10)
y = np.linspace(-45, 45, 10)

# 格子点を作成
X, Y = np.meshgrid(x, y)
# Zの作成
Z = (h-a*X-b*Y)/c

p1 = [ax.plot_surface(X, Y, Z, color="tomato",antialiased=False, alpha=0.3)]


def update(i):
    p1[0].remove()
    aa= a1[i]
    bb= b1[i]
    cc= c1[i]
    hh= h1[i]
    ZZ = (hh-aa*X-bb*Y)/cc
    p1[0]= ax.plot_surface(X, Y, ZZ, color="tomato",antialiased=False, alpha=0.3)


ani = animation.FuncAnimation(fig, update, interval=100, frames=100)

# 描画
ani.save('sample.gif', writer="pillow",dpi=100)
plt.show()