读取xml数据，绘制直方图和散点图

最新推荐文章于 2025-08-14 09:57:17 发布

原创

最新推荐文章于 2025-08-14 09:57:17 发布 · 770 阅读

1 ·

CC 4.0 BY-SA版权

该博客介绍如何读取XML文件中的坐标数据，通过计算中心点生成'x.txt'和'y.txt'文件，并利用matplotlib库绘制直方图和散点图，展示了数据的分布情况。

import xml.etree.ElementTree as ET
import os
import matplotlib.pyplot as plt


def get_center(filepath,sku):
    num_x = []
    num_y = []
    for root, folder, files in os.walk(filepath):
        for file in files:
            if file.endswith('xml'):
                dir = os.path.join(root,file)
                tree = ET.parse(dir)
                rt = tree.getroot()
                for object1 in rt.findall('object'):
                    sku_name = object1.find('name').text
                    if (sku_name == str(sku)):
                        bndbox1 = object1.find('bndbox')
                        x1 = bndbox1.find('xmin')
                        x2 = bndbox1.find('xmax')
                        y1 = bndbox1.find('ymin')
                        y2 = bndbox1.find('ymax')
                        x = (float(x1.text) + float(x2.text))/2
                        y = (float(y1.text) + float(y2.text)) / 2
                        num_x.append(x)
                        num_y.append(y)