python读取文件夹中多个txt文件并写入一个excel工作表

这是使用python写的一个能够读取一个文件夹中的所有txt文件并能够将读取到的数据写入一个excel文件的代码。

import pandas as pd
import os
# 设置文件夹路径
filedir = r'D:\desktop\keyan\loss_2023_05_28_22_26_33'
#设置工作路径
os.chdir(filedir)
# 创建Excel写入器
writer = pd.ExcelWriter('data_4.xlsx', engine='xlsxwriter')
# 获取工作表对象
worksheet = writer.book.add_worksheet('Sheet1')
# 遍历TXT文件列表
for i, file in enumerate(os.listdir(filedir)):
    #判断文件是否是txt文件
    if file.endswith('.txt'):
        # 读取TXT文件
        data = pd.read_csv(file, delimiter='\t', header=None)
        # 获取文件名（不带扩展名）
        filename = os.path.splitext(file)[0]
        # 第一行写入文件名
        worksheet.write(0, i, filename)
        # 将剩余数据写入Excel文件的工作表中，从第二行开始
        for j in range(len(data)):
            worksheet.write(j+1, i, data.iloc[j][0])
# 保存Excel文件
writer.save()