python中实现写入xml文件

import xml.etree.ElementTree as ET
# 创建根元素
root = ET.Element("annotation")
# 创建子元素
folder = ET.SubElement(root, "folder")
folder.text = "image"
filename = ET.SubElement(root, "filename")
filename.text = "601.JPG"
path = ET.SubElement(root, "path")
path.text = r"E:\python\VOCdevkit\image\601.JPG"
source = ET.SubElement(root, "source")
database = ET.SubElement(source, "database")
database.text = "Unknown"
size = ET.SubElement(root, "size")
width = ET.SubElement(size, "width")
width.text = "1280"
height = ET.SubElement(size, "height")
height.text = "1920"
depth = ET.SubElement(size, "depth")
depth.text = "3"
segmented = ET.SubElement(root, 'segmented')
segmented.text = "0"
object_info = ET.SubElement(root, 'object')
name = ET.SubElement(object_info, 'name')
name.text = 'plain_clothe'
pose = ET.SubElement(object_info, 'pose')
pose.text = 'Unspecified'
truncated = ET.SubElement(object_info, 'truncated')
truncated.text = "0"
difficult = ET.SubElement(object_info, 'difficult')
difficult.text = "0"
bndbox = ET.SubElement(object_info, 'bndbox')
xmin = ET.SubElement(bndbox, 'xmin')
xmin.text = "585"
ymin = ET.SubElement(bndbox, 'ymin')
ymin.text = "1032"
xmax = ET.SubElement(bndbox, 'xmax')
xmax.text = "865"
ymax = ET.SubElement(bndbox, 'ymax')
ymax.text = "1883"
# 创建XML树
tree = ET.ElementTree(root)
# 将XML写入文件
tree.write("example.xml")