从原始VTK (.vtu)文件读取数据
从原始VTK (.vtu)文件读取数据
提问于 2019-01-05 03:32:15
我想使用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数组中的数据存储需要做什么更改？
回答 2
Stack Overflow用户
回答已采纳
发布于 2019-01-07 18:51:14
您使用的读卡器不正确，这是一个.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")Stack Overflow用户
发布于 2020-02-22 07:18:51
一种更轻量级的解决方案是使用meshio (这是我编写的)。除了numpy之外，它没有任何必需的依赖项。安装时使用
pip install meshio并使用以下命令读取文件：
import meshio
mesh = meshio.read("foo.vtu")
# mesh.points, mesh.cells, mesh.point_data, ...页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接：
https://stackoverflow.com/questions/54044958
复制相关文章
相似问题
腾讯云开发者
