xml python et_python xml处理

一、xml.etree.ElementTree(以下简称ET)

Python标准库中,提供了ET的两种实现。一个是纯Python实现的xml.etree.ElementTree,另一个是速度更快的C语言实现xml.etree.cElementTree。请记住始终使用C语言实现,因为它的速度要快很多,而且内存消耗也要少很多。如果你所使用的Python版本中没有cElementTree所需的加速模块,你可以这样导入模块:

try:

import xml.etree.cElementTree as ET

except ImportError:

import xml.etree.ElementTree as ET

如果某个API存在不同的实现,上面是常见的导入方式。当然,很可能你直接导入第一个模块时,并不会出现问题。请注意,自Python 3.3之后,就不用采用上面的导入方法,因为ElemenTree模块会自动优先使用C加速器,如果不存在C实现,则会使用Python实现。因此,使用Python 3.3+的朋友,只需要import xml.etree.ElementTree即可。

以country.xml为例,内容如下:

2

2008

141100

5

2011

59900

69

2011

13600

1.1解析

1.1.1调用 parse() 方法,返回解析树

import xml.etree.ElementTree as ET

tree = ET.parse("country.xml") #

root = tree.getroot() # 获取根节点

本质上和方法三相同,parse() 源码如下:

def parse(source, parser=None):

"""Parse XML document into element tree.

*source* is a filename or file object containing XML data,

*parser* is an optional parser instance defaulting to XMLParser.

Return an ElementTree instance.

"""

tree = ElementTree()

tree.parse(source, parser)

return tree

2)调用 from_string() ,返回解析树的根元素

import xml.etree.ElementTree as ET

data = open("country.xml").read()

root = ET.fromstring(data) #

3)调用 ElementTree模块的 ElementTree(self, element=None, file=None)类 # 这里的element作为根节点

import xml.etree.ElementTree as ET

tree = ET.ElementTree(file="country.xml") #

root = tree.getroot() #

1.2遍历

1.2.1简单遍历

import xml.etree.ElementTree as ET

tree = ET.parse("country.xml")

root = tree.getroot()

print(root.tag, ":", root.attrib) # 打印根元素的tag和属性

# 遍历xml文档的第二层

for child in root:

# 第二层节点的标签名称和属性

print("" + child.tag,":", child.attrib)

# 遍历xml文档的第三层

for children in child:

# 第三层节点的标签名称和属性

print("" + children.tag, ":", children.attrib)

可以通过下标的方式直接访问节点

# 访问根节点下第一个country的第二个节点year,获取对应的文本

year = root[0][1].text # 2008

1.2.2ElementTree提供的方法

find(match)    # 查找第一个匹配的子元素, match可以时tag或是xpaht路径

findall(match) # 返回所有匹配的子元素列表

findtext(match, default=None) #

iter(tag=None) # 以当前元素为根节点 创建树迭代器,如果tag不为None,则以tag进行过滤

iterfind(match) #

例子:

# 过滤出所有neighbor标签

for neighbor in root.iter("neighbor"):

print(neighbor.tag, ":", neighbor.attrib)

---

# 遍历所有的counry标签

for country in root.findall("country"):

# 查找country标签下的第一个rank标签

rank = country.find("rank").text

# 获取country标签的name属性

name = country.get("name")

print(name, rank)

1.3 修改xml结构

1.3.1 属性相关

# 将所有的rank值加1,并添加属性updated为yes

for rank in root.iter("rank"):

new_rank = int(rank.text) + 1

rank.text = str(new_rank) # 必须将int转为str

rank.set("updated", "yes") # 添加属性

# 再终端显示整个xml

ET.dump(root)

# 注意 修改的内容存在内存中 尚未保存到文件中

# 保存修改后的内容

tree.write("output.xml")

---

import xml.etree.ElementTree as ET

tree = ET.parse("output.xml")

root = tree.getroot()

for rank in root.iter("rank"):

# attrib为属性字典

# 删除对应的属性updated

del rank.attrib[‘updated‘]

ET.dump(root)

小结: 关于class xml.etree.ElementTree.``Element 属性相关

attrib    # 为包含元素属性的字典

keys() # 返回元素属性名称列表

items() # 返回(name,value)列表

get(key, default=None) # 获取属性

set(key, value) # 跟新/添加 属性

del xxx.attrib[key] # 删除对应的属性

1.3.2节点/元素 相关

删除子元素remove()

import xml.etree.ElementTree as ET

tree = ET.parse("country.xml")

root = tree.getroot()

# 删除rank大于50的国家

for country in root.iter("country"):

rank = int(country.find("rank").text)

if rank > 50:

# remove()方法 删除子元素

root.remove(country)

ET.dump(root)

添加子元素

import xml.etree.ElementTree as ET

tree = ET.parse("country.xml")

root = tree.getroot()

country = root[0]

last_ele = country[len(list(country))-1]

last_ele.tail = ‘

# 创建新的元素, tag为test_append

elem1 = ET.Element("test_append")

elem1.text = "elem 1"

# elem.tail = ‘

country.append(elem1)

# SubElement() 其实内部调用的时append()

elem2 = ET.SubElement(country, "test_subelement")

elem2.text = "elem 2"

# extend()

elem3 = ET.Element("test_extend")

elem3.text = "elem 3"

elem4 = ET.Element("test_extend")

elem4.text = "elem 4"

country.extend([elem3, elem4])

# insert()

elem5 = ET.Element("test_insert")

elem5.text = "elem 5"

country.insert(5, elem5)

ET.dump(country)

添加子元素方法总结:

append(subelement)

extend(subelements)

insert(index, element)

1.4创建xml文档

?想创建root Element,然后创建SubElement,最后将root element传入ElementTree(element),创建tree,调用tree.write()方法写入文件

?对于创建元素的3个方法: 使用ET.Element、Element对象的makeelement()方法以及ET.SubElement

import xml.etree.ElementTree as ET

def subElement(root, tag, text):

ele = ET.SubElement(root, tag)

ele.text = text

ele.tail = ‘

root = ET.Element("note")

to = root.makeelement("to", {})

to.text = "peter"

to.tail = ‘

root.append(to)

subElement(root, "from", "marry")

subElement(root, "heading", "Reminder")

subElement(root, "body", "Don‘t forget the meeting!")

tree = ET.ElementTree(root)

tree.write("note.xml", encoding="utf-8", xml_declaration=True)

效果:

由于原生保存的XML时默认无缩进,如果想要设置缩进的话, 需要修改保存方式

import xml.etree.ElementTree as ET

from xml.dom import minidom

def subElement(root, tag, text):

ele = ET.SubElement(root, tag)

ele.text = text

def saveXML(root, filename, indent="", newl="

", encoding="utf-8"):

rawText = ET.tostring(root)

dom = minidom.parseString(rawText)

with open(filename, ‘w‘) as f:

dom.writexml(f, "", indent, newl, encoding)

root = ET.Element("note")

to = root.makeelement("to", {})

to.text = "peter"

root.append(to)

subElement(root, "from", "marry")

subElement(root, "heading", "Reminder")

subElement(root, "body", "Don‘t forget the meeting!")

# 保存xml文件

saveXML(root, "note.xml")

python xml处理

你可能感兴趣的:(xml,python,et)