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

我不是女神ヾ 2023-10-12 16:06 8阅读 0赞

#### 文章目录 ####

*  一、问题的提出
 *  二、问题的解决
 *   *  2.1 解决问题
     *  2.2 把bboxes写入txt文件中

## 一、问题的提出 ##

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

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写入到指定的输出文件中，并在完成后打印出"写入完成！"的消息。请确保具有适当的文件写入权限，并提供合适的文件名和路径。