修改txt文件内容每行第一列

修改文件夹下所有txt文件的每一行的第一列, 其它内容不变:
修改txt文件内容每行第一列_第1张图片
转为为:
在这里插入图片描述

import os
import random
import numpy as np
from numpy import *

txtfilepath = r"D:\DataSet\Human\train\labels" #原始txt文件所存文件夹,文件夹可以有一个或多个txt文件
savefilepath = r"D:\DataSet\Human\train\labels" #更改后txt文件存放的文件夹
total_txt = os.listdir(txtfilepath) # 返回指定的文件夹包含的文件或文件夹的名字的列表
num = len(total_txt)
list = range(num) #创建从0到num的整数列表
files = os.listdir(savefilepath)

for file in total_txt:
    filepath = os.path.join(txtfilepath, file)
    savepath = os.path.join(savefilepath, file)
    b = []
    with open(filepath, 'r', encoding='utf8') as f:
        print(f)
        while True:
            _ = f.readline()
            print(_)

            if _:
                b.append(_.split(' '))
            else:
                break

        for i in range(len(b)):
            print(b[i])
            if b[i][0] == '6':
                b[i][0] = '0'
            # elif b[i][0] == '1':
            #     b[i][0] = '15'
    with open(savepath, 'w+', encoding='utf8') as f:
        for i in b:
            f.writelines(' '.join(i))

你可能感兴趣的:(机器学习,numpy,python,机器学习)