Question

我将“vtu”格式文件（用于paraview）作为文本。格式如下：

<?xml version="1.0"?>
<VTKFile type="UnstructuredGrid" version="0.1" byte_order="LittleEndian" >
<UnstructuredGrid>
<Piece NumberOfPoints="21" NumberOfCells="20" >
<Points>
<DataArray type="Float64" Name="coordinates" NumberOfComponents="3" format="ascii" >
                  -3.3333333333e-01 1.1111111111e-01 0.0000000000e+00
 -2.7777777778e-01 1.1111111111e-01 0.0000000000e+00
 -1.1111111111e-01 4.4444444445e-01 0.0000000000e+00
</DataArray>
</Points>
<Cells>
<DataArray type="UInt64" Name="connectivity" NumberOfComponents="1" format="ascii" >
       0 1
 2 3
 5 4

它代表一个网格文件。我想提取NumberOfPoints的值以及前两个坐标并将它们存储在另一个文件中，如下所示：

21
-3.3333333333e-01
1.1111111111e-01
-2.7777777778e-01
1.1111111111e-01
-1.1111111111e-01
4.4444444445e-01

我不熟悉python，我只能逐行读取文件，但我不知道构造上面的文件。

到目前为止我学到的很简单。对于第一个文件，我能够检测到

包含的NumberOfPoints行

import xml.etree.ElementTree as ET
tree = ET.parse('read.vtu')
root = tree.getroot()

for Piece in root.iter('Piece'):
  print Piece.attrib
  nr = Piece.get('NumberOfPoints')
  print nr

我可以拥有21 :)下一步是添加坐标。但我不知道如何解析它们，因为我找不到任何连接到它们的节点。

Answer 1

试试这个：

import xml.etree.ElementTree as ET
try:
    from cStringIO import StringIO
except:
    from StringIO import StringIO

o = file('out.txt', 'w')

tree = ET.parse('read.vtu')
root = tree.getroot()

for Piece in root.iter('Piece'):
  nr = Piece.get('NumberOfPoints')
  o.write(nr+ '\n')

piece = root.iter('Piece') 
piece = piece.next()
point = piece.getchildren()[0]
dataArr = point.getchildren()
data = dataArr[0]

# Writing to a buffer
output = StringIO()
output.write(data.text)

# Retrieve the value written
crds = output.seek(1)

for l in output:
  ls = l.split( );
  o.write(ls[0]+ '\n')
  o.write(ls[1]+ '\n')

output.close()
o.close()

Answer 2

meshio（我的一个项目）知道VTU格式，所以你可以简单地

pip install meshio

然后

import meshio
points, cells, _, _, _ = meshio.read('file.vtu')

提取值并构造新文件

2 个答案: