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文件里。