码迷,mamicode.com
首页 > 其他好文 > 详细

独热编码

时间:2019-01-12 13:21:15      阅读:331      评论:0      收藏:0      [点我收藏+]

标签:read   das   array   roc   样本   img   proc   写入文件   gen   

独热编码OneHotEncoder可以将分类字符串变量数值化,例如:

技术分享图片

这里有三个样本,每个样本有三个特征,我们可以用独热编码将其转换成数字,代码如下

from sklearn import preprocessing
X=[[male,USA,firefox],[female,UK,google],[male,China,ie]]
enc=preprocessing.OneHotEncoder()
enc.fit(X) #训练模型
print(enc.transform([[male,USA,google],[female,UK,firefox]]).toarray())#测试集得出结果

运行结果如下可见三个特征并不是转换成三个数字,而是一个特征转换成了多个数字

技术分享图片

接下来看看从文件中导入离散数据并将这些离散特征通过独热编码数值化,关键是要先把数据存入一个二维矩阵中(原文件数据中第一行是变量名称,否则会少一行)。

import pandas as pd
import numpy as np
from sklearn import preprocessing
from pandas import DataFrame
data=pd.read_excel(r"C:\Users\Liugengxin\Desktop\课程\数据科学\大作业\小数据集测试\离散.xlsx") #这里用pandas的read_excel方法将数据读入
arr=np.array(data) #将data转换为数组放在arr中
enc=preprocessing.OneHotEncoder()
enc.fit(arr) #训练模型
data=enc.transform(arr).toarray() #将测试集用该模型转换成数字
DataFrame(data).to_excel(rc:\Users\Liugengxin\Desktop\2.xlsx)  #将结果写入文件中

 

独热编码

标签:read   das   array   roc   样本   img   proc   写入文件   gen   

原文地址:https://www.cnblogs.com/Liu269393/p/10259124.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!