ElementTree是Python常用的處理XML文件的類。下面將介紹使用ElementTree解析、查找、修改XML的方法。
1、引用方法
import xml.etree.ElementTree as ET
2、一個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>
先對XML的格式做一些說明:
? ? Tag: 標簽,如country標簽
? ? Element:被Tag包圍的部分,值,如 68,2011 等
? ? Attribute:標簽的屬性,如country標簽的name
3、解析XML
3.1 讀取
讀入XML數據有三種途徑,從文件讀入和從字符串讀入,文件讀入有2種
1、python3.3之后ElementTree模塊會自動尋找可用的C庫來加快速度
import xml.etree.ElementTree as ETtree = ET.parse('sample.xml')
root = tree.getroot()
2、調用ElementTree類ElementTree(self, element=None, file=None) # 這里的element作為根節點
import xml.etree.ElementTree as ETtree = ET.ElementTree(file="country.xml")
root = tree.getroot() # <Element 'data' at 0x030EA600>
3、從字符串讀入:
import xml.etree.ElementTree as ETdata = open("country.xml").read()
root = ET.fromstring(data)
# root = ET.fromstring(sample_as_string)
3.2 查看Tag和Attribute
# 當要獲取屬性值時,用attrib方法。 返回一個字典,如
print(root.attrib)
# 可以Element.get(AttributeName)來代替Element.attrib[AttributeName]來訪問。# 當要獲取節點值時,用text方法。返回一個字符串,如
print(root.text)# 當要獲取節點名時,用tag方法。 返回一個字符串,如
print(root.tag)
3.3 查看孩子
.root.attrib返回的是一個空字典,如果看root的孩子,可以得到非空的attrib字典。
1、使用for...in...查詢根節點下二級節點的所有元素
print(root.tag)# 根節點標簽
for child in root:print(child.tag,child.attrib) # 二級節點標簽、屬性、內容,結果得到
# country {'name': 'Liechtenstein'}
# country {'name': 'Singapore'}
# country {'name': 'Panama'}
2、使用findall等查詢根節點下的元素
Element.iter()用來尋找所有符合要求的Tag,注意,這里查找的范圍是所有孩子和孩子的孩子 and so
Element.findall()只查找直接的孩子,返回所有符合要求的Tag的Element
Element.find()只返回符合要求的第一個Element。如果查看Singapore的year的值
for child in root.findall('.//country'):print(child.tag,child.attrib)
4、修改XML
前面已經介紹了如何獲取一個Element的對象,以及查看它的Tag、Attribute、值和它的孩子。下面介紹如何修改一個Element并對XML文件進行保存
修改Element。修改Element可以直接訪問Element.text。
import xml.etree.ElementTree as ET# 修改Element的Attribute,也可以用來新增Attribute:
ET.Element.set('AttributeName','AttributeValue')# 新增孩子節點:
root.append(childElement)# 刪除孩子節點:
root.remove(childElement)
保存XML。我們從文件解析的時候,我們用了一個ElementTree的對象tree,在完成修改之后,還用tree來保存XML文件。
tree.write('output.xml')
構建XML。ElementTree提供了兩個靜態函數(直接用類名訪問,這里我們用的是ET)可以很方便的構建一個XML,如:
import xml.etree.ElementTree as ETroot = ET.Element('data')
country = ET.SubElement(root,'country', {'name':'Liechtenstein'}) # root 增加 country 節點
rank = ET.SubElement(country,'rank') # country 增加 rank 節點
rank.text = '1'
year = ET.SubElement(country,'year') # country 增加 year節點
year.text = '2008'
ET.dump(root)
5、XPath支持
XPath表達式用來在XML中定位Element,下面給一個例子來說明:
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]")
6、namespace?
6.1 namespace 獲取
# namespace 為 正則匹配獲取,可能不同版本的會獲取不到
root = tree.getroot()
namespace = re.search('\{.*?\}', root.tag)[0]
6.2?xml.etree.Element 在處理含有namespace的xml文件時寫入會產生ns0
root = tree.getroot()
namespace = re.search('\{.*?\}', root.tag)[0]
ET.register_namespace('', namespace.strip('{}')) # 部分人說這步需要在parse之前做,在我的版本上不需要
7、其他錯誤
7.1?Error parsing XML: junk after document element
一般合法的XML文件只有一個主根節點,比如
<android123><item1/><item2/><item3/>
</android123>
如果出現了Error parsing XML: junk after document element這樣的錯誤,你的想法可能只要主根有多個節點,比如說
<android123><item1/>
</android123>
<android123><item2/><item3/>
</android123>
xml不能解析,原因有可能為xml格式寫的不對,考慮根節點不對。