0

0

如何用Python构建特征工程—sklearn预处理全流程

絕刀狂花

絕刀狂花

发布时间:2025-07-12 14:27:02

|

650人浏览过

|

来源于php中文网

原创

在机器学习项目中,特征工程是提升模型性能的关键,而sklearn库提供了完整的预处理工具。1. 首先使用pandas加载数据并检查缺失值与数据类型,缺失严重则删除列,少量缺失则填充均值、中位数或标记为“missing”。2. 使用labelencoder或onehotencoder对类别变量进行编码,前者适用于有序类别,后者适用于无序类别,避免手动替换导致错误。3. 对数值特征应用standardscaler或minmaxscaler进行标准化或归一化,注意训练集用fit_transform,测试集仅transform。4. 利用columntransformer结合pipeline构建预处理流程,统一处理不同类型的特征,提高代码可复用性与效率。

如何用Python构建特征工程—sklearn预处理全流程

在做机器学习项目时,特征工程往往是决定模型性能的关键一环。而用Python做特征预处理,最常用的就是sklearn这个库了。它提供了很多实用的工具,能帮你把原始数据变得更适合模型训练。

如何用Python构建特征工程—sklearn预处理全流程

下面我来一步步讲讲怎么用sklearn完成一个完整的特征预处理流程,包括缺失值处理、标准化、编码类别变量等常见步骤。


1. 数据加载与初步检查

不管你用的是CSV文件还是数据库,第一步肯定是把数据读进来。一般会用pandas来做这一步:

立即学习Python免费学习笔记(深入)”;

如何用Python构建特征工程—sklearn预处理全流程
import pandas as pd
df = pd.read_csv('data.csv')

然后要看看有没有缺失值,哪些是数值型,哪些是类别型。可以用df.info()df.isnull().sum()快速查看。

建议:

如何用Python构建特征工程—sklearn预处理全流程
  • 如果某一列缺失太多(比如超过50%),可以考虑直接删掉。
  • 少量缺失的话,数值型可以用均值/中位数填充,类别型可以用众数或者加个“Missing”标记。

2. 编码类别变量:LabelEncoder 与 OneHotEncoder

机器学习模型不能直接处理字符串,所以类别变量需要转换成数字。

紫东太初
紫东太初

中科院和武汉AI研究院推出的新一代大模型

下载
  • LabelEncoder:适合有序类别,比如“小、中、大”,可以映射成0、1、2。
  • OneHotEncoder:适合无序类别,比如“红、绿、蓝”,会变成三个二进制列。

举个例子,假设你有个"Color"列,包含'red'、'blue'、'green':

from sklearn.preprocessing import OneHotEncoder

encoder = OneHotEncoder()
encoded = encoder.fit_transform(df[['Color']])

这样就会生成三列,每一列代表一种颜色是否存在。

注意点:

  • 不要手动写循环去替换类别值,容易出错。
  • 如果类别特别多,可以考虑用频率编码或目标编码来降维。

3. 处理数值特征:标准化与归一化

很多模型(比如线性回归、KNN)对数值范围敏感,所以要做标准化或归一化。

  • StandardScaler:减去均值再除以标准差,适合分布接近正态的数据。
  • MinMaxScaler:把数据缩放到[0,1]区间,适合有明显上下限的数据。
from sklearn.preprocessing import StandardScaler

scaler = StandardScaler()
scaled = scaler.fit_transform(df[['Age', 'Income']])

小细节:

  • 一定要先fit_transform训练集,测试集只transform
  • 对于偏态分布的数据,可以先做对数变换再标准化。

4. 管道化处理:使用ColumnTransformer组合不同操作

如果你既有类别型又有数值型变量,一个个处理太麻烦。可以用ColumnTransformer统一处理:

from sklearn.compose import ColumnTransformer
from sklearn.pipeline import Pipeline

numeric_features = ['Age', 'Income']
categorical_features = ['Color']

numeric_transformer = StandardScaler()
categorical_transformer = OneHotEncoder(handle_unknown='ignore')

preprocessor = ColumnTransformer(
    transformers=[
        ('num', numeric_transformer, numeric_features),
        ('cat', categorical_transformer, categorical_features)
    ])

pipeline = Pipeline(steps=[('preprocessor', preprocessor)])

这样整个预处理过程就可以封装起来,后续调用只需要一行:

X_processed = pipeline.fit_transform(X_train)

基本上就这些。整个流程看起来有点长,但其实每一步都很清晰。只要搞清楚每种方法适用的场景,组合起来就能应对大多数预处理任务了。

