python的XML模块

构建XML文件

介绍

xml即可扩展标记语言，它可以用来标记数据、定义数据类型，是一种允许用户对自己的标记语言进行定义的源语言。从结构上，很像HTML超文本标记语言。但他们被设计的目的是不同的，超文本标记语言被设计用来显示数据，其焦<br>点是数据的外观。它被设计用来传输和存储数据，其焦点是数据的内容。<br>xml是实现不同语言或程序之间进行数据交换的协议，跟json差不多，但json使用起来更简单。

主要学习的ElementTree是python的XML处理模块，它提供了一个轻量级的对象模型。在使用ElementTree模块时，需要import xml.etree.ElementTree的操作。ElementTree表示整个XML节点树，而Element表示节点数<br>中的一个单独的节点。

XML文件

<data>
    <country name="Liechtenstein">
        <rank updated="yes">2</rank>
        <year>2008</year>
        <gdppc>141100</gdppc>
        <neighbor name="Austria" direction="E"/>
        <neighbor name="Switzerland" direction="W"/>
    </country>
</data>

构建XML文件

注意：我们现在就是按照上面的文件格式构建

import xml.etree.ElementTree as ET

new_xml = ET.Element("data")  # 设置根节点，标签为data
country = ET.SubElement(new_xml, "country", attrib={"name": "Liechtenstein"})  # 在根节点建立子节点
rank = ET.SubElement(country, "rank", attrib={"updated": "yes"})
rank.text = "2"
year = ET.SubElement(country, "year")
year.text = "2008"
gdppc = ET.SubElement(country, "gdpcc")
gdppc.text = "141100"
neighbor1 = ET.SubElement(country, "neighbor", attrib={"name": "Austria", "direction": "E"})
neighbor2 = ET.SubElement(country, "neighbor", attrib={"name": "Switzerland", "direction": "W"})

et = ET.ElementTree(new_xml)  # 生成文档对象
et.write("test.xml")

运行结果

解析XML文件

遍历XML文件

import xml.etree.ElementTree as ET
 
tree=ET.parse("test.xml")#将xml数据转为对象
root=tree.getroot() #获取根节点
print(root.tag) #打印标签名
for child in root: #遍历根节点的子标签，有一个
   print(child.tag,child.attrib)
   for i in child: #遍历子标签下的子标签，有五个
    print(i.tag,i.text)

运行结果

data
country {'name': 'Liechtenstein'}
rank 2
year 2008
gdpcc 141100
neighbor None
neighbor None

只遍历指定的节点

注意：如果你想要遍历所有的，那么你要从根节点开始遍历。

import xml.etree.ElementTree as ET

tree=ET.parse("test.xml")#将xml数据转为对象
root=tree.getroot() #获取根节点
for node in root.iter('year'):
  print(node.tag,node.text)

运行结果

1	`year 2008`

修改XML文件

注意：修改完必须存储才是真的修改。

修改操作

import xml.etree.ElementTree as ET
 
tree=ET.parse("test.xml")#将xml数据转为对象
root=tree.getroot() #获取根节点
for node in root.iter('year'):
    new_year = int(node.text) + 1
    # 修改属性值
    node.text = str(new_year)
    # 修改属性名
    node.set("updated", "yes")
 
# 存储
tree.write("xmltest.xml")

运行结果

删除操作

注意：这里先事先修改rank标签的值为51

import xml.etree.ElementTree as ET
 
tree=ET.parse("test.xml")#将xml数据转为对象
root=tree.getroot() #获取根节点
for country in root.findall('country'): #先找到country标签,findall()可以找多个
    rank = int(country.find('rank').text) #在country标签下找rank标签,find()可以找一个
    if rank > 50:
        root.remove(country)
 
tree.write('test.xml')