python读取/创建XML文件

Python中定义了很多处理XML的函数,如xml.dom,它会在处理文件之前,将根据xml文件构建的树状数据存在内存。还有xml.sax,它实现了SAX API,这个模块牺牲了便捷性,换取了速度和减少内存占用。

本文将要说明的是xml.tree.ElementTree的使用。与DOM比较,它使用起来更快更方便,和SAX比较呢,性能相仿,但使用起来更快捷。

ET(ElementTree)提供了两个对象:ElementTree和Element

ElementTree:将整个XML转化为树,对整个XML文档进行操作(读取,写入,查找等)一般在ElementTree层面进行。

Element:代表树上单个节点,对单个XML元素及其子元素进行操作,则是在Element层面进行。

1)加载整个文档(demo.xml):

  import xml.etree.ElementTree as ET

  tree = ET.ElementTree(file="demo.xml")

2)获取根元素

  root = tree.getroot()

  根元素是一个Element对象,它具有以下属性:

  root.tag:返回元素的标签名

  root.attrib:以字典形式返回属性名和值

3)根元素本身就是一个可迭代对象,和其他Element对象一样,也具备直接遍历子元素的接口

  for child in root:

    print(child.tag, child.attrib)

  也可以通过索引来访问特定的子元素 root[1].tag

4)查找需要的元素:find,findall, findText,iterfind等

  find(tagName):总是返回第一个匹配的元素

  findall(tagName):返回当前元素下一级所有匹配的元素列表

  findtext:

  iterfind(tagName):作用和findall一样,但是它返回的是一个生成器。

4)要想找到当前元素下所有元素,而不是只找到下一级元素

  list(root.iter())  #列出根元素下所有子节点列表

  list(root.iter(tagName))  # 列出所有标签名为tagName的子节点

5)正则表达式的使用:

  *:所有 ---------> root.find("Menues/*") 查找路径Menus下面的所有子节点

  .:当前元素---------->root.find(./*)  查找当前元素下的所有子节点

  //:------------> root.findall(".//Menu"):查找当前目录下任意层级的标签名为Menu的子元素

  ..:------------>root.findall(".//Menu/.."):查找当前目录下任意层级的标签名为Menu的子元素的父元素

  [@attrib]:根据指定的属性搜索元素

  [@attrib='value']:根据给定属性名搜索元素--------->root.findall("Tab[@type='subabsent']"):找到所有type为subabsent的Tab标签

  [tag]:----->root.findall("Tab[Menues]"):找到包含子元素为Menues的Tab标签

  [tag='text']:---------->root.findall("Tab[Menues=5]"):找到包含Menues标签,且Menues标签中间text值为5的Tab元素

  [position]:----->根据元素位置找相应的元素,从1开始: root.findall("Tab[1]")  root.findall("Tab[last()-1]"):找到倒数第二个元素

  

 写入XML文件:

  调用ElementTree的write函数:ElementTree.write(file)

  def pretty(e,level=0):

    # 格式化xml文件

    if len(e)>0:

      e.text = " "+" "*(level+1)

      for child in e:

        pretty(child,level+1)

      child.tail = child.tail[:-1]

    e.tail = " "+" "*level

  

  把如下CSV文件写入XML文件

  

---》

        

  import xml.etree.ElementTree as ET

  from xml.etree.ElementTree import ElementTree,Element

  import csv

  def WriteXML(csvfile):

    # 把CSV文件写入到xml文件

    with open(csvfile,"r") as rf:

      reader = csv.reader(rf)

      header = next(reader)

      root = Element("Data")

      for row in reader:

        eRow = Element("Row")

        root.append(eRow)

        for tag, text in zip(header, row):

          e = Element(tag.strip())

          e.text = text.strip()

          eRow.append(e)

   pretty(root)

   return ElementTree(root)

原文地址:https://www.cnblogs.com/fiona-zhong/p/9910548.html