Mayavi的時間相關數據

Question

假設我有一個像這樣的4d numpy數組：my_array[x,y,z,t]。

有沒有簡單的方法將整個數組加載到Mayavi中，並簡單地選擇我想調查的t？

我知道可以對數據進行動畫製作，但我想旋轉我的數字「在旅途中」。

Answer 1

可以在輸入框中設置對話框，您可以在其中設置t。 您必須使用traits.api，它看起來是這樣的：

from traits.api import HasTraits, Int, Instance, on_trait_change 
from traitsui.api import View, Item, Group 
from mayavi.core.ui.api import SceneEditor, MlabSceneModel, MayaviScene 

class Data_plot(HasTraits): 
    a = my_array 
    t = Int(0) 
    scene = Instance(MlabSceneModel,()) 
    plot = Instance(PipelineBase) 

    @on_trait_change('scene.activated') 
    def show_plot(self): 
    self.plot = something(self.a[self.t]) #something can be surf or mesh or other 

    @on_trait_change('t') 
    def update_plot(self): 
    self.plot.parent.parent.scalar_data = self.a[self.t] #Change the data 

    view = View(Item('scene', editor=SceneEditor(scene_class=MayaviScene), 
       show_label=False), 
       Group('_', 't'), 
       resizable=True, 
       ) 

my_plot = Data_plot(a=my_array) 
my_plot.configure_traits() 

您還可以設置滑塊與命令Range，而不是Int如果你喜歡這一點。