看代碼吧~
# -*- coding:utf-8 -*-
import os
import json
import numpy as np
#from xml.etree import ElementTree as etree
from xml.etree.ElementTree import Element
from xml.etree.ElementTree import SubElement
from xml.etree.ElementTree import ElementTree
imagePath = r'E:\Desktop\SteelCoilsDetection\test\images'
jsonPath = r'E:\Desktop\SteelCoilsDetection\test\json'
savePath = r'E:\Desktop\SteelCoilsDetection\test\xml'
jsonList = os.listdir(jsonPath)
for jsonName in jsonList:
print(jsonName)
readPath = os.path.join(jsonPath, jsonName)
# 打開json文件
with open(readPath, 'r') as file_loader:
jsonDic = json.load(file_loader)
# print(jsonDic.keys())
# dict_keys(['version', 'flags', 'shapes', 'imagePath', 'imageData', 'imageHeight', 'imageWidth'])
# 生成xml文件
annotation = Element('annotation')
folder = SubElement(annotation, 'folder')
folder.text = "images"
filename = SubElement(annotation, 'filename')
filename.text = jsonName.split('.')[0]
path = SubElement(annotation, 'path')
path.text = imagePath + jsonName.split('.')[0]
source = SubElement(annotation, 'source')
database = SubElement(source, 'database')
database.text = "Unknown"
size = SubElement(annotation, 'size')
width = SubElement(size, 'width')
width.text = str(jsonDic['imageWidth'])
height = SubElement(size, 'height')
height.text = str(jsonDic['imageHeight'])
depth = SubElement(size, 'depth')
depth.text = "3"
segmented = SubElement(annotation, 'segmented')
segmented.text = "0"
for shape in jsonDic['shapes']:
if shape["label"] == 'a':
continue
object = SubElement(annotation, 'object')
name = SubElement(object, 'name')
name.text = shape["label"]
pose = SubElement(object, 'pose')
pose.text = 'Unspecified'
truncated = SubElement(object, 'truncated')
truncated.text = str(0)
difficult = SubElement(object, 'difficult')
difficult.text = str(0)
points = shape['points']
mritx = np.array(points)
xxmin = min(mritx[:, 0])
xxmax = max(mritx[:, 0])
yymin = min(mritx[:, 1])
yymax = max(mritx[:, 1])
bndbox = SubElement(object, 'bndbox')
xmin = SubElement(bndbox, 'xmin')
xmin.text = str(int(xxmin))
ymin = SubElement(bndbox, 'ymin')
ymin.text = str(int(yymin))
xmax = SubElement(bndbox, 'xmax')
xmax.text = str(int(xxmax))
ymax = SubElement(bndbox, 'ymax')
ymax.text = str(int(yymax))
tree = ElementTree(annotation)
tree.write(os.path.join(savePath, jsonName.split('.')[0]+'.xml'), encoding = 'utf-8')
美化：
# -*- coding:utf-8 -*-
import os
from xml.etree import ElementTree # 導(dǎo)入ElementTree模塊
# elemnt為傳進(jìn)來的Elment類，參數(shù)indent用于縮進(jìn)，newline用于換行
def prettyXml(element, indent, newline, level = 0):
# 判斷element是否有子元素
if element:
# 如果element的text沒有內(nèi)容
if element.text == None or element.text.isspace():
element.text = newline + indent * (level + 1)
else:
element.text = newline + indent * (level + 1) + element.text.strip() + newline + indent * (level + 1)
# 此處兩行如果把注釋去掉，Element的text也會另起一行
#else:
#element.text = newline + indent * (level + 1) + element.text.strip() + newline + indent * level
temp = list(element) # 將elemnt轉(zhuǎn)成list
for subelement in temp:
# 如果不是list的最后一個元素，說明下一個行是同級別元素的起始，縮進(jìn)應(yīng)一致
if temp.index(subelement) (len(temp) - 1):
subelement.tail = newline + indent * (level + 1)
else: # 如果是list的最后一個元素， 說明下一行是母元素的結(jié)束，縮進(jìn)應(yīng)該少一個
subelement.tail = newline + indent * level
# 對子元素進(jìn)行遞歸操作
prettyXml(subelement, indent, newline, level = level + 1)
dir = r'E:\Desktop\SteelCoilsDetection\test\xml'
for fileName in os.listdir(dir):
print(fileName)
tree = ElementTree.parse(os.path.join(dir, fileName)) #解析test.xml這個文件，該文件內(nèi)容如上文
root = tree.getroot() #得到根元素，Element類
prettyXml(root, '\t', '\n') # 執(zhí)行美化方法
#ElementTree.dump(root) #顯示出美化后的XML內(nèi)容
tree.write(os.path.join(dir, fileName), encoding = 'utf-8')
補充：Python 標(biāo)準(zhǔn)庫 xml 詳解
對于簡單的 XML 解析處理, 可以使用標(biāo)準(zhǔn)庫 xml, 相對于第三方庫 lxml, xml 無需額外安裝, 但 xml 是用 Python 實現(xiàn)的, 性能不如 lxml
XML 的解析功能主要由 xml.etree.ElementTree 模塊完成, 其中包含兩個類, ElementTree 用于表示整個 XML 文檔, 而 Element 表示文檔中的一個節(jié)點
示例數(shù)據(jù), 命名為 book.xml
?xml version="1.0"?>
bookstore>
book name="西游記">
author>吳承恩/author>
dynasty>明朝/dynasty>
similar name="封神演義" author="許仲琳"/>
/book>
book name="紅樓夢">
author>曹雪芹/author>
dynasty>清朝/dynasty>
/book>
book name="三國演義">
author>羅貫中/author>
dynasty>明末清初/dynasty>
similar name="三國志" author="陳壽"/>
/book>
/bookstore>
導(dǎo)入要解析的 XML 文檔, 并獲取文檔的根節(jié)點
import xml.etree.ElementTree as ET
tree = ET.parse("./book.xml")
root = tree.getroot()
也可以直接解析字符串
with open("./book.xml") as fp:
root = ET.fromstring(fp.read())
對于每一個節(jié)點 Element:
通過列表接口可以訪問直接子節(jié)點
通過字典接口可以訪問屬性節(jié)點, 也可通過 attrib 屬性(例如 root.attrib)得到真正的字典
其他還有 tag 屬性表示標(biāo)簽名, text 表示其包含的文本內(nèi)容
# 遍歷直接子節(jié)點
for book in root:
print(book.tag, book.attrib, book.get("name"))
# 訪問根節(jié)點下的第2個子節(jié)點, 再向下訪問第1個子節(jié)點的文本, 也就是 "author>曹雪芹/author>"
author = root[1][0].text
print(type(author), author)
打印輸出
book {'name': '西游記'} 西游記
book {'name': '紅樓夢'} 紅樓夢
book {'name': '三國演義'} 三國演義
class 'str'> 曹雪芹
獲取到的文本結(jié)果與 lxml 不同, 這里的結(jié)果直接是字符串類型
遞歸函數(shù), 可以遍歷所有的后代節(jié)點
# 遞歸選擇所有標(biāo)簽名為 "similar" 的節(jié)點
for book in root.iter("similar"):
print(book.attrib)
打印輸出
{'name': '封神演義', 'author': '許仲琳'}
{'name': '三國志', 'author': '陳壽'}
XPath 語法
XPath 類似于文件路徑, 路徑中最末尾的部分表示要提取的內(nèi)容, 分隔符有兩種, "/"表示直接子節(jié)點的關(guān)系, "http://"表示所有的子節(jié)點
| 語法 |
含義 |
| tag |
匹配特定標(biāo)簽 |
| * |
匹配所有元素 |
| . |
當(dāng)前節(jié)點, 用于相對路徑 |
| … |
父節(jié)點 |
| [@attrib] |
匹配包含 attrib 屬性的節(jié)點 |
| [@attrib=‘value'] |
匹配 attrib 屬性等于 value 的節(jié)點 |
| [tag] |
匹配包含直接子節(jié)點 tag 的節(jié)點 |
| [tag=‘text'] |
匹配包含直接子節(jié)點 tag 且子節(jié)點文本內(nèi)容為 text 的節(jié)點 |
| [n] |
匹配第 n 個節(jié)點 |
[] 前面必須有標(biāo)簽名, book[@name][similar] 匹配帶有 name 屬性以及 similar 直接子節(jié)點的 book 節(jié)點, 然后將 book[@name][similar] 置于 XPath 路徑中, 例如 “/bookstore/book[@name][similar]”
可以通過 Element 對象的方法 findall(path) 和 find(path) 使用 XPath 語法, 次時路徑是從 Element 代表的節(jié)點開始, 也可以通過 ElementTree 對象調(diào)用 findall 與 find, 相當(dāng)于路徑從根節(jié)點開始
匹配到節(jié)點, findall 返回所有匹配節(jié)點的列表, find 返回首個匹配節(jié)點, 沒有匹配到節(jié)點時, findall 返回空列表, find 返回 None
# . 表示 bookstore 節(jié)點
author_1 = tree.find("./book[@name='紅樓夢']/author").text
author_2 = tree.findtext("./book[@name='紅樓夢']/author")
print("紅樓夢作者:", author_1, author_2)
author_3 = root.find("./book/similar[@name='三國志']").get("author")
print("三國志作者:", author_3)
打印結(jié)果
紅樓夢作者: 曹雪芹 曹雪芹
三國志作者: 陳壽
findtext 類似于 find, 直接獲取節(jié)點的文本內(nèi)容
books_1 = root.findall("./book[similar]")
# 對于直接子節(jié)點, 可以省略 ./
books_2 = root.findall("book[similar]")
print(books_1 == books_2)
for book in books_1:
print(book[0].text, book[1].text)
打印結(jié)果
True
吳承恩 明朝
羅貫中 明末清初
以上為個人經(jīng)驗，希望能給大家一個參考，也希望大家多多支持腳本之家。如有錯誤或未考慮完全的地方，望不吝賜教。
您可能感興趣的文章:- Python實現(xiàn)K-means聚類算法并可視化生成動圖步驟詳解
- python3使用迭代生成器實現(xiàn)減少內(nèi)存占用
- python實現(xiàn)由數(shù)組生成對稱矩陣
- python自動化之如何利用allure生成測試報告
- python基于opencv批量生成驗證碼的示例
- 用python自動生成日歷
- python2利用wxpython生成投影界面工具的圖文詳解
- 用Python生成N層的楊輝三角的實現(xiàn)方法
- Python生成九宮格圖片的示例代碼
- python生成隨機數(shù)、隨機字符、隨機字符串的方法示例
- python學(xué)習(xí)之可迭代對象、迭代器、生成器
- python opencv 找出圖像中的最大輪廓并填充(生成mask)
- Python .py生成.pyd文件并打包.exe 的注意事項說明
- python實戰(zhàn)之用emoji表情生成文字
