python批量修改txt文件内容以更换标签编号

import os
import argparse

# path = 'img0/labels/'
# path_list = os.listdir(path)

def main(path_list):
    for name in path_list:
        line_data = ""
        with open(path+name, 'r', encoding='unicode_escape') as c2:
            for line in c2.readlines():
                data = line.split(' ')
                if data[0] == '15':
                    data[0] = '0'
                if data[0] == '16':
                    data[0] = '4'

                for i in range(len(data)):
                    if i == len(data) - 1:
                        line_data += data[i]
                    else:
                        line_data += data[i] + ' '

            with open(path+name, 'w') as c3:
                c3.write(line_data)
            c3.close()
        c2.close()

if __name__ == '__main__':
    parser = argparse.ArgumentParser(description='test')
    parser.add_argument('--label-dir')
    args = parser.parse_args()
    path = args.label_dir
    path = path+'/labels/'
    path_list = os.listdir(path)
    main(path_list)

yolov5好几个人分开标的，每个人的标签编号都不一样，批量修改一下，主要是先打开只读再打开只写这一步有点儿麻烦