Python Apriori算法 载入数据修改 可以读取dataframe

def loadDataSet(dataSet):
    data1=list(map(lambda x:str(x) + '=',list(dataSet.columns))) + 
    dataSet.applymap(str)
    train_data = np.array(data1)#np.ndarray()
    data1_list=train_data.tolist()#list
    X = []
    for LIST1 in data1_list:
        Y = []
        for LIST2 in LIST1:
            if 'nan' not in LIST2:
                Y.append(LIST2)
        X.append(Y)
    return X