如何動畫matplotlib中的3d plot_surface

我已經從一個文件創建了一個3D繪圖曲面，並且我正在試圖爲這個繪圖設置動畫。我已經閱讀了matplotlib網頁中的示例和其他示例，並注意到我需要創建一個更新函數來遍歷文件中的值，然後創建一個對象，但我不明白該怎麼做。如何動畫matplotlib中的3d plot_surface

如果有人能向我解釋更新函數的語法以及如何在matplotlib.animation對象中使用它，我將非常感激。

我的數據是一個具有498行的多維數組，每行我有一個64x128值的數組。數據的組織方式如下：

數據是來自力盤的時間序列，500行中的每一行都是一幀，這意味着該試驗持續10秒。對於每一幀我有另一個64x128值的數組。

這是我的代碼至今：

from mpl_toolkits.mplot3d import * 
import matplotlib.pyplot as plt 
import numpy as np 
from random import random, seed 
from matplotlib import cm 
from mpl_toolkits.mplot3d import axes3d 
import matplotlib.animation as animation 

source_path = "c:\\Projecto\\" 
destination_path = "c:\\Projecto\\EntirePlate\\" 
#fid = np.loadtxt(source_path + "rolloff_xls.txt",dtype=str) 

fid_MP = open(source_path + "101mp - Entire plate roll off.xls","Ur") 
lines_MP = fid_MP.readlines() 
fid_MP.close() 

values_MP = [] 

for i in lines_MP: 
     if i[0].isdigit(): 
      values_MP.append(i) 

values = np.loadtxt(values_MP,dtype=float) 

new_values_MP =[] 

for i in range(0,(len(values_MP)/64)): 
    for j in range(0,64): 
     new_values_MP.append([[i],[j],values[j]]) 

new_values_MP = np.asarray(new_values_MP) 

fig = plt.figure() 
ax = fig.gca(projection='3d')    # to work in 3d 
plt.hold(True) 

x_surf = np.arange(0,128)    # generate a mesh 
y_surf = np.arange(0,64) 
x_surf, y_surf = np.meshgrid(x_surf, y_surf) 
z_surf = [] 

for i in range(0,64): 
    # print(new_values[i]) 
    z_surf.append(np.asarray(new_values_MP[i][2])) # ex. function, which depends on x and y 

z_surf = np.asarray(z_surf).reshape([64,128]) 

ax.plot_surface(x_surf, y_surf, z_surf, rstride=2, cstride=2 ,cmap=cm.jet) # plot a 3d surface plot 

ax.set_xlabel('Medio Lateral - Axis') 
ax.set_ylabel('Anterior Posterior - Axis') 
ax.set_zlabel('Pressure (P)') 

def update(x_values, y_values, z_values): 
    for i in range(0,len(values_MP)/64): 
     x_surf = x_values 
     y_surf = y_values 
     z_surf.set_data(new_values_MP[i,2]) 
    return z_surf 

ani = animation.FuncAnimation(fig, update, frames=xrange(len(values_MP)/64), 
           interval=50, blit=False) 
plt.show()

來源

2013-06-25 TMoover

這可能不是最佳的方式，但我發現的文件/例子不夠了。

我採取的是以下解決方案：使用animation.FuncAnimation來調用一個函數。在該功能中清除並重繪，如下所示：

from __future__ import division 
from matplotlib import cm 
from mpl_toolkits.mplot3d import Axes3D 
import matplotlib.pyplot as plt 
import mpl_toolkits.mplot3d.axes3d as p3 
import matplotlib.animation as animation 
import numpy as np 

plot_args = {'rstride': 1, 'cstride': 1, 'cmap': 
      cm.bwr, 'linewidth': 0.01, 'antialiased': True, 'color': 'w', 
      'shade': True} 

soln = np.zeros((size, size)) 
midpoint = size // 2 
soln[midpoint, midpoint] = 1 

#first frame 
X = range(size) 
Y = range(size) 
X, Y = np.meshgrid(X, Y) 
plot = ax.plot_surface(X, Y, soln, **plot_args) 
pam_ani = animation.FuncAnimation(fig, data_gen, fargs=(soln, plot), 
           interval=30, blit=False) 

def data_gen(framenumber, soln, plot): 
    #change soln variable for the next frame 
    ... 
    ax.clear() 
    plot = ax.plot_surface(X, Y, soln, **plot_args) 
    return plot,

來源

2013-10-13 19:15:24

如何動畫matplotlib中的3d plot_surface

回答

相關問題