离散型数据编码

最新推荐文章于 2023-08-31 16:50:41 发布

人鱼线

最新推荐文章于 2023-08-31 16:50:41 发布

阅读量717

点赞数 1

分类专栏：机器学习

本文链接：https://blog.csdn.net/qfikh/article/details/103299221

版权

机器学习专栏收录该内容

43 篇文章 7 订阅

订阅专栏

对比LabelEncoder 和OneHot编码：

LabelEncoder编码：

from sklearn import preprocessing
import pandas as pd
df = pd.DataFrame([
            [-1 , 'A'],
            [2   , 'B'],
            [1  , 'A']])
df.columns = ['age',  'class']

le = preprocessing.LabelEncoder()
age_df=le.fit_transform(df['age'])
print(list(le.classes_))
print(age_df)
# 三个类别分别为0 1 2
age_df = pd.get_dummies(age_df, prefix='color_df')
age_df

OneHot编码

import pandas as pd
df = pd.DataFrame([
            [-1 , 'A'],
            [2  , 'B'],
            [1  , 'A']])
df.columns = ['age',  'class']
print(df)
print(pd.get_dummies(df['age'],prefix='age'))