pandas的get_dummies进行one-hot编码

最新推荐文章于 2024-11-08 18:29:04 发布

原创最新推荐文章于 2024-11-08 18:29:04 发布 · 231 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#pandas

python 专栏收录该内容

24 篇文章

订阅专栏

本文详细介绍了Pandas库中get_dummies()函数的使用方法，包括如何将分类数据转换为虚拟变量，以及参数data、prefix、columns的用法示例。通过实例展示了在不同场景下如何应用此函数进行数据预处理。

pandas.get_dummies(data, prefix=None, prefix_sep=’_’, dummy_na=False, columns=None, sparse=False, drop_first=False, dtype=None)
说下常用参数
data:的话就是我们要处理的数据
prefix:就是我们制定的前缀，
columns:这是我们直接在原数据集上使用

eg:

data=pd.DataFrame({'a':['x','y','z','y','x'],'b':[1,2,3,1,2]})
data
   a  b
0  x  1
1  y  2
2  z  3
3  y  1
4  x  2

pd.get_dummies(data,prefix='love')
   b  love_x  love_y  love_z
0  1       1       0       0
1  2       0       1       0
2  3       0       0       1
3  1       0       1       0
4  2       1       0       0

pd.get_dummies(data['b'])
   1  2  3
0  1  0  0
1  0  1  0
2  0  0  1
3  1  0  0
4  0  1  0

pd.get_dummies(data,prefix='love',columns=['b'])
   a  love_1  love_2  love_3
0  x       1       0       0
1  y       0       1       0
2  z       0       0       1
3  y       1       0       0
4  x       0       1       0