# 3.离散值进行LabelEncoder  #处理数据的三个步骤，去重，处理缺失值，离散值LabelEncoder

from sklearn import preprocessing
from sklearn.externals import joblib

categorical_ix = ["1","2","3"]  #离散值的列名

for i in categorical_ix:
    le = preprocessing.LabelEncoder()
    k = le.fit_transform(df_train[i])
    joblib.dump(le,i+"_LabelEncoder.model")
    df_train[i] = k

for i in categorical_ix:
    le = joblib.load(i+"_LabelEncoder.model")
    k = le.transform(df_train[i])
    df_test[i] = k