python3-将xml文件转成json

2020-01-19 15:21:55 浏览数 (1)

代码语言:javascript复制
from xml.etree import ElementTree as et
import json
def readxml_et():
    tree = et.ElementTree(file="D:\vscode\xml2json\Annotations\timg.xml")
    root = tree.getroot()
    A = dict()
    listbigoption=[]
    for child_root in root:
        if child_root.tag == 'filename':
            imagePath = child_root.text
        if child_root.tag == 'object':
            listobject=dict()
            for xylabel in child_root:
                if xylabel.tag == 'name':
                    label = xylabel.text                    
                if xylabel.tag == 'polygon':
                    listpoints=[]
                    for pt in xylabel:
                        if pt.tag == 'pt':
                            listxy = []
                            for i in pt:                                
                                if i.tag == 'x':
                                    listxy.append(int(i.text))
                                if i.tag == 'y':
                                    listxy.append(int(i.text))
                                listpoints.append(listxy)
                    listobject['points'] = listpoints
                    listobject['line_color'] = 'null'
                    listobject['label'] = label
                    listobject['fill_color'] = 'null'
                    listbigoption.append(listobject)
                # print(listbigoption)
    A['lineColor'] = [0,255,0,128]
    A['imageData'] = 'imageData'
    A['fillColor'] = [255,0,0,128]
    A['imagePath'] = imagePath
    A['shapes'] = listbigoption
    A['flags'] = {}
    with open('1.json','w') as f:
        json.dump(A,f)

readxml_et()

  同事给了我一个xml文件让我转成他想要的json格式

代码用了好多for和if,用for循环来找子节点,先完成任务,得改进,还不知道怎么改。

0 人点赞