python - 用 Python 读取 vtu 文件

Question

我想用非结构化的 vtu 数据制作 npy 矩阵，下面是它的代码。

file_name = 'data/0_0000.vtu'
reader = vtk.vtkXMLUnstructuredGridReader()
reader.SetFileName(file_name)
reader.Update()  

polydata = reader.GetOutput()
points = polydata.GetPoints()
array1 = vtk_to_numpy(points.GetData())

cells = polydata.GetCells()
array2 = vtk_to_numpy(cells.GetData())
velocity = vtk_to_numpy(polydata.GetCellData().GetArray("velocity"))

但是在我运行这段代码之后，它只是部分转换了数据，而不是完全转换。vtu 文件包含超过 30000 个单元，但不知何故只有 4000 个单元被转换。

谁能帮我解决这个问题？

谢谢！

标签： pythonvtk