mat文件转excel

大强想进步

于 2023-04-15 16:02:38 发布

阅读量1.6k

点赞数 2

文章标签： excel python 开发语言

本文链接：https://blog.csdn.net/weixin_64651338/article/details/130171306

版权

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位置的相对路径
df.to_excel(output_path, index=False)