使用python中的vtk库提取vtk数据作为数组

Question

我希望使用python2.7中的vtk库从vtk非结构化网格文件中提取数据，并将这些数据转换为numpy或pythonlist格式。vtk文件的结构如下：

ASCII 
DATASET UNSTRUCTURED_GRID 
POINTS       96 float 
CELLS       96     192 
CELL_TYPES       96 
POINT_DATA       96 
VECTORS vector1 float 
VECTORS vector2 float 
etc 

由于vtk图书馆文档似乎没有对其功能进行公正的处理，所以我自己做了一些改进：

import vtk
Filename = 'test.vtk'
reader = vtk.vtkUnstructuredGridReader()
reader.SetFileName(FileName)
reader.Update()

然后呢？对dir变量执行reader显示了许多不同的方法/属性，例如GetNumberOfVectorsInFile获取ASCII文件中包含的向量层数，GetVectorsNameInFile(5)输出文件中包含的第五个向量信息数组的名称。如何获得文件中实际值的列表？

Answer 1

Python中VTK函数的名称类似于C++函数。所以您可以使用用于C++的vtkUnstructuredGridReader文档中给出的函数

import vtk
Filename = 'test.vtk'
reader = vtk.vtkUnstructuredGridReader()
reader.SetFileName(FileName)
reader.ReadAllScalarsOn()
reader.ReadAllVectorsOn()
reader.Update()
usg = reader.GetOutput()

vec1 = usg.GetPointData().GetVector('vector1')