python解析xml文件

python解析xml文件

import lxml

def parse_xml(xml):
    if len(xml) == 0:
        return {xml.tag: xml.text}
    result = {}
    for child in xml:
        child_result = parse_xml(child)
        if child.tag not in result.keys():
            result.update(child_result)
        else:
            if isinstance(result[child.tag], list):
                result[child.tag].append(child_result[child.tag])
            else:
                result[child.tag] = [result[child.tag]]
    return {xml.tag: result}


anatation_xml = r'C:\Users\tao19\Desktop\机器学习笔记\机器学习笔记\深度学习\目标检测算法\yolov2-tf2\data\train\annotation\X2-10-1.xml'
xml = lxml.etree.fromstring(open(anatation_xml).read())
result = parse_xml(xml)