首页 > 技术文章 > 数据预处理 | 使用 sklearn.preprocessing.OrdinalEncoder 将分类特征转换为数值型

ykit 2020-03-08 08:41 原文

from sklearn.preprocessing import OrdinalEncoder

model_oe = OrdinalEncoder()

string_data_array = model_oe.fit_transform(string_data)
string_data_pd = pd.DataFrame(string_data_con,columns=string_data.columns)

说明:

1  string_data  是挑出来的 需要转成数值型特征的 分类特征

2  转换后,通常要将 array 类型的结果转成 DataFrame,与其他的特征合并

推荐阅读