xml.etree.ElementTree模块为xml文件的提取和建立提供了简单有效的API。下文中使用ET来代表xml.etree.ElementTree模块。
XML是一种内在的分层的数据形式,展示这种数据最自然的方式就是树形结构。ET有两种类来实现这个目标——Elementree将整个XML文件构造成一棵树,Element代表书树一个节点。与整个文件的交互(从文件读或者写)一般在Elementtree层次完成,与一个简单的XML element以及它的sub-element交互一般在Element层次完成。
- 解析XML
用下面的XML文件来完成后面的代码中对xml的提取
<?xml version = "1.0" ?> <bounding_boxes> <object id = "00"> <bndbox> <xmin>76.7371</xmin> <ymin>112.0698</ymin> <xmax>93.5142</xmax> <ymax>126.666</ymax> </bndbox> <name>ship</name> </object> <object id = "01"> <bndbox> <xmin>51.4033</xmin> <ymin>179.954</ymin> <xmax>62.1409</xmax> <ymax>178.3398</ymax> </bndbox> <name>ship</name> </object> </bounding_boxes>
可以通过读取一个文件来导入数据:
import xml.etree.ElementTree as ET tree = ET.pares("Annotations0001.xml") root = tree.getroot()
也可以直接从一个字符串读入:
root = ET.fromstring(Annotations)
fromstring()直接将一个字符串从xml解析成树根的Element。
作为一个Elment,root有一个tag和一个字典属性:
>>>root.tag 'bounding_boxes' >>>root.attrib {}
它也有可以迭代的孩子节点:
for child in root: print(child.tag,child.attrib) object {'id': '00'} object {'id': '01'}
孩子是可以嵌套的,因此我们可以通过索引来访问特定的孩子:
>>>root[0][1].text 'ship'
注意:并不是所有的XML输入的element都是以解析树的element结束。现在,这个模块跳过了输入中所有的XML注释、处理指令和文件类型说明。然而,通过这个模块的API构造而不是通过解析XML文件构造的树可以包括注释和处理指令,这些会在生成XML输出中体现出来。一个文件类型的声明可以通过传递一个TreeBuilder实例给XMLParser构造函数来实现。
- 非阻塞解析
在ET模块中的大多数解析函数都要求在返回结果之前一次性读入整个文件。使用XMLParser并且逐渐的填入数据是可能的,但是它是一个调用回调目标的push API,对于大多数需求这样很不方便并且层次很低。有时候在享受构造整个Element目标方便的同时,使用者只是想逐渐解析XML而不要阻塞操作。
完成这个功能最强大的工具就是XMLPullParser。它不要求阻塞式读入所有数据,而是通过调用XMLPullParser.feed()函数来逐渐填入数据。为了的到解析的XML element,可以调用XMLPullParser.read_events()。下面是例子:
>>>parser = ET.XMLPullParser(['start','end']) >>>parser.feed("<book>python") >>>list(parser.read_events()) [('start', <Element 'book' at 0x032FA6F0>)] >>>parser.feed(" tutorial</book>") >>>for event,elem in parser.read_events(): ... print(event) ... print(elem.tag,"text=",elem.text) ... end book text= python tutorial
- 找到感兴趣的element
Element有一些有用的方法来帮助用户递归迭代该节点的子树:
>>>for obj in root.iter('name'): ... print(obj.text) ... ship ship
Element.findall()可以找到所有该Element下所有的直接孩子。Element.find()特定标签下的第一个孩子,Element.text是指该element的文本。Element.get()的事element的属性。
for obj in root.findall("object"): name = obj.find('name').text print(name,obj.get("id")) ship 00 ship 01
- 修改一个XML文件
ElemenrtTree提供了一个简单的构造XML文件并写入的方法。ElementTree.write()方法完成这个功能。
一旦一个Element object被创建,就可以直接修改它的域(比如Element.text),添加和修改属性(Element.set()方法),也可以添加孩子结点(Element.append())
>>>obj = root.find("object") >>>obj.set("updated","yes") >>>obj.attrib {'id': '00', 'updated': 'yes'}
也可以通过使用函数Element.remove()来移除element
>>>name = obj.find("name") >>>obj.remove(name) >>>tree.write("Annotations0001.xml")
移除之后XML文件变成如下这个样子:
bounding_boxes> <object id="00" updated="yes"> <bndbox> <xmin>76.7371</xmin> <ymin>112.0698</ymin> <xmax>93.5142</xmax> <ymax>126.666</ymax> </bndbox> </object> <object id="01"> <bndbox> <xmin>51.4033</xmin> <ymin>179.954</ymin> <xmax>62.1409</xmax> <ymax>178.3398</ymax> </bndbox> <name>ship</name> </object> </bounding_boxes>
- 创建XML文件
SubElement()函数提供了一个创建给定element的sub-element的简单方法:
>>>root = ET.Element("root") >>>child1 = ET.SubElement(root,"child1") >>>child2 = ET.SubElement(root,"child2") >>>grandson = ET.SubElement(child2,"grandson") >>>ET.dump(root) <root><child1 /><child2><grandson /></child2></root>
参考文件:https://docs.python.org/3.5/library/xml.etree.elementtree.html