pandas之get_dummies
方法：pandas.get_dummies(data,prefix=None,prefix_sep="_",dummy_na=False,columns=None,sparse=False,drop_first=False)
该方法可以将类别变量转换成新增的虚拟变量/指示变量
参数说明：
-
data：array-like、Series 、 DataFrame ， 输入数据
-
prefix：string、list of strings、dict of strings ，default为None，get_dummies转换后，列名的前缀
-
columns：list-like， default为False，指定需要实现类别转换的列名
-
dummy_na：bool， default为False，增加一列表示空缺值，如果False就忽略空缺值
-
drop_first：fool，default为False ，获取K中的K-1个类别之，去除第一个
举例：
下面通过例子来进一步说明get_dummies()
1、首先构造一个数据列
1
import
pandas as pd
2 s=pd.Series(list(‘abca‘))
34 s_1=pd.get_dummies(s)
1、两个变量：s为Series、s_1为DataFrame
变成→
2、去除第一列
1 b=pd.get_dummies(s,drop_first=True)
得到：
3、查看dummy_na功能
创建数据
1
import
numpy as np
2 s_2=["a","b",np.nan]
结果为：
1 pd.get_dummies(s_2,dummy_na=True)
2 pd.get_dummies(s_2,dummy_na=False)
结果为：
、
4、创建数据框
1 df=pd.DataFrame({
2"A":["a","b","a"],"B":["b","a","c"],
3"C":[1,2,3]})
4 pd.get_dummies(df,prefix=["col_1","col_2"])
结果为：
变为→
实例：
1
import
pandas
2
3 data=pandas.read_csv(
4"C:\Users\Jw\Desktop\python_work\Python数据挖掘实战课程课件\5.2\data1.csv",
5 encoding=‘utf-8‘)
6 7 8 dummyColumns=[‘症状‘,‘职业‘]
910for column in dummyColumns:
11 data[column]=data[column].astype(‘category‘)
121314 dummiesData=pandas.get_dummies(
15 data,
16 columns=dummyColumns,
17 prefix=dummyColumns,
18 prefix_sep=‘‘)
1920 dummiesData=pandas.get_dummies(
21 data,
22 columns=dummyColumns,
23 prefix=dummyColumns,
24 prefix_sep=‘‘,
25 drop_first=True)
构造虚拟变量
drop_first后
实例
步骤：
- 导入数据，设置虚拟变量，将虚拟变量转变为category类，(category变量)类别变量转换成新增的虚拟变量/指示变量
- 通过pandas自带的get_dummies功能，将所有分类扁平化扩张以增加列的形式实现，离散数据按照[0，1]分布。
知识点：
Categorical Type：什么是categorical Type?不知道确切的英文翻译,但是可以按照字面意思来也就是分类数据,比如皮肤的颜色,可以分为黄色,白色,黑色等等,但是这些数据的均值以及数值计算比如加减的结果是没有意义的;但是我们可以将不同的数据分为这几类,在比如人类的性别,男女也属于categorical 类别; 英文中欧冠也可以称之为Nominal Data.
1
import
pandas;
2
3 data = pandas.read_csv(
4"C:\Users\Jw\Desktop\python_work\Python数据挖掘实战课程课件\5.2\data1.csv",
5 encoding=‘utf8‘ 6)
7 8 dummyColumns = [‘症状‘, ‘职业‘]
910for column in dummyColumns:
11 data[column]=data[column].astype(‘category‘)
1213 dummiesData = pandas.get_dummies( #调用get_dummyColumns方法进行不可比较大小虚拟变量的转换14 data,
15 columns=dummyColumns,
16 prefix=dummyColumns,
17 prefix_sep=""18)
1920 dummiesData = pandas.get_dummies(
21 data,
22 columns=dummyColumns,
23 prefix=dummyColumns,
24 prefix_sep="",
25 drop_first=True
26 )
1
#
伯努利贝叶斯
2
from sklearn.naive_bayes import BernoulliNB
3 BNBModel = BernoulliNB()
45 fNames = [‘症状 打喷嚏‘, ‘职业 建筑工人‘, ‘职业 护士‘, ‘职业 教师‘]
6 tData = dummiesData[‘疾病‘]
7 fData = dummiesData[fNames]
89 BNBModel.fit(fData, tData)
上述代码：建模，构造伯努利方程，设置自变量和因变量，训练变量，得到训练集
1
#
病症是打喷嚏的建筑工人
2 newData = pandas.DataFrame({
3‘症状‘:[‘打喷嚏‘],
4‘职业‘:[‘建筑工人‘]
5})
6 7for column in dummyColumns:
8 newData[column] = newData[column].astype(
9‘category‘,
10 categories=data[column].cat.categories
11 )
1213 dummiesNewData = pandas.get_dummies(
14 newData,
15 columns=dummyColumns,
16 prefix=dummyColumns,
17 prefix_sep="",
18 drop_first=True
19)
2021 pData = dummiesNewData[fNames]
22 BNBModel.predict(pData)
训练：
1
#
病症是打喷嚏的建筑工人
2 newData=pandas.DataFrame({
3‘症状‘:[‘打喷嚏‘],
4‘职业‘:[‘建筑工人‘]})
5 6for column in dummyColumns:
7 newData[column]=newData[column].astype(
8‘category‘,
9 categories=data[column].cat.categories
10)
1112 dummiesNewData=pandas.get_dummies(
13 newData,
14 columns=dummyColumns,
15 prefix=dummyColumns,
16 prefix_sep=‘‘,
17 drop_first=True)
1819 pData=dummiesNewData[fNames]
20 BNBModel.predict(pData)
原文：https://www.cnblogs.com/U940634/p/9745737.html
【说明】：本文章由站长整理发布，文章内容不代表本站观点，如文中有侵权行为，请与本站客服联系(QQ:254677821)！