python中get_dummies()用法示例详解

 更新时间:2025年09月28日 09:21:35   作者:sofaraway13  
get_dummies函数是Python中的一个函数,用于将分类变量转换为哑变量,下面这篇文章主要介绍了python中get_dummies()用法的相关资料,文中通过代码介绍的非常详细,需要的朋友可以参考下

1. 基本作用

pandas.get_dummies() 用于将 分类变量(categorical variable) 转换为 哑变量/指示变量(Dummy Variables),也叫 独热编码(One-Hot Encoding)

👉 通俗点说:把字符串或类别列拆分成若干个二进制列(0 或 1),机器学习更容易处理。

2. 语法

pd.get_dummies(data, prefix=None, prefix_sep="_", dummy_na=False, columns=None, drop_first=False)

常用参数:

  • data:要转换的数据(Series 或 DataFrame)

  • prefix:新生成列的前缀

  • prefix_sep:前缀与值之间的分隔符(默认 _

  • dummy_na:是否为缺失值生成一列(默认 False)

  • columns:指定要转换的列(DataFrame 时用)

  • drop_first:是否删除第一个类别(避免多重共线性,回归时常用)

3. 示例讲解

(1)对 Series 使用

import pandas as pd

s = pd.Series(["red", "green", "blue", "red"])
dummies = pd.get_dummies(s)
print(dummies)

👉 输出:

   blue  green  red
0     0      0    1
1     0      1    0
2     1      0    0
3     0      0    1

(2)对 DataFrame 中的列使用

df = pd.DataFrame({
    "id": [1, 2, 3, 4],
    "color": ["red", "green", "blue", "red"],
    "size": ["S", "M", "L", "M"]
})

# 对 color 列进行独热编码
df_dummies = pd.get_dummies(df, columns=["color"])
print(df_dummies)

👉 输出:

   id size  color_blue  color_green  color_red
0   1    S           0            0          1
1   2    M           0            1          0
2   3    L           1            0          0
3   4    M           0            0          1

(3)加前缀

df_dummies = pd.get_dummies(df, columns=["size"], prefix="SIZE")
print(df_dummies)

👉 输出:

   id color  SIZE_L  SIZE_M  SIZE_S
0   1   red       0       0       1
1   2 green       0       1       0
2   3  blue       1       0       0
3   4   red       0       1       0

(4)避免虚拟变量陷阱(drop_first=True)

df_dummies = pd.get_dummies(df, columns=["color"], drop_first=True)
print(df_dummies)

👉 输出(少一列,避免多重共线性):

   id size  color_green  color_red
0   1    S            0          1
1   2    M            1          0
2   3    L            0          0
3   4    M            0          1

✅ 总结:

  • get_dummies() → 分类变量转 0/1 矩阵

  • columns 指定哪些列编码

  • prefix/prefix_sep 控制列名

  • drop_first=True → 回归模型常用,避免虚拟变量陷阱

总结 

到此这篇关于python中get_dummies()用法的文章就介绍到这了,更多相关python get_dummies()用法内容请搜索脚本之家以前的文章或继续浏览下面的相关文章希望大家以后多多支持脚本之家!

相关文章

  • Python使用LSTM实现销售额预测详解

    Python使用LSTM实现销售额预测详解

    大家经常会遇到一些需要预测的场景,比如预测品牌销售额,预测产品销量。本文给大家分享一波使用 LSTM 进行端到端时间序列预测的完整代码和详细解释,需要的可以参考一下
    2022-07-07
  • Python中往列表中插入字典时,数据重复问题

    Python中往列表中插入字典时,数据重复问题

    这篇文章主要介绍了Python中往列表中插入字典时,数据重复问题,具有很好的参考价值,希望对大家有所帮助,如有错误或未考虑完全的地方,望不吝赐教
    2024-02-02
  • 5行Python代码实现图像分割的步骤详解

    5行Python代码实现图像分割的步骤详解

    这篇文章主要介绍了5行Python代码实现图像分割的步骤详解,文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需要的朋友们下面随着小编来一起学习学习吧
    2020-05-05
  • Python如何调用外部系统命令

    Python如何调用外部系统命令

    这篇文章主要介绍了Python如何调用外部系统命令,文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需要的朋友可以参考下
    2019-08-08
  • Python中list初始化方法示例

    Python中list初始化方法示例

    这篇文章主要介绍了Python中list初始化方法,分析了list初始化常用的方法与相关使用注意事项,需要的朋友可以参考下
    2016-09-09
  • python中bisect模块用法实例

    python中bisect模块用法实例

    这篇文章主要介绍了python中bisect模块用法实例,以实例形式介绍了bisect模块中几种常见函数的用法,非常具有实用价值,需要的朋友可以参考下
    2014-09-09
  • python之multimethod包多分派解读

    python之multimethod包多分派解读

    这篇文章主要介绍了python之multimethod包多分派问题,具有很好的参考价值,希望对大家有所帮助,如有错误或未考虑完全的地方,望不吝赐教
    2023-08-08
  • Python Black代码格式化终极指南

    Python Black代码格式化终极指南

    Black是一款自动化的Python代码格式化工具,旨在通过强制一致的代码格式来提高代码的可读性和维护性,本文将深入探讨Black的使用方法、高级特性以及与其他格式化工具的比较,帮助你更好地理解并成功应用Black在你的项目中
    2024-01-01
  • 从入门到实战详解Python解析Excel的完整指南

    从入门到实战详解Python解析Excel的完整指南

    在日常工作中,Excel是存储和分析数据的常用工具,本文将介绍如何用Python解析Excel,并通过实战案例带你快速上手,感兴趣的小伙伴可以跟随小编一起学习一下
    2026-04-04
  • 使用Python实现计算DICOM图像两点真实距离

    使用Python实现计算DICOM图像两点真实距离

    这篇文章主要为大家详细介绍了如何使用Python实现计算DICOM图像两点真实距离,文中的示例代码讲解详细,感兴趣的小伙伴可以跟随小编一起学习一下
    2023-11-11

最新评论