西储大学轴承数据读取程序python

西储大学（CWRU）轴承数据集是机械故障诊断中广泛使用的公开数据集，通常包含正常、内圈故障、外圈故障和滚动体故障等多种工况下的振动信号数据。这些数据通常以 `.mat` 文件格式提供，可以通过 MATLAB 生成，也可在 Python 中使用 `scipy.io` 模块进行读取。以下是一个用于读取 CWRU 轴承数据集 `.mat` 文件的 Python 程序示例： ```python import scipy.io import numpy as np # 指定.mat文件路径 file_path = '97.mat' # 示例文件名，根据实际文件名修改 # 使用scipy.io加载.mat文件 mat_data = scipy.io.loadmat(file_path) # 查看文件中包含的变量名 print("Variables in .mat file:") for key in mat_data.keys(): if not key.startswith('__'): # 过滤系统变量 print(f"{key}: {mat_data[key].shape}") # 读取某一通道的振动信号数据（假设变量名为'X97_DE_time'） if 'X97_DE_time' in mat_data: vibration_signal = mat_data['X97_DE_time'].flatten() # 展平为一维数组 print(f"Vibration signal shape: {vibration_signal.shape}") print(f"First 10 samples: {vibration_signal[:10]}") else: print("指定的变量名不存在于.mat文件中。") ``` 该程序使用 `scipy.io.loadmat` 方法加载 `.mat` 文件，并输出文件中包含的变量名及其维度。随后，它读取某一通道的振动信号数据（如 `X97_DE_time`），并将其展平为一维数组以便后续处理，例如时域分析、频域变换或使用深度学习模型进行分类。在实际应用中，可能需要根据具体的数据文件名和变量名进行相应调整。此外，如果数据集是以 `.csv` 或 `.txt` 格式提供的，可以使用 `pandas` 或 `numpy` 的 `loadtxt` 方法进行读取。

创作声明：本文部分内容由AI辅助生成（AIGC），仅供参考

下一篇 python df.iterrows读取excel时，获取excel的行数和列数