对西安交大轴承数据集XJTU-SY_Bearing_Datasets进行读取和处理:
读取交大全寿命数据并显示
1.python
读取任意一个工况里的任意一个轴承数据的任意文件csv
#读取数据集的CSV文件并显示
import csv
import matplotlib.pyplot as plt
def csv_read(CSV_data,CSV_number):#第CSV_data工况,第CSV_number组轴承
data_csv = []
data_H = []#水平传感器测量的诊断数据的数组
data_L = []#垂直传感器测量的诊断数据的数组
CSV = [[123, 161, 158, 122, 52], [491, 161, 533, 42, 339], [2538, 2496, 371, 1515, 114]]#3种工况分别都有5个轴承,CSV数据集样本总数
CSV_path = ["", "35Hz12kN", "37.5Hz11kN", "40Hz10kN"]
#样本数
# 35Hz12kN 1 1-123 2-161 3-158 4-122 5-52
# 37.5Hz11kN 2 1-491 2-161 3-533 4-42 5-339
# 40Hz10kN 3 1-2538 2-2496 3-371 4-1515 5-114
path = "E://故障诊断//XJTU-SY_Bearing_Datasets//Data//XJTU-SY_Bearing_Datasets//" + CSV_path[CSV_data] + "//Bearing" + str(CSV_data) + "_" + str(CSV_number) + "//"
print(path)
for i in range(50,CSV[CSV_data-1][CSV_number-1]):#二维数组从0开始,显示部分周期修改range(里的1)
csv_data=csv.reader(open(path+"%d.csv"% i,"r"))
for list in csv_data:
data_csv.append(list)
for j in range(1, len(data_csv)):
data_H.append(float(data_csv[j][1]))
data_L.append(float(data_csv[j][0]))
data_csv = []
retu