我在使用Python解析比較大型的xml文件時,為了提高效率,決定使用iterparse()方法,但是發現根據網上的例子:每次if event == 'end':之后elem.clear()或者是每次 if elem.tag == '':之后clear(),都只能去到當前標簽的相關內容,如果想繼續讀取得到標簽的子標簽,則會返回為空,也就是取不到。
其實iterparse()方法的原理是當遇到標簽的“>”符號時觸發start,當遇到標簽的結束標志是會觸發end,比如:
<item>
<country>
<city></city>
<city></city>
</country>
</item>
在這個xml栗子中,用iterparse()方法捕捉end來獲取標簽,如果要獲取<country>標簽,那elem先獲取的是country下的<city>標簽,因為在遍歷到<country>標簽的結束標志</>前,先遍歷到<city>的結束標志,所以elem先獲取<country>標簽的子標簽<city>,這類似於深度優先遍歷,所以如果每次if完都clear()的話,會clear掉相應標簽的子標簽,所以得不到其子標簽。
我嘗試了各種方法來解決這個問題,后來通過這么寫來實現的:
from xml.etree import ElementTree
for event, elem in ElementTree.iterparse(xml):
if event == 'end':
if elem.tag == 'name' and elem.find('children').get('name') == attr_name:
相應標簽的操作
elem.clear()
if ....:
elem.clear()
我把clear()方法放在了if里面,這樣只有在找到相應標簽並且讀取完其子標簽后再clear(),這樣就可以讀取子標簽以及多層節點了。
功能是實現了,但是不知道其效率有沒有影響,本人菜鳥剛接觸Python,還望大牛們指點。