如何解析大型xml文件中的一些数据?

时间:2013-06-06 21:10:47

标签: python xml python-2.7 xml-parsing

我需要从大型xml文件中提取位置和半径数据,格式如下,并将数据存储在二维ndarray中。这是我第一次使用Python,但是找不到最好的方法。

<species name="MyHeterotrophEPS" header="family,genealogy,generation,birthday,biomass,inert,capsule,growthRate,volumeRate,locationX,locationY,locationZ,radius,totalRadius">
0,0,0,0.0,0.0,0.0,77.0645361927206,-0.1001871531330136,-0.0013358287084401814,4.523853439106942,234.14575280979898,123.92820420047076,0.0,0.6259920275663835;
0,0,0,0.0,0.0,0.0,108.5705297969604,-0.1411462759900182,-0.001881950346533576,1.0429122163754276,144.1066875513379,72.24884428367467,0.0,0.7017581019907897;
.
.
.
</species>

编辑:我的意思是人类标准的“大”。我没有任何内存问题。

3 个答案:

答案 0 :(得分:4)

您基本上拥有XML文本值中的CSV数据。

使用ElementTree解析XML,然后使用numpy.genfromtxt()将该文本加载到数组中:

from xml.etree import ElementTree as ET

tree = ET.parse('yourxmlfilename.xml')
species = tree.find(".//species[@name='MyHeterotrophEPS']")
names = species.attrib['header']
array = numpy.genfromtxt((line.rstrip(';') for line in species.text.splitlines()), 
    delimiter=',', names=names)

注意生成器表达式,str.splitlines()调用;这会将XML元素的文本转换为一系列行,.genfromtxt()非常乐意接收这些行。我们会从每行中删除尾随的;字符。

对于您的样本输入(减去.行),结果为:

array([ (0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 77.0645361927206, -0.1001871531330136, -0.0013358287084401814, 4.523853439106942, 234.14575280979898, 123.92820420047076, 0.0, 0.6259920275663835),
       (0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 108.5705297969604, -0.1411462759900182, -0.001881950346533576, 1.0429122163754276, 144.1066875513379, 72.24884428367467, 0.0, 0.7017581019907897)], 
      dtype=[('family', '<f8'), ('genealogy', '<f8'), ('generation', '<f8'), ('birthday', '<f8'), ('biomass', '<f8'), ('inert', '<f8'), ('capsule', '<f8'), ('growthRate', '<f8'), ('volumeRate', '<f8'), ('locationX', '<f8'), ('locationY', '<f8'), ('locationZ', '<f8'), ('radius', '<f8'), ('totalRadius', '<f8')])

答案 1 :(得分:2)

如果您的XML只是那个species节点,那么它非常简单,Martijn Pieters已经比我更好地解释了它。

但是如果你在文档中有大量的species个节点,并且它太大而无法将整个内容放入内存中,那么你可以使用iterparse代替parse

import numpy as np
import xml.etree.ElementTree as ET

for event, node in ET.iterparse('species.xml'):
    if node.tag == 'species':
        name = node.attr['name']
        names = node.attr['header']
        csvdata = (line.rstrip(';') for line in node.text.splitlines())
        array = np.genfromtxt(csvdata, delimiter=',', names=names)
        # do something with the array.

如果您只有一个超级巨大的species节点,这将无济于事,因为即使iterparse(或像SAX解析器这样的类似解决方案)也会一次解析整个节点。您需要找到一个允许您流式传输大型节点的文本的XML库,而且我不认为任何stdlib或流行的第三方解析器都可以那样做。

答案 2 :(得分:0)

如果文件非常大,请使用ElementTreeSAX

如果文件不是那么大(即适合内存),minidom可能更容易使用。

每一行似乎都是一个逗号分隔数字的简单字符串,因此你可以吝啬地做line.split(',')