python操作XML

XML是实现不同语言或程序之间进行数据交换的协议，XML文件格式如下：

<data>
    <country name="Liechtenstein">
        <rank updated="yes">2</rank>
        <year age="25">2033</year>
        <gdppc>141100</gdppc>
        <neighbor direction="E" name="Austria" />
        <neighbor direction="W" name="Switzerland" />
    </country>
    <country name="Singapore">
        <rank updated="yes">5</rank>
        <year age="25">2036</year>
        <gdppc>59900</gdppc>
        <neighbor direction="N" name="Malaysia" />
    </country>
    <country name="Panama">
        <rank updated="yes">69</rank>
        <year age="25" type="date">2036</year>
        <gdppc>13600</gdppc>
        <neighbor direction="W" name="Costa Rica" />
        <neighbor direction="E" name="Colombia" />
    </country>
</data>

我们创建操作XMl对象有2种创建方法 如下

# -*- coding: utf-8 -*-

from xml.etree import  ElementTree as ET

#打开文件 读取XML内容   第一种创建方法
str_xml= open('data.xml','r').read()
#将字符串解析成xml特殊对象，root代指xml文件的根节点
root = ET.XML(str_xml)  #data
print(root)   #返回一个对象
#输出 <Element 'data' at 0x1006feb50>


# 第二种创建方法
# 直接解析xml文件
tree = ET.parse('data.xml')
#获取xml文件的根节点
root = tree.getroot()
print(root) 
#输出 <Element 'data' at 0x101821650>

我们可以查看data.xml所有的节点

#便利所有的子节点
for child in root:
    #第二层
    print(child.tag,child.attrib)  #输出一个元组
    #对第二层节点下面的子节点进行便利
    for i in child:
        print(i.tag,i.attrib)

获取指定的节点

#获取xml中的制定节点
for child in root.iter('year'):
    print(child.tag,child.attrib)

对XML中的操作一般是在内存中进行的，不会影响到文件中的内容，因此我们在内存中写完之后需要重新写入文件

#对xml文件进行操作 增删改查
for node in root.iter('year'):
    # 将year内容+10
    new_year = int(node.text)+10
    node.text = str(new_year)

    # 设置属性
    node.set('name','eric')
    node.set('age','25')

    #删除属性
    del node.attrib['age']

#保存到文件
tree = ET.ElementTree(root)
tree.write('data.xml',encoding='utf-8')


#删除节点
for country in root.findall('country'):
   # 获取每一个country节点下rank节点的内容
   rank =  int(country.find('rank').text)
   if rank > 50:
       root.remove(country)  #会删除掉  整个country节点和下面的内容
#保存到文件
tree = ET.ElementTree(root)
tree.write('data.xml',encoding='utf-8')