Python读取txt文件并保存为csv格式

import xlwt
import numpy as np
import os
import re

def read_txt(path):

    filelist = os.listdir(path)
    file_names = np.array([file for file in filelist if file.endswith('.txt')], dtype=object)

    return file_names

workbook = xlwt.Workbook(encoding = 'utf-8')
path = './data_test/predictions_manual'

sheet1 = workbook.add_sheet('result', cell_overwrite_ok=True)
sheet1_row = read_txt(path)
row = 1
for i in range(len(sheet1_row)):
    paths = path + '/' +sheet1_row[i]
    f = open(paths)
    name = paths[31:-4]
    lines = f.readlines()
    for line in lines:
        index = 1
        data = line.strip().split(" ")
        for val in data:
            val = val.split(' ')
            val = [x for x in val if x != '']
            for key in val:
                #key = str(key)
                if index > 3:
                    sheet1.write(row, index, int(key))
                if index == 3:
                    sheet1.write(row, index, float(key))
                if index < 3:
                    sheet1.write(row, index, str(key))
                index += 1
        sheet1.write(row, index, str(name))
        row += 1
workbook.save('test.csv')

写入txt文件部分

                    if save_txt:  # Write to file
                        xywh = (xyxy2xywh(torch.tensor(xyxy).view(1, 4)) ).view(-1).tolist()  # normalized xywh
                        with open(txt_path + '.txt', 'a') as f:
                            #识别类别
                            f.write(str(cfg.textnames[int(cls)])+' ')
                            f.write(str(obj_conf)+' ')
                            f.write(('%g ' * 5 + '\n') % (cls, *xyxy))  # label format