特征工程之字典数据特征抽取

击上方

“蓝色字”

可关注我们！

今日分享：字典数据特征抽取

一：sklearn 特征抽取API

sklearn库提供了特征抽取模块 sklearn.feature_extraction 对数据进行特征值化，包括字典数据及文本数据的特征抽取。

二：字典数据特征抽取

DictVectorizer类

DictVectorizer语法

三：字典数据特征抽取流程

实例化类 DictVectorizer()

调用 fit_transform() 方法输入数据并转换

四：代码演示

运行结果

从结果可以看出，数据转换后，特征数量变成了四个，原始数据中的三个不同的城市分别作为一个特征进行one-hot编码（对每一个类别生成一个布尔列，以保证每一列中只有一个样本的值取1），而 temperature 特征本身已是数值型，所以不再转换。

公众号QQ群

扫QQ群二维码进交流学习群