【xml】操作,append增加结点,remove删除结点
·
remove:删除结点append:添加节点
import xml.etree.ElementTree as ET
# 删除某一固定类别的结点
def removecls(xml_path, cls):
tree = ET.parse(xml_path) # 保存xml文件是tree
root = tree.getroot() # 修改xml问价是对root操作
# 通常用find来获得某一结点的地址,然后再继续访问一些元素值
# 通过findall或者iterfind,都可以直接获取到所有object的结点地址
objects = root.findall('object')
for obj in objects:
cls = obj.find('name').text
if cls == 'loc':
root.remove(obj)
tree.write('caogao.xml') # 保存xml
# 加入一个object节点
def appendobj(xml_path, name, xmin, ymin, xmax, ymax):
tree = ET.parse(xml_path)
root = tree.getroot()
new_obj = ET.Element('object')
new_obj.tail = '\n'
root.append(new_obj)
new_obj_name = ET.Element('name')
new_obj_name.text = name
new_obj.append(new_obj_name)
new_obj_box = ET.Element('bndbox')
new_obj.append(new_obj_box)
new_obj_box_xmin = ET.Element('xmin')
new_obj_box_xmin.text = str(xmin)
new_obj_box.append(new_obj_box_xmin)
new_obj_box_ymin = ET.Element('ymin')
new_obj_box_ymin.text = str(ymin)
new_obj_box.append(new_obj_box_ymin)
new_obj_box_xmax = ET.Element('xmax')
new_obj_box_xmax.text = str(xmax)
new_obj_box.append(new_obj_box_xmax)
new_obj_box_ymax = ET.Element('ymax')
new_obj_box_ymax.text = str(ymax)
new_obj_box.append(new_obj_box_ymax)
tree.write('caogao2.xml') # 保存xml
if __name__ == '__main__':
xml_path = r'G:\pycharmprojects\ScriptCodes_lwd\linepaste\lines\xmlloc\1a.xml'
# 1.删除某一结点
removecls(xml_path, '')
# 2.添加一个结点
# appendobj(xml_path, 'loc', 200, 300, 300, 400)
魔乐社区(Modelers.cn) 是一个中立、公益的人工智能社区,提供人工智能工具、模型、数据的托管、展示与应用协同服务,为人工智能开发及爱好者搭建开放的学习交流平台。社区通过理事会方式运作,由全产业链共同建设、共同运营、共同享有,推动国产AI生态繁荣发展。
更多推荐


所有评论(0)