【跑实验02】如何提取名称相同的部分，比如obj365_train_000000000002.xml，换成相应的坐标格式

一、问题的提出

我有一个文件格式如下所示：

obj365_train_000000000002.xml: 236, 41, 263, 180
obj365_train_000000000002.xml: 218, 174, 248, 207
obj365_train_000000000002.xml: 306, 10, 332, 175
obj365_train_000000000002.xml: 396, 0, 426, 162
obj365_train_000000000002.xml: 189, 281, 528, 512
obj365_train_000000000002.xml: 255, 293, 388, 512
obj365_train_000000000002.xml: 193, 285, 302, 476
obj365_train_000000000002.xml: 146, 280, 213, 445
obj365_train_000000000002.xml: 339, 113, 450, 229
obj365_train_000000000002.xml: 406, 49, 508, 177
obj365_train_000000000002.xml: 499, 82, 537, 229
obj365_train_000000000002.xml: 521, 0, 683, 167
obj365_train_000000000002.xml: 5, 281, 88, 389
obj365_train_000000000002.xml: 162, 275, 212, 366
obj365_train_000000000002.xml: 10, 79, 67, 235
obj365_train_000000000002.xml: 142, 97, 196, 234
obj365_train_000000000002.xml: 325, 275, 533, 411
obj365_train_000000000002.xml: 65, 82, 154, 165
obj365_train_000000000002.xml: 462, 251, 488, 285
obj365_train_000000000002.xml: 422, 277, 502, 286
obj365_train_000000000002.xml: 501, 254, 512, 290
obj365_train_000000000002.xml: 511, 255, 521, 288
obj365_train_000000000002.xml: 532, 151, 676, 451
......

以上只是部分内容。

我如何提取名称相同的部分，比如obj365_train_000000000002.xml，换成下面的坐标形式：

bboxes = [
    (236, 41, 263, 180), (218, 174, 248, 207), (306, 10, 332, 175), (396, 0, 426, 162),
    (189, 281, 528, 512), (255, 293, 388, 512), (193, 285, 302, 476),
    (146, 280, 213, 445), (339, 113, 450, 229), (406, 49, 508, 177), (499, 82, 537, 229),
    (521, 0, 683, 167), (5, 281, 88, 389), (162, 275, 212, 366), (10, 79, 67, 235),
    (142, 97, 196, 234), (325, 275, 533, 411), (65, 82, 154, 165), (462, 251, 488, 285),
    (422, 277, 502, 286), (501, 254, 512, 290), (511, 255, 521, 288), (532, 151, 676, 451),
    (61, 171, 163, 198), (64, 268, 166, 291), (83, 291, 166, 377), (502, 254, 520, 283),
    (11, 250, 32, 273), (12, 272, 34, 282), (288, 284, 332, 301)
]

二、问题的解决

2.1 解决问题

可以使用Python的字典（Dictionary）来实现这个任务。首先，需要遍历给定的数据，并将相同名称的边界框坐标存储在字典中。下面是一个示例代码：

data = [
    "obj365_train_000000000002.xml: 236, 41, 263, 180",
    "obj365_train_000000000002.xml: 218, 174, 248, 207",
    "obj365_train_000000000002.xml: 306, 10, 332, 175",
    "obj365_train_000000000002.xml: 396, 0, 426, 162",
    # 剩下的数据...
]

bboxes = {}  # 创建一个空字典用于存储边界框坐标

for item in data:
    filename, bbox = item.split(":")  # 拆分文件名和边界框坐标
    filename = filename.strip()  # 去除文件名中的空格
    coords = tuple(map(int, bbox.split(",")))  # 将边界框坐标转换为整数元组
    if filename in bboxes:
        bboxes[filename].append(coords)  # 如果字典中已存在该文件名，则将坐标添加到对应的列表中
    else:
        bboxes[filename] = [coords]  # 否则，在字典中创建新的键，并将坐标作为列表的第一个元素

# 输出结果
for filename, coords_list in bboxes.items():
    print(filename)
    for coords in coords_list:
        print(coords)

运行以上代码后，将输出名称相同的部分以及对应的坐标形式。请注意，代码假设给定的数据格式是准确的，并且每个文件名都有相应的边界框坐标。

但是我们的data是一个文件，我们需要换掉：

如果数据存储在一个文本文件中，您可以使用Python的文件操作来读取文件内容。以下是相应的代码示例：

filename = 'data.txt'  # 指定数据文件名

bboxes = {}  # 创建一个空字典用于存储边界框坐标

with open(filename, 'r') as file:
    for line in file:
        item = line.strip()  # 去除行末尾的换行符和空格
        filename, bbox = item.split(":")  # 拆分文件名和边界框坐标
        filename = filename.strip()  # 去除文件名中的空格
        coords = tuple(map(int, bbox.split(",")))  # 将边界框坐标转换为整数元组
        if filename in bboxes:
            bboxes[filename].append(coords)  # 如果字典中已存在该文件名，则将坐标添加到对应的列表中
        else:
            bboxes[filename] = [coords]  # 否则，在字典中创建新的键，并将坐标作为列表的第一个元素

# 输出结果
for filename, coords_list in bboxes.items():
    print(filename)
    for coords in coords_list:
        print(coords)

请将文件名指定为适当的路径和文件名，并确保文件存在于该路径下。代码将逐行读取文件内容，并按照之前提到的方式提取和存储边界框坐标。最后，它会打印输出相同文件名的部分以及对应的坐标形式。

2.2 把bboxes写入txt文件中

要将bboxes写入到文本文件中，我们可以使用Python的文件操作来完成。以下是相应的代码示例：

filename = 'output.txt'  # 指定输出文件名

with open(filename, 'w') as file:
    for filename, coords_list in bboxes.items():
        file.write(f"{filename}:\n")
        for coords in coords_list:
            file.write(f"{', '.join(str(coord) for coord in coords)}\n")

print("写入完成！")

上述代码将迭代bboxes字典中的键值对，并将文件名和对应的边界框坐标写入到指定的输出文件中。每个文件名后面的冒号（:）用于分隔文件名和边界框坐标。每个坐标都写入到新的一行中，用逗号分隔。请将filename设置为您想要保存输出的路径和文件名。

运行代码后，它将将bboxes写入到指定的输出文件中，并在完成后打印出"写入完成！"的消息。请确保具有适当的文件写入权限，并提供合适的文件名和路径。