CWRU的轴承振动数据集读取以及可视化

 

 

import scipy.io as scio
from matplotlib import pyplot as plt

def Raw_data(FilePath):
    data = scio.loadmat(file_path)  # 读取.mat数据
    data_key = list(data.keys())  # 因为mat文件是作为字典储存，所以需要将mat数据字典所有的键并转换为list类型
    print(data_key)
    accl_key = data_key[5]  # 获取'X105_BA_time'键，索引（0，1，2，3，4，5，6）
    accl_data = data[accl_key].flatten() # 获取'X105_BA_time'键在.mat文件所有值，然后将二维数组展平成一维数组
    return accl_data

file_path = r'D:\data\105.mat'
raw_data = Raw_data(file_path)
plt.figure()
plt.plot(raw_data)
print(raw_data)
plt.show()