os.listdir()读取文件夹下特定命名的文件并合并保存

本文介绍了如何使用Python的pandas库和tqdm库从D:/code/data/Learning_set/Bearing1_1文件夹中逐个读取并合并acc.csv和temp.csv文件,最终将结果分别保存到output_file_1和output_file_2中,处理过程中遇到的ParserError也被妥善处理。
摘要由CSDN通过智能技术生成
import pandas as pd
from tqdm import tqdm
import os
# 合并振动信号的所有数据
path = 'D:/code/data/Learning_set/Bearing1_1'
acc_csv_files = os.listdir(path)

acc_data = pd.DataFrame()
temp_data = pd.DataFrame()
# 逐个读取并合并CSV文件
# tqdm的作用是显示进度条,也可以不加tqdm
for file in tqdm(acc_csv_files):
    try:
        if 'acc' in file:
            df1 = pd.read_csv(path+'/'+file, header=None)
            acc_data =  pd.concat([acc_data, df1], ignore_index=True)
        else:
            df2 = pd.read_csv(path+'/'+file, header=None)
            temp_data =  pd.concat([temp_data, df2], ignore_index=True)
    except pd.errors.ParserError as e:
        print(f"解析文件{file}时出错:{e}")

# 指定保存文件夹
output_file_1 = r"./data/acc_data_bearing1_1.csv"
acc_data.to_csv(output_file_1, index=False)
output_file_2 = r"./data/temp_data_bearing1_1.csv"
temp_data.to_csv(output_file_2, index=False)

D:/code/data/Learning_set/Bearing1_1文件夹下的文件是:既有“acc***.csv”,又有“temp***.csv”文件,我想把两种数据分别整合并存在一个csv文件里。
在这里插入图片描述

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值