如何使用VTK和Python在时间步长中生成字段的快照

import vtk
import numpy as np

data = vtk.vtkImageData()

dimensions = (100, 100, 100)  # 数据的维度
data.SetDimensions(dimensions)

field_array = vtk.vtkFloatArray()
field_array.SetNumberOfComponents(1)  # 设置数据的组件数
field_array.SetNumberOfTuples(dimensions[0] * dimensions[1] * dimensions[2])  # 设置数据的元素数

for t in range(num_time_steps):
    # 生成数据，可以根据具体需求进行操作
    field_data = np.random.rand(dimensions[0] * dimensions[1] * dimensions[2])
    
    # 将数据存储到字段数组中
    for i in range(dimensions[0] * dimensions[1] * dimensions[2]):
        field_array.SetValue(i, field_data[i])

data.GetPointData().SetScalars(field_array)

renderer = vtk.vtkRenderer()
render_window = vtk.vtkRenderWindow()
render_window.AddRenderer(renderer)

mapper = vtk.vtkDataSetMapper()
mapper.SetInputData(data)
actor = vtk.vtkActor()
actor.SetMapper(mapper)
renderer.AddActor(actor)

interactor = vtk.vtkRenderWindowInteractor()
interactor.SetRenderWindow(render_window)
interactor.Start()