遍历修改文件夹内所有txt文件内第一列内容

遍历修改文件夹内所有txt文件内第一列内容

想法源于：修改数据集标签，将所有的分类（0，1，2，3…）合并为一个大分类（0）

实现效果：

​ 转换前：

​ 转换后：

代码如下：

import os
for root, dirs, files in os.walk(r"/media/xy/XY/VOCdevkit/VOC2007/convert/1"):		# 文件夹路径
    for file in files:
        # 获取文件name
        print(root)
        print(file)
        root1 = str(root+'/'+file) 	# 需要转换文件的路径
        f = open(root1)	# 打开txt文件
        line = f.readlines()	# 以行的形式进行读取文件
        list1 = []
        for i,line in enumerate(line):
            a = line.split()
            b = a[0:5]	# 这是选取需要读取/修改的列；txt文件内一共五列
            list1.append(b)  # 将其添加在列表之中
        f.close()

        root2 = str('/media/xy/XY/VOCdevkit/VOC2007/convert/1ed'+'/'+file)  # 转换后存放文件夹路径+转换后新的txt文件名称
        path_out = root2
        with open(path_out, 'w+') as f_out:
            for i in list1:
                fir = '0'  # 第一列变成0
                sec = i[1]  # 第二、三、四、五列内容保持不变
                third = i[2]
                fourth = i[3]
                fifth = i[4]
                f_out.write(fir + ' ' + str(sec) + ' ' + str(third) + ' ' + str(fourth) + ' ' + str(fifth) + '\n')  # 把前两列写入新的txt文件