热门AI工具

更多
DeepSeek
DeepSeek

幻方量化公司旗下的开源大模型平台

豆包大模型
豆包大模型

字节跳动自主研发的一系列大型语言模型

通义千问
通义千问

阿里巴巴推出的全能AI助手

腾讯元宝
腾讯元宝

腾讯混元平台推出的AI助手

文心一言
文心一言

文心一言是百度开发的AI聊天机器人,通过对话可以生成各种形式的内容。

讯飞写作
讯飞写作

基于讯飞星火大模型的AI写作工具,可以快速生成新闻稿件、品宣文案、工作总结、心得体会等各种文文稿

即梦AI
即梦AI

一站式AI创作平台,免费AI图片和视频生成。

ChatGPT
ChatGPT

最最强大的AI聊天机器人程序,ChatGPT不单是聊天机器人,还能进行撰写邮件、视频脚本、文案、翻译、代码等任务。

相关专题

更多
Python 时间序列分析与预测
Python 时间序列分析与预测

本专题专注讲解 Python 在时间序列数据处理与预测建模中的实战技巧,涵盖时间索引处理、周期性与趋势分解、平稳性检测、ARIMA/SARIMA 模型构建、预测误差评估,以及基于实际业务场景的时间序列项目实操,帮助学习者掌握从数据预处理到模型预测的完整时序分析能力。

79

2025.12.04

Python 数据清洗与预处理实战
Python 数据清洗与预处理实战

本专题系统讲解 Python 在数据清洗与预处理中的核心技术,包括使用 Pandas 进行缺失值处理、异常值检测、数据格式化、特征工程与数据转换,结合 NumPy 高效处理大规模数据。通过实战案例,帮助学习者掌握 如何处理混乱、不完整数据,为后续数据分析与机器学习模型训练打下坚实基础。

32

2026.01.31

数据类型有哪几种
数据类型有哪几种

数据类型有整型、浮点型、字符型、字符串型、布尔型、数组、结构体和枚举等。本专题为大家提供相关的文章、下载、课程内容,供大家免费下载体验。

337

2023.10.31

php数据类型
php数据类型

本专题整合了php数据类型相关内容,阅读专题下面的文章了解更多详细内容。

224

2025.10.31

c语言 数据类型
c语言 数据类型

本专题整合了c语言数据类型相关内容,阅读专题下面的文章了解更多详细内容。

138

2026.02.12

js 字符串转数组
js 字符串转数组

js字符串转数组的方法:1、使用“split()”方法;2、使用“Array.from()”方法;3、使用for循环遍历;4、使用“Array.split()”方法。本专题为大家提供js字符串转数组的相关的文章、下载、课程内容,供大家免费下载体验。

760

2023.08.03

js截取字符串的方法
js截取字符串的方法

js截取字符串的方法有substring()方法、substr()方法、slice()方法、split()方法和slice()方法。本专题为大家提供字符串相关的文章、下载、课程内容,供大家免费下载体验。

221

2023.09.04

java基础知识汇总
java基础知识汇总

java基础知识有Java的历史和特点、Java的开发环境、Java的基本数据类型、变量和常量、运算符和表达式、控制语句、数组和字符串等等知识点。想要知道更多关于java基础知识的朋友,请阅读本专题下面的的有关文章,欢迎大家来php中文网学习。

1566

2023.10.24

C# ASP.NET Core微服务架构与API网关实践
C# ASP.NET Core微服务架构与API网关实践

本专题围绕 C# 在现代后端架构中的微服务实践展开,系统讲解基于 ASP.NET Core 构建可扩展服务体系的核心方法。内容涵盖服务拆分策略、RESTful API 设计、服务间通信、API 网关统一入口管理以及服务治理机制。通过真实项目案例,帮助开发者掌握构建高可用微服务系统的关键技术,提高系统的可扩展性与维护效率。

76

2026.03.11

热门下载

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

精品课程

更多
相关推荐
/
热门推荐
/
最新课程
ASP 教程
ASP 教程

共34课时 | 5.8万人学习

麻省理工大佬Python课程
麻省理工大佬Python课程

共34课时 | 5.5万人学习

关于我们 免责申明 举报中心 意见反馈 讲师合作 广告合作 最新更新
php中文网:公益在线php培训,帮助PHP学习者快速成长!
关注服务号 技术交流群
PHP中文网订阅号
每天精选资源文章推送

Copyright 2014-2026 https://www.php.cn/ All Rights Reserved | php.cn | 湘ICP备2023035733号