西安交通大学轴承实验集XJTU-SY_Bearing_Datasets调用函数代码

def csv_read(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]]
    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(1, CSV[CSV_data - 1][CSV_number - 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 = []
    return data_H, data_L
复制

使用方法：
使用前，请添加csv包：

import csv
复制

如果没有这个包就使用pip install csv命令在CMD中安装。

使用语句：

data_1, data_2 = csv_read(3, 5)  # 调用工况40Hz10kN的第五个数据集
复制

一共15个数据集，调用时请更改csv_read里的两个数字：
第一个代表第几个工况（共3种，“35Hz12kN”, “37.5Hz11kN”, “40Hz10kN”）
第二个代表第几个数据集（每种工况5个数据集）
（1,1）代表工况"35Hz12kN"下的第一个数据集
（2.3）代表工况"37.5Hz11kN"下的第三个数据集
以此类推

使用前更改path里的调用路径，我放在E盘下，大家自行更改，两种振动数据分别在data_1和data_2里。