python操作xml

$$python操作xml$$

一 生成xml

from xml.dom.minidom import Document # 引入Document模块
doc = Document()  # 创建Document
# 1.创建节点
train_config = doc.createElement("train_config")  # 创建节点 （根节点）
param = doc.createElement("param")  # 创建节点（二级）
epoch = doc.createElement("epoch")  # 创建节点（三级）
batchsize = doc.createElement("batchsize")  # 创建节点（三级）
dataset_path = doc.createElement("dataset_path")  # 创建节点（三级）

# 2.设置节点内容
epoch_content = doc.createTextNode("100")
batchsize_content = doc.createTextNode("2")
dataset_path_content = doc.createTextNode(r"D:\2.png")

# 3.在节点中加入内容
epoch.appendChild(epoch_content)
batchsize.appendChild(batchsize_content)
dataset_path.appendChild(dataset_path_content)

# 4.设置节点层级
doc.appendChild(train_config)  # 3.1 添加根节点
train_config.appendChild(param) # 3.2 添加二级节点
param.appendChild(epoch) # 3.2 添加三级节点
param.appendChild(batchsize) # 3.2 添加三级节点
param.appendChild(dataset_path) # 3.2 添加三级节点


# 5.保存
filename = "train_config.xml"
f = open(filename, "w")
f.write(doc.toprettyxml(indent="   "))  # 设置间隔
f.close()

生产的xml文件内容

<?xml version="1.0" ?>
<train_config>
   <param>
      <epoch>100</epoch>
      <batchsize>2</batchsize>
   </param>
</train_config>

二 遍历xml节点

for child in root:
    print(child.tag, child.attrib)  # 获取下一级的标签和属性
    for i in child:
        print('  ', i.tag, i.text)  # 获取再下一级的标签和文本

# 只遍历epoch节点
for node in root.iter('epoch'):  # 在 根节点的子节点中过滤出标签‘epoch’
    print(node.tag, node.text)  # 打印节点标签和文本

三 解析xml节点

import xml.etree.ElementTree as ET # 导入ElementTree模块

tree = ET.parse('train_config.xml') # 获取解析对象
root = tree.getroot()  # 获取根节点
print(root.tag)
for child in root:
    print(child.tag, child.attrib)  # 获取下一级的标签和属性
    for i in child:
        print('  ', i.tag, i.text)  # 获取再下一级的标签和文本

四 修改xml节点

for node in root.iter('epoch'):
    new_year = 123  # 修改获取到的文本 （先转为int 型）
    node.text = str(new_year)  # 再转为string 型
tree.write("train_config.xml")  # 持久化

五 删除xml节点