python xml.etree.ElementTree模块

时间：2022-08-17 17:52:31

使用的XML文件如下：file.xml

<?xml version="1.0"?>

<data name="ming">

    <country name="Singapore">

        <rank>4</rank>

        <year>2011</year>

        <gdppc>59900</gdppc>

        <neighbor name="Malaysia" direction="N"/>

    </country>

    <country name="Panama">

        <rank>68</rank>

        <year>2011</year>

        <gdppc>13600</gdppc>

        <neighbor name="Costa Rica" direction="W"/>

        <neighbor name="Colombia" direction="E"/>

    </country>

</data>

导入模块

#!/usr/bin/env python

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

# @Time    : 2017/11/30 0030 18:18

# @Author  : ming

import xml.etree.ElementTree as ET

读取文件

result = ET.parse("file.xml")  # 打开文件

root = result.getroot()  # 获取根节点

打印一下根节点信息

print(root.tag, root.attrib)  # 根节点标签和属性值

#运行结果：data {'name': 'ming'}

打印一下根节点的孩子节点

for i in root:  # 循环根节点的孩子节点

    print(i.tag, i.attrib)  # 打印孩子节点的标签和属性值

    print(i[0].tag, i[0].text)  # 打印孙子节点中第一个节点的标签和属性值。多层节点都可以通过下标访问

# 运行结果：

# country {'name': 'Singapore'}

# rank 4

# country {'name': 'Panama'}

# rank 68

findall 直接定位节点

for k in root.findall('country'):  # 直接定位该节点

    print(k.tag, k.attrib)  # 打印孩子节点的标签和属性值

    print(k[1].tag, k[1].text)  # 打印孙子节点中第一个节点的标签和属性值。多层节点都可以通过下标访问

#运行结果

# country {'name': 'Singapore'}

# year 2011

# country {'name': 'Panama'}

# year 2011

iter 遍历所有节点

for m in root.iter():  # 遍历所有节点

    print(m.tag, m.attrib, m.text)  # 打印所有节点的标签、属性、值
data {'name': 'ming'} 

country {'name': 'Singapore'} 

rank {} 4

year {} 2011

gdppc {} 59900

neighbor {'name': 'Malaysia', 'direction': 'N'} None

country {'name': 'Panama'} 

rank {} 68

year {} 2011

gdppc {} 13600

neighbor {'name': 'Costa Rica', 'direction': 'W'} None

neighbor {'name': 'Colombia', 'direction': 'E'} None

运行结果

iter 遍历特定的节点，不遍历子节点

for p in root.iter("country"):  # 遍历所有country节点自身，不遍历子节点

    print(p.tag, p.attrib)  # 打印所有country节点的标签、属性

#运行结果

# country {'name': 'Singapore'}

# country {'name': 'Panama'}

find 查找第一个匹配的节点

firstCountry = root.find("country")  # 返回第一个tag为country的element，如没有，返回None

print(firstCountry.attrib)  # 第一个country节点的属性值

# 运行结果

# {'name': 'Singapore'}

append 增加一个新的节点

newEle = ET.Element("Newtag")  # 新节点的标签

newEle.attrib = {"name": "NewElement", "age": ""}  # 新节点的属性

newEle.text = "This is a new element"  # 新节点的值

# a = root.find("country")  # 第一个country节点

# a.append(newEle)  # 在第一个country中添加

#

# root[0].append(newEle)  # 根节点中第一个子节点下添加

root.append(newEle)  # 把这个新节点放在哪个节点下，这里的root代表根节点

result.write("file.xml")  # 写回原文件
<data name="ming">

    <country name="Singapore">

        <rank>4</rank>

        <year>2011</year>

        <gdppc>59900</gdppc>

        <neighbor direction="N" name="Malaysia"/>

    </country>

    <country name="Panama">

        <rank>68</rank>

        <year>2011</year>

        <gdppc>13600</gdppc>

        <neighbor direction="W" name="Costa Rica"/>

        <neighbor direction="E" name="Colombia"/>

    </country>

    <Newtag age="20" name="NewElement">This is a new element</Newtag>

</data>

运行结果

修改节点信息

# 修改节点属性

sub1 = root.find("country")

sub1.set("name", "Chinese") # 修改name=chinese

result.write("file.xml")  # 写回原文件

# 修改节点值

sub2 = root.find("country")

sub2.find("year").text = "" #修改year的值为2018

result.write("file.xml")  # 写回原文件
<data name="ming">

    <country name="Chinese">

        <rank>4</rank>

        <year>2018</year>

        <gdppc>59900</gdppc>

        <neighbor direction="N" name="Malaysia" />

    </country>

    <country name="Panama">

        <rank>68</rank>

        <year>2011</year>

        <gdppc>13600</gdppc>

        <neighbor direction="W" name="Costa Rica" />

        <neighbor direction="E" name="Colombia" />

    </country>

</data>

删除节点

sub3 = root.find("country")

root.remove(sub3)

result.write("file.xml")  # 写回原文件
<data name="ming">

    <country name="Panama">

        <rank>68</rank>

        <year>2011</year>

        <gdppc>13600</gdppc>

        <neighbor direction="W" name="Costa Rica"/>

        <neighbor direction="E" name="Colombia"/>

    </country>

</data>

运行结果

创建新的XML文件

#!/usr/bin/env python

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

# @Time    : 2017/12/2 0002 14:44

# @Author  : ming

import xml.etree.ElementTree as ET

# 创建一个新的XML文件，根节点为yangyongming

new_xml = ET.Element("yangyongming")

# 孩子节点为name，指点父亲节点为new_xml sttrib属性

new_1 = ET.SubElement(new_xml, "name", attrib={"name": "hello"})

# 孙子节点sex 父亲节点new_1

new_s1 = ET.SubElement(new_1, "sex")

# 孙子节点sex 赋值"man"

new_s1.text = "man"

# 孙子节点 age， 父亲节点new_1

new_s2 = ET.SubElement(new_1, "age")

# 孙子节点age 赋值“18”

new_s2.text = ""

# 转换为XML文件，写入文件

et = ET.ElementTree(new_xml)

et.write("file.xml", encoding="utf-8")
<yangyongming>

    <name name="hello">

        <sex>man</sex>

        <age>18</age>

    </name>

</yangyongming>

运行结果

相关文章