ElementTree是Python常用的處理XML文件的類(lèi)嫉拐。本文將介紹使用ElementTree解析迅耘、查找、修改XML的方法魄藕。

一、引用方法

ElementTree所在文件保存在Lib/xml/etree/ElementTree.py撵术，所以我們通過(guò)下面的代碼引用它背率，之后就可以使用ET.來(lái)訪問(wèn)ElementTree中的函數(shù)。

import xml.etree.ElementTree as ET

二、一個(gè)XML例子

下面所有的操作都將下面這段XML為例寝姿，我們將它保存為sample.xml交排。

<?xml version="1.0"?>
<data>
    <country name="Liechtenstein">
        <rank>1</rank>
        <year>2008</year>
        <gdppc>141100</gdppc>
        <neighbor name="Austria" direction="E"/>
        <neighbor name="Switzerland" direction="W"/>
    </country>
    <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>

先對(duì)XML的格式做一些說(shuō)明：

• Tag： 使用<和>包圍的部分，如<rank>成為start-tag饵筑，</rank>是end-tags埃篓；
• Element：被Tag包圍的部分，如<rank>68</rank>根资，可以認(rèn)為是一個(gè)節(jié)點(diǎn)架专，它可以有子節(jié)點(diǎn)；
• Attribute：在Tag中可能存在的name/value對(duì)嫂冻，如<country name="Liechtenstein">中的name="Liechtenstein"胶征，一般表示屬性。

三桨仿、解析XML

讀入XML數(shù)據(jù)

首先讀入XML，有兩種途徑案狠，從文件讀入和從字符串讀入服傍。
從文件讀入：

import xml.etree.ElementTree as ET
tree = ET.parse('sample.xml')
root = tree.getroot()

從字符串讀入：

root = ET.fromstring(sample_as_string)

tree和root分布是ElementTree中兩個(gè)很重要的類(lèi)的對(duì)象：

• ElementTree
• Element

查看Tag和Attribute

這時(shí)得到的root是一個(gè)指向<data>Element對(duì)象，我們可以通過(guò)查看root的tag和attrib來(lái)驗(yàn)證這一點(diǎn)：

>>> root.tag
'data'
>>> root.attrib
{}

上面的代碼說(shuō)明了查看一個(gè)Element的Tag和Attribute的方法骂铁，Tag是一個(gè)字符串吹零，而Attribute得到的是一個(gè)字典。

另外拉庵，還可以使用

• Element.get(AttributeName)

來(lái)代替Element.attrib[AttributeName]來(lái)訪問(wèn)灿椅。

查看孩子

root.attrib返回的是一個(gè)空字典，如果看root的孩子钞支，可以得到非空的attrib字典茫蛹。

1、使用for...in...訪問(wèn)

for child in root:
    print child.tag, child.attrib

得到

country {'name': 'Liechtenstein'}
country {'name': 'Singapore'}
country {'name': 'Panama'}

2烁挟、使用下標(biāo)訪問(wèn)

如:

>>> print root[0].tag
country
>>> print root[0][0].tag
rank

3婴洼、使用Tag名稱(chēng)訪問(wèn)

下標(biāo)訪問(wèn)的方法雖然簡(jiǎn)單，但是在未知XML具體結(jié)構(gòu)的時(shí)候并不適用撼嗓，通過(guò)Tag名稱(chēng)訪問(wèn)的方法更具有普適性柬采。這里用到Element類(lèi)的幾個(gè)函數(shù)，分別是

• Element.iter()
• Element.findall()
• Element.find()

這兩個(gè)函數(shù)使用的場(chǎng)景有所差異：
Element.iter()用來(lái)尋找所有符合要求的Tag且警，注意粉捻，這里查找的范圍是所有孩子和孩子的孩子 and so on。如果查看所有的year斑芜，可以使用下面的代碼：

for neighbor in root.iter('year'):
    print neighbor.text

返回

2008
2011
2011

Element.findall()只查找直接的孩子肩刃，返回所有符合要求的Tag的Element，而Element.find()只返回符合要求的第一個(gè)Element。如果查看Singapore的year的值树酪，可以使用下面的代碼：

for country in root.findall('country'):
        if country.attrib['name'] == 'Singapore':
            year = country.find('year')  # 使用Element.find()
            print year.text

for country in root.findall('country'):
        if country.attrib['name'] == 'Singapore':
            years = country.findall('year')  # 使用Element.findall()
            print years[0].text  # 注意和上段的區(qū)別

查看Element的值

我們可以直接用Element.text來(lái)得到這個(gè)Element的值浅碾。

四、修改XML

前面已經(jīng)介紹了如何獲取一個(gè)Element的對(duì)象续语，以及查看它的Tag垂谢、Attribute、值和它的孩子疮茄。下面介紹如何修改一個(gè)Element并對(duì)XML文件進(jìn)行保存

修改Element

修改Element可以直接訪問(wèn)Element.text滥朱。
修改Element的Attribute，也可以用來(lái)新增Attribute：

Element.set('AttributeName','AttributeValue')

新增孩子節(jié)點(diǎn)：

Element.append(childElement)

刪除孩子節(jié)點(diǎn)：

Element.remove(childElement)

保存XML

我們從文件解析的時(shí)候力试，我們用了一個(gè)ElementTree的對(duì)象tree徙邻，在完成修改之后，還用tree來(lái)保存XML文件畸裳。

tree.write('output.xml')

構(gòu)建XML

ElementTree提供了兩個(gè)靜態(tài)函數(shù)（直接用類(lèi)名訪問(wèn)缰犁，這里我們用的是ET）可以很方便的構(gòu)建一個(gè)XML，如：

root = ET.Element('data') 
    country = ET.SubElement(root,'country', {'name':'Liechtenstein'})
    rank = ET.SubElement(country,'rank')
    rank.text = '1'
    year = ET.SubElement(country,'year')
    year.text = '2008'
    ET.dump(root)

就可以得到

<data><country name="Liechtenstein"><rank>1</rank><year>2008</year></country></data>

五怖糊、XPath支持

XPath表達(dá)式用來(lái)在XML中定位Element帅容，下面給一個(gè)例子來(lái)說(shuō)明：

import xml.etree.ElementTree as ET

root = ET.fromstring(countrydata)

# Top-level elements
root.findall(".")

# All 'neighbor' grand-children of 'country' children of the top-level
# elements
root.findall("./country/neighbor")

# Nodes with name='Singapore' that have a 'year' child
root.findall(".//year/..[@name='Singapore']")

# 'year' nodes that are children of nodes with name='Singapore'
root.findall(".//*[@name='Singapore']/year")

# All 'neighbor' nodes that are the second child of their parent
root.findall(".//neighbor[2]")

參考

ElementTree主頁(yè)
ElementTree的函數(shù)與類(lèi)介紹