LabelEncoder 自然数编码

September 1, 2019 · 机器学习 · 564次阅读

利用自然数给一种特征进行数值化,比如:
有一个地区特征:[重庆,湖北,湖南]
自然数编码将其数值化则为:[0,1,2],代码example如下:

第一步,导入相关库

import pandas as pd
import numpy as np
from sklearn.preprocessing import LabelEncoder
地区喜好
重庆
黑龙江不辣
湖南特辣
贵州

第二步,导入数据集

fruit_data = pd.DataFrame({
    '地区':  ['重庆','黑龙江','湖南','贵州'],
    '喜好':  ['辣','不辣','特辣','辣'],
})

第三步 自然编码

fruit_data[['fruit','color']]=fruit_data[['fruit','color']].apply(LabelEncoder().fit_transform)
地区喜好
02
11
20
32

标签:机器学习

最后编辑于:2019/09/28 18:45

添加新评论

控制面板