Plot vtk 文件使用 Python

Question

我无法使用 python 显示 vtk 文件。 Spyder 代码分析提示我 vtkDataSetMapper 是一个未定义的名称。

我知道 vtk 文件是有序的，因为我已经使用 Paraview 显示了它。

我的 vtk 文件如下所示：

# vtk DataFile Version 2.0
 velocity field
 ASCII
 DATASET STRUCTURED_POINTS
 DIMENSIONS         108         103          31
 ORIGIN   0.0000000000000000        0.0000000000000000       -297.50000000000000     
 SPACING   15.465818554775332        12.565027060488859        10.000000000000000     
 POINT_DATA      344844
 SCALARS scalars float
 LOOKUP_TABLE default
 8.4405251
 8.4405251
...
...
...

在最后显示的行之后，vtk 文件包含 rest 的信息，它们只是数字（~ 300000 个值）

我的代码如下所示：

import vtk

# Read vtk file data 
reader = vtk.vtkDataSetReader()
reader.SetFileName("seaust.vtk")
reader.ReadAllScalarsOn()  # Activate the reading of all scalars
reader.ReadAllVectorsOn()  # Activate the reading of all vectors
reader.ReadAllTensorsOn()  # Activate the reading of all tensors
reader.Update()
data = reader.GetOutput()
scalar_range = data.GetScalarRange()

# Create the mapper that corresponds the objects of the vtk file
# into graphics elements
mapper = vtkDataSetMapper()
mapper.SetInput(data)

尝试编译代码时，python 提示我这个错误：

AttributeError: 'vtkRenderingCorePython.vtkDataSetMapper' object has no attribute 'SetInput'

我期待我的数据的 3D 可视化。

你能帮我得到它吗？

Answer 1

我猜你错过了vtk。 在mapper = vtk.vtkDataSetMapper()

你可能需要使用mapper.SetInputData(data)