python修改xml文件_python-----批量操作xml文件（新建、增、删、改、查）

#!/usr/bin/env python#-*- coding: utf-8 -*-#@Time : 2019/7/9 15:16#@Author : xiaodai

importxml.etree.ElementTree as ETfrom lxml importetreefrom tqdm importtqdmimportosdefcreat_xml(xml_file,x,y,w,h):"""新建xml文件

:param xml_file: xml

:param x: xmin

:param y: ymin

:param w: 宽

:param h: 高

:return:"""root= etree.Element("annotation") #创建根节点 annotation

child1 = etree.SubElement(root, "folder")

child1.text= "VOC2007"object= etree.SubElement(root, "object") #创建annotation节点的子节点 object

namen = etree.SubElement(object, "name") #创建object节点的子节点 name

namen.text = 'person' #name的值为person

bndbox = etree.SubElement(object, "bndbox")

xminn= etree.SubElement(bndbox, "xmin")

xminn.text=str(x)

yminn= etree.SubElement(bndbox, "ymin")

yminn.text=str(y)

xmaxn= etree.SubElement(bndbox, "xmax")

xmaxn.text= str(x +w)

ymaxn= etree.SubElement(bndbox, "ymax")

ymaxn.text= str(y +h)

weight= etree.SubElement(bndbox, "box_w")

weight.text=str(w)

height= etree.SubElement(bndbox, "box_h")

height.text=str(h)

tree=etree.ElementTree(root)

tree.write(xml_file, pretty_print=True, xml_declaration=False, encoding='utf-8')defedit_xml(xml_file):"""修改xml文件

:param xml_file:xml文件的路径

:return:"""tree=ET.parse(xml_file)

objs= tree.findall('object')for ix, obj inenumerate(objs):

name= obj.find('name').textif name == 'mouse':print(xml_file)

obj_new= obj.find('bndbox')

xmin= str( 1280 - int(obj_new.find('xmin').text))

xmax= str( 1280 - int(obj_new.find('xmax').text))

obj_new.find('xmin').text = xmin #修改节点值

obj_new.find('xmax').text =xmax

conf= obj_new.find('conf')

conf.tag= 'fenshu' #修改节点名

newEle= ET.Element("NewElement1")

newEle.text= "new element"obj_new.append(newEle)#新增节点

newEle= obj_new.find('xmin')

obj_new.remove(newEle)#删除节点

obj_new.set('key','value') #给bndbox节点设置属性

tree.write(xml_file,method='xml',encoding='utf-8') #更新xml文件

if __name__ == '__main__':

path= r'F:\temp\tmp_0709'xml_files= [os.path.join(rootdir, file) for rootdir, _, files in os.walk(path) for file in files if(file.endswith('.xml'))]for xml intqdm(xml_files):

edit_xml(xml)#修改xml文件

for i in tqdm(range(10)):

xml= os.path.join(path,str(i)+'.xml')

creat_xml(xml,32,120,80,80) #新建xml文件