用 python 将文件夹内的 .txt 文件存储到 .csv 文件

本文记录了利用python将.txt文件的数据集 label 写入到一个.csv文件中。

文件夹内文件如下：

txt文件如下图所示，每行代表一个坐标（x，y）共有19个点，将所有点按顺序存储到csv文件中的一行中，第一列为txt文件的文件名。

存储后的csv文件如下：

代码如下：

import os
import csv

data_path = './AnnotationsByMD/400_junior/'
data_names = os.listdir(data_path)

with open('400_junior.csv', 'a', newline='') as csvfile:
    writer = csv.writer(csvfile)
    for file in data_names:   # 遍历文件夹中的txt文件
        file_path = os.path.join(data_path, file)
        data_list = []
        data_list.append(file)   # 将文件名存到第一列
        with open(file_path,'r') as f: 
            lines=f.readlines()  # 按行读取文件内容
            for line in lines:   
                xy = line.strip("\n").split(",")
                if len(xy) > 1:
                    data_list.append(xy[0])  # 提取txt文件每行的第一个元素
                    data_list.append(xy[1])  # 提取txt文件每行的第二个元素
                else:
                    data_list.append(xy[0]) 
        writer.writerow(data_list)