如何在Python中使用Sklearn对类别型特征进行高效的一热编码?

陌敏大大_8908

陌敏大大_8908

2026-07-31

830人浏览

原创

onehotencoder报“expected 2d array, got 1d array”错误,因其严格要求输入为二维数组(shape为(n_samples, n_features)),而df['color']返回一维series,正确做法是用df[['color']]或.values.reshape(-1, 1)。

如何在python中使用sklearn对类别型特征进行高效的一热编码?

为什么 OneHotEncoder 会报错 “Expected 2D array, got 1D array”?

因为 OneHotEncoder 默认只接受二维输入(即 shape 为 (n_samples, n_features)),而你直接传入一维的 Series 或 list(比如 df['color'])时,它会拒绝处理。

正确做法是把单列转成二维:用 .values.reshape(-1, 1) 或 [[x] for x in col],但更稳妥的是始终用 pd.DataFrame 包一层:

from sklearn.preprocessing import OneHotEncoder
encoder = OneHotEncoder(sparse_output=False)  # 注意新版默认 sparse_output=True
encoded = encoder.fit_transform(df[['color']])  # ✅ 必须是双括号
  • 别写 df['color'] —— 这是 Series,一维
  • 别依赖 reshape 后再传,容易漏掉 dtype 转换(比如 object → str)
  • 如果列含 NaN,OneHotEncoder 默认会报错;加参数 handle_unknown='ignore' 和 drop=None 之前先确认是否真要保留 NaN 列

pd.get_dummies() 和 OneHotEncoder 选哪个?

简单场景(训练/预测都在同一套类别内、不涉及 pipeline 部署)直接用 pd.get_dummies() 更快、更直观;但只要涉及模型上线、新数据预测或需要和 ColumnTransformer 配合,必须用 OneHotEncoder。

关键差异点:

  • pd.get_dummies(df, columns=['color']) 生成列名如 color_red,可读性强;OneHotEncoder 默认列名是数字索引,需配合 encoder.get_feature_names_out() 手动还原
  • pd.get_dummies() 对测试集里出现的新类别(比如新颜色 'purple')直接新增列,导致训练/预测维度不一致;OneHotEncoder 设置 handle_unknown='ignore' 后,新类别对应全零向量,维度稳定
  • OneHotEncoder 支持 drop='first' 做哑变量编码防共线性;pd.get_dummies(drop_first=True) 也能做到,但无法跨数据集复用编码逻辑

如何用 ColumnTransformer 对混合类型特征统一编码?

真实数据总有数值列+类别列混在一起,硬拆再拼接容易错位。用 ColumnTransformer 把编码逻辑锁死在预处理流程里:

python-parallelization
python-parallelization

将顺序 Python 代码转换为并行/并发实现,用于在需要并行化 Python 代码、提升性能时使用。

下载
from sklearn.compose import ColumnTransformer
from sklearn.preprocessing import StandardScaler
<p>preprocessor = ColumnTransformer(
transformers=[
('num', StandardScaler(), ['age', 'income']),
('cat', OneHotEncoder(handle_unknown='ignore'), ['city', 'gender'])
],
remainder='passthrough'  # 不参与变换的列(如 ID)原样保留
)</p><p>X_processed = preprocessor.fit_transform(X_train)</p>
  • 注意 remainder='passthrough' 是默认行为,但显式写出更安全;设为 'drop' 会丢弃未声明列,容易误删
  • OneHotEncoder 的 handle_unknown='ignore' 必须在 fit 前设定,否则 transform 遇到未知类别仍会崩
  • 如果类别列含空值,OneHotEncoder 在 fit 时会把 np.nan 当作一个合法类别(生成一列 x0_nan),除非你提前用 fillna() 处理

为什么编码后特征数量爆炸?怎么控制?

当某列有几百个唯一值(比如用户 ID、商品 SKU),OneHotEncoder 会生成同等数量的列,内存暴增且无泛化能力。这不是 bug,是设计使然 —— 它本就不该用于高基数类别特征。

应对方式不是调参,而是换方法:

  • 高基数列(>10–20 个唯一值)优先考虑 TargetEncoder(来自 category_encoders 库)或频率编码(value_counts().to_dict() 映射)
  • 想强行用 one-hot?先做截断:df['city'].value_counts().index[:10] 取 Top-N,其余归为 'other'
  • OneHotEncoder 的 min_frequency 参数(v1.3+)可自动合并低频项,但只支持整数阈值,且要求输入是字符串或类别型(astype('category'))

真正麻烦的从来不是怎么编码,而是没在 EDA 阶段看清每列的唯一值分布 —— 一热编码只是工具,不是万能解药。

Python免费学习笔记(深入):立即使用
在学习笔记中,你将探索 Python 的核心概念和高级技巧!

相关专题

更多
python打包成可执行文件
python打包成可执行文件

本专题为大家带来python打包成可执行文件相关的文章,大家可以免费的下载体验。

2023.07.20

1751

4

python能做什么
python能做什么

python能做的有:可用于开发基于控制台的应用程序、多媒体部分开发、用于开发基于Web的应用程序、使用python处理数据、系统编程等等。本专题为大家提供python相关的各种文章、以及下载和课程。

2023.07.25

4484

7

format在python中的用法
format在python中的用法

Python中的format是一种字符串格式化方法,用于将变量或值插入到字符串中的占位符位置。通过format方法,我们可以动态地构建字符串,使其包含不同值。php中文网给大家带来了相关的教程以及文章,欢迎大家前来阅读学习。

2023.07.31

1729

3

python教程
python教程

Python已成为一门网红语言,即使是在非编程开发者当中,也掀起了一股学习的热潮。本专题为大家带来python教程的相关文章,大家可以免费体验学习。

2023.08.03

26417

23

python环境变量的配置
python环境变量的配置

Python是一种流行的编程语言,被广泛用于软件开发、数据分析和科学计算等领域。在安装Python之后,我们需要配置环境变量,以便在任何位置都能够访问Python的可执行文件。php中文网给大家带来了相关的教程以及文章,欢迎大家前来学习阅读。

2023.08.04

3207

5

python eval
python eval

eval函数是Python中一个非常强大的函数,它可以将字符串作为Python代码进行执行,实现动态编程的效果。然而,由于其潜在的安全风险和性能问题,需要谨慎使用。php中文网给大家带来了相关的教程以及文章,欢迎大家前来学习阅读。

2023.08.04

3227

5

scratch和python区别
scratch和python区别

scratch和python的区别:1、scratch是一种专为初学者设计的图形化编程语言,python是一种文本编程语言;2、scratch使用的是基于积木的编程语法,python采用更加传统的文本编程语法等等。本专题为大家提供scratch和python相关的文章、下载、课程内容,供大家免费下载体验。

2023.08.11

1183

5

python合并两个列表
python合并两个列表

Python是一种强大的编程语言,具有许多方便的功能和工具。在Python中,有多种方法可以合并两个列表。php中文网给大家带来了相关的教程以及文章,欢迎大家前来学习阅读。

2023.08.10

596

4

python是前端还是后端
python是前端还是后端

Python属于前端也属于后端,其灵活性和丰富的生态系统使得开发人员能够在不同的领域中灵活运用。本专题为大家提供python相关的文章、下载、课程内容,供大家免费下载体验。

2023.08.11

2503

5

热门下载

更多
网站特效
/
网站源码
/
网站素材
/
前端模板

精品课程

更多
相关推荐
/
热门推荐
/
最新课程
Uvicorn官方安装与入门手册
Uvicorn官方安装与入门手册

共0课时 | 0人学习

Python uv官方项目管理指南
Python uv官方项目管理指南

共0课时 | 0人学习

Python uv官方安装与入门文档
Python uv官方安装与入门文档

共0课时 | 0人学习