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

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文件里。
os.listdir()读取文件夹下特定命名的文件并合并保存_第1张图片

你可能感兴趣的:(python,pandas,开发语言)