python总结——处理XML(xml.etree.ElementTree)

2016-05-03  橙子 

python处理XML

我们使用下面的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>

 

1、导入ElementTree

在使用xml.etree.ElementTree时,一般都按如下导入:

try:

    import xml.etree.cElementTree as ET

except ImportError:

    import xml.etree.ElementTree as ET

 

2、导入数据

#从硬盘的xml文件读取数据
tree = ET.parse('E:/country_data.xml')
#从字符串读取数据
root = ET.fromstring(country_data_as_string)

 

3、读取数据

Element.iter(tag):遍历该Element所有后代,也可以指定tag进行遍历寻找。

Element.findall(tag/path):查找当前元素下tag或path能够匹配的直系节点。

Element.find(tag/path):查找当前元素下tag或path能够匹配的首个直系节点。

Element.text: 获取当前元素的text值。

Element.get(key):获取元素指定key对应的属性值,如果没有该属性,则返回default值。

Element.iter(tag)

>>> for neighbor in root.iter('neighbor'):
...   print neighbor.attrib
...
{'name': 'Austria', 'direction': 'E'}
{'name': 'Switzerland', 'direction': 'W'}
{'name': 'Malaysia', 'direction': 'N'}
{'name': 'Costa Rica', 'direction': 'W'}
{'name': 'Colombia', 'direction': 'E'}

findall,find,get

>>> for country in root.findall('country'):
...   rank = country.find('rank').text
...   name = country.get('name')
...   print name, rank
...
Liechtenstein 1
Singapore 4
Panama 68

 

4、修改xml文件

Element.text=new_value  修改节点值为new_value

ElementTree.write(path) 保存修改后的文件

 

write(file, encoding="utf-8")

file:文件名,或者打开进行写操作的文件对象

encoding:结果文件的编码格式,默认是US-ASCII,可以改为’utf=8’

>>> for rank in root.iter('rank'):
...   new_rank = int(rank.text) + 1
...   rank.text = str(new_rank)
...   rank.set('updated', 'yes')
...
>>> tree.write('output.xml')

 

520°|5203 人阅读|0 条评论
登录 后发表评论