python学习——pandas使用get_dummies,对无大小区分分类字段进行one-hot热编码

最新推荐文章于 2025-07-30 21:33:20 发布

如是雨林

最新推荐文章于 2025-07-30 21:33:20 发布

阅读量1.4k

点赞数 1

CC 4.0 BY-SA版权

分类专栏： python 文章标签： pandas

本文链接：https://blog.youkuaiyun.com/qq_23418043/article/details/82556628

python 专栏收录该内容

21 篇文章

订阅专栏

本文介绍如何使用 Pandas 的 get_dummies 方法进行 one-hot 编码，并通过实例展示了如何处理分类数据，如衣服的颜色和大小等。get_dummies 方法能够将分类变量转换为数值形式，便于后续的数据分析与机器学习。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

参考博客：

python中get_dummies实践

pandas使用get_dummies进行one-hot编码

在数据集中，会有一些分类字段，比如衣服的大小(X,XL,M)，衣服的颜色(RED,GREEN),在进行训练的时候，我们肯定要让category变为nummerical表达形式。

对于有大小区分的category，直接map就行了。如下：

df = pd.DataFrame([
            ['green', 'M', 10.1, 'class1'], 
            ['red', 'L', 13.5, 'class2'], 
            ['blue', 'XL', 15.3, 'class1']])
 
df.columns = ['color', 'size', 'prize', 'class label']
 
size_mapping = {
           'XL': 3,
           'L': 2,
           'M': 1}
df['size'] = df['size'].map(size_mapping)

对于颜色这种无法区分大小的数据，我们就使用one-hot编码，pandas自带方法帮我们完成编码。

>>> df = pd.DataFrame({'A': ['a', 'b', 'a'], 'B': ['b', 'a', 'c'],
...                    'C': [1, 2, 3]})
>>> pd.get_dummies(df, prefix=['col1', 'col2'])
   C  col1_a  col1_b  col2_a  col2_b  col2_c
0  1       1       0       0       1       0
1  2       0       1       1       0       0
2  3       1       0       0       0       1

从上面的结果可以看到，热编码是将我们一个分类字段变为多个分类字段，1代表是这个分类，0代表不是这个分类。

最后说明一下：

get_dummies()不会改变原来的df，这个方法返回的是改变后的df，所以如果你想得到改变后的df，就df = pd.get_dummies()