西储大学轴承数据集mat转excel代码python_西储大学轴承python

import os
import scipy.io
import pandas as pd
import numpy as np

# 读取数据文件

data_dir = 'C:/Users/喃文强/Desktop/新建文件夹/Normal Baseline Data'   #要导出excel的存放文件位置
file_name = 'C:/Users/喃文强/Desktop/cwru/Normal Baseline Data/99.mat'   #原mat文件位置
data = scipy.io.loadmat(os.path.join(data_dir, file_name))

# 将数据转换为 DataFrame
dfs = []
for key in data.keys():
    if isinstance(data[key], np.ndarray):
        if data[key].ndim == 2:
            df = pd.DataFrame(data[key][:,0], columns=[key])
        else:
            df = pd.DataFrame(data[key][0], columns=[key])
        dfs.append(df)
df = pd.concat(dfs, axis=1)

# 将 DataFrame 写入 Excel 文件
output_file_name ='99.xlsx'
output_path = os.path.join(os.path.expanduser("~"), 'Desktop', '新建文件夹/12k Fan End Bearing Fault Data', output_file_name)
#新建文件夹原本保存在新建文件夹中，但是我想在分类的话在新建文件夹中又建了个12k Drive End Bearing Fault Data
# 上一行代码变为'新建文件夹/12k Drive End Bearing Fault Data',即要保存excel位置的相对路径。
# 如果分类不同每个文件夹的相对路径要随，生成的excel存放位置data_dir = 'C:/Users/喃文强/Desktop/新建文件夹/Normal Baseline Data'   改变
df.to_excel(output_path, index=False)