从原始 VTK (.vtu) 文件中读取数据

Question

我想使用 Python VTK 模块从 .vtu 文件中提取数据数组。该文件如下所示（省略了文件末尾的原始数据）：

<?xml version="1.0"?>
<VTKFile type="UnstructuredGrid" version="0.1" byte_order="LittleEndian">
  <UnstructuredGrid>
    <Piece NumberOfPoints="10471" NumberOfCells="64892">
      <PointData>
        <DataArray type="Float64" Name="potential" NumberOfComponents="1" format="appended" offset="0"/>
        <DataArray type="Float64" Name="electric field" NumberOfComponents="3" format="appended" offset="83772"/>
      </PointData>
      <CellData>
        <DataArray type="Int32" Name="GeometryIds" format="appended" offset="335080"/>
      </CellData>
      <Points>
        <DataArray type="Float64" NumberOfComponents="3" format="appended" offset="594652"/>
      </Points>
      <Cells>
        <DataArray type="Int32" Name="connectivity" format="appended" offset="845960"/>
        <DataArray type="Int32" Name="offsets" format="appended" offset="1865068"/>
        <DataArray type="Int32" Name="types" format="appended" offset="2124640"/>
      </Cells>
    </Piece>
  </UnstructuredGrid>
<AppendedData encoding="raw">

我尝试使用以下 python 代码提取数据：

import numpy
from vtk import vtkUnstructuredGridReader
from vtk.util import numpy_support as VN

reader = vtkUnstructuredGridReader()
reader.SetFileName("charged_wire.vtu")
reader.ReadAllVectorsOn()
reader.ReadAllScalarsOn()
reader.Update()

data = reader.GetOutput()
potential = data.GetPointData().GetScalars("potential")

print(type(potential))

不幸的是，这个程序打印 NoneType 作为输出，我不确定我需要更改什么才能提取 potential 数组中的数据存储？

Answer 1

你没有使用正确的reader，这是一个.vtu文件，你必须使用vtkXMLUnstructuredGridReader.

import vtk.vtk

# The source file
file_name = "path/to/your/file.vtu"

# Read the source file.
reader = vtk.vtkXMLUnstructuredGridReader()
reader.SetFileName(file_name)
reader.Update()  # Needed because of GetScalarRange
output = reader.GetOutput()
potential = output.GetPointData().GetArray("potential")

Answer 2

一个更轻量级的解决方案是使用 meshio（我编写的）。除了 numpy 之外，它没有任何必需的依赖项。安装

pip install meshio

并使用

读取文件

import meshio

mesh = meshio.read("foo.vtu")
# mesh.points, mesh.cells, mesh.point_data, ...

从原始 VTK (.vtu) 文件中读取数据

Reading data from a raw VTK (.vtu) file

numpy

vtk

python-3.x