背景

Python有三種方法解析xml：SAX，DOM，Elementree。本文記錄ElementTree方法解析xml。
目前自己用的是Python3.6，但在該版本中并沒有xml的縮進(jìn)函數(shù)ET.indent，不過就我所知3.9版本是有的，所以當(dāng)前3.6寫出來的xml是無法調(diào)用函數(shù)來美化xml排版，文中的xml排版是手動擋 :）。當(dāng)然，也可以寫個函數(shù)來自動優(yōu)化。

正文

xml是一種固有的分層數(shù)據(jù)格式，最好的描述方式就是使用樹形結(jié)構(gòu)。在ElementTree模塊中，使用ElementTree對象來表示一棵樹，Element對象來表示樹中的一個單一結(jié)點。讀取、寫入一個xml文件一般都是在ElementTree層面上操作，而對xml元素（結(jié)點）及其子元素（子結(jié)點）的操作是在Element層面上進(jìn)行。

說明：下面的內(nèi)容有時候使用明確的node名稱來代替Element進(jìn)行操作，有時候使用Element泛指一個結(jié)點。ElementTree和Element是一個類，創(chuàng)建一棵樹tree或者一個結(jié)點node相當(dāng)于類的實例化。

解析xml文件

import xml.etree.ElementTree as ET # 導(dǎo)入ElementTree模塊 tree = ET.parse(xml_file_path) # 解析xml文件，得到樹形結(jié)構(gòu) root = tree.getroot() # 獲取根節(jié)點

結(jié)點基礎(chǔ)：node.tag，node.attrib，node.get()，node.text

每個結(jié)點都有標(biāo)簽（tag）和屬性（attrib），標(biāo)簽名一般不為空，屬性可為空，比如：

<data, attrib=[]><daughter_node, name="child_1", age="20"><chichild_node>...</chichild_node></daughter_node>...<daughter_node>2021</daughter_node><son_node>...<son_node>...<son_node>...<son_node> </data>

上面為一個根結(jié)點及其子結(jié)點的例子，根結(jié)點的標(biāo)簽名為data，沒有屬性（attrib為空：[]）。注意區(qū)分結(jié)點的屬性和結(jié)點的子結(jié)點，屬性包含在結(jié)點的括號<>中，而子結(jié)點是夾在一對標(biāo)簽內(nèi)。獲取一個結(jié)點的標(biāo)簽和屬性（此處結(jié)點為根節(jié)點）：

root.tag # 返回結(jié)點的標(biāo)簽名data root.atrrib # 返回結(jié)點的屬性，此時根節(jié)點的屬性為[]

而對于非空屬性的結(jié)點，其屬性可以單個或者多個，比如下面這個結(jié)點child node，它帶有兩個屬性name和age：

<child_node, name="child_1" age="20">... </child_node>

可以使用如下代碼來訪問child_node屬性：

child_node.attrib # 返回一個字典，字典包含每個屬性名和屬性內(nèi)容 child_node.get("name") # 返回name屬性的屬性內(nèi)容:child_1

假定C結(jié)點（node結(jié)點）沒有子結(jié)點，但夾有文本信息：

<node>2021</node>

則要獲取node結(jié)點的文本信息，有：

node.text # 返回“2021”

結(jié)點拔高：遍歷，索引，遍歷指定結(jié)點Element.iter()，查找指定結(jié)點Element.findall()

• 有子結(jié)點的父結(jié)點是可迭代循環(huán)的，可以用for循環(huán)遍歷父結(jié)點的所有子結(jié)點

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

• 通過索引的方式獲取結(jié)點

  node = root[0][1]

  此處表示返回root結(jié)點（A）的第0個子結(jié)點（B）的第1個子結(jié)點（C），ABC三個的關(guān)系是，A是B的父節(jié)點，A是C的爺爺節(jié)點，B是C的父節(jié)點。

• 指定遍歷某一類結(jié)點

  假定要遍歷上面例子中data結(jié)點下的所有daughter_node，則可以使用Element.iter()來指定遍歷結(jié)點：

  for daughter in data.iter("daughter_node"):print(daughter.tag)print(daughter.attrib)

  ps：可以用data.findall()替換data.iter()，進(jìn)行同樣的迭代，但區(qū)別在哪里自己探索吧。

創(chuàng)建xml并保存

• 創(chuàng)建結(jié)點：ElementTree.Element()
• 創(chuàng)建樹形結(jié)構(gòu)：ElementTree.ElementTree()
• 保存為xml文件：ElementTree.write()
• 添加子結(jié)點：ElementTree.append()

要創(chuàng)建xml文件，那首先就要創(chuàng)建一個樹形結(jié)構(gòu)，對于樹形結(jié)構(gòu)，肯定是在ElementTree這個層次上創(chuàng)建，相當(dāng)于創(chuàng)建了一個樹架子，而Element是創(chuàng)建結(jié)點，創(chuàng)建樹架子和結(jié)點后，你需要將Element結(jié)點掛到樹架子ElementTree上，因此：

xml_2_path = r"path\to\save\your\xml\file.xml" root = ET.Element("data", {"year":"2021", "age":"21"}) # 創(chuàng)建根節(jié)點 new_tree = ET.ElementTree(element=root) # 創(chuàng)建樹形結(jié)構(gòu)，再將根節(jié)點傳遞到樹中 child = ET.Element("child_1") # 創(chuàng)建一個結(jié)點 child.text = "Anya" # 創(chuàng)界該結(jié)點的text內(nèi)容 root.append(child) # 將該結(jié)點連接到根結(jié)點root，此時該結(jié)點便成了root的子結(jié)點 new_tree.write(xml_2_path)

修改xml

• 修改文本：node.text = 2022
• 修改屬性：node.set()
• 移除結(jié)點：Element.remove()
• 查找特定結(jié)點：Element.findall()

假定目前已有xml文件如下：

<data, attrib=[]><daughter_node, name="child_1", age="20"><chichild_node>...</chichild_node></daughter_node>...<daughter_node>2021</daughter_node><son_node>...<son_node>...<son_node>...<son_node> </data>

• 對于創(chuàng)建的Element對象，可以通過直接對其結(jié)點域（fields）賦值，達(dá)到修改的目的，例如：

  daughter_node.text = 2022 # 原值為2021，經(jīng)賦值后，當(dāng)前值為2022

• 對于一個結(jié)點的屬性，可以使用Element.set()來新增或修改結(jié)點屬性：

  data[0].set("age", "21") data[0].set("where", "home")

  data[0]代表data的第一個子結(jié)點daughter_node，有屬性name和age，沒有屬性where，因此上面的代碼第一條修改了age屬性，將20修改為21；新增了屬性where，其屬性值為home。此時該節(jié)點有三個屬性。

• 假定當(dāng)前for循環(huán)遍歷所有son_node，刪掉滿足判斷條件（此處為True）的son_node，那么：

  for one_son_node in data.findall("son_node"):if True:data.remove(one_son_node)

  注意，此處不能用data.iter()替代data.findall()，因為后者只是查找，返回的是查找的結(jié)果；而前者是迭代，如果在迭代的過程中修改，會導(dǎo)致迭代發(fā)生錯誤

保存修改后的xml樹

在經(jīng)過上面的一系列修改后，此時的xml文件里的內(nèi)容并沒有修改，因此需要將修改后的樹重寫進(jìn)文件中：

ElementTree.write(r"path\to\save\your\xml\file.xml")

結(jié)語

寫到到這里，自己完全明白了怎么建立樹形結(jié)構(gòu)、結(jié)點及兩者的相關(guān)操作，能準(zhǔn)確區(qū)分ElementTree和Element到底是什么。本文只講了一些基礎(chǔ)的操作，看完后完全可以自行進(jìn)官網(wǎng)查看兩者的文檔，發(fā)覺更多其他更操作：

• ElementTree
• Element

總結(jié)

以上是生活随笔為你收集整理的「Python」 ElementTree模块解析xml文件，建议小白阅读全文的全部內(nèi)容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網(wǎng)站內(nèi)容還不錯，歡迎將生活随笔推薦給好友。