news 2026/4/16 14:21:53

one-hot编码

作者头像

张小明

前端开发工程师

1.2k 24
文章封面图
one-hot编码

我来详细介绍一下one-hot 编码(独热编码)。

什么是 One-Hot 编码?

One-Hot 编码是一种将分类变量转换为二进制向量的技术,其中每个类别都表示为一个二进制向量,只有一个元素为1("热"),其余都为0。

工作原理

假设有一个颜色类别变量:["红", "绿", "蓝"]

原始数据:

text

红 绿 蓝 红

One-Hot 编码后:

text

红 → [1, 0, 0] 绿 → [0, 1, 0] 蓝 → [0, 0, 1] 红 → [1, 0, 0]

为什么要使用 One-Hot 编码?

  1. 消除类别间的序数关系:避免机器学习模型误以为类别之间有大小顺序关系

  2. 兼容算法:许多机器学习算法(如线性回归、神经网络)需要数值输入

  3. 保持类别独立性:每个类别都成为独立的特征维度

在 Python 中的实现

方法1:使用 pandas

python

import pandas as pd # 创建示例数据 df = pd.DataFrame({'颜色': ['红', '绿', '蓝', '红', '绿']}) # 进行 one-hot 编码 one_hot = pd.get_dummies(df['颜色'], prefix='颜色') print(one_hot)

方法2:使用 scikit-learn

python

from sklearn.preprocessing import OneHotEncoder import numpy as np # 创建示例数据 data = np.array(['红', '绿', '蓝', '红', '绿']).reshape(-1, 1) # 创建编码器 encoder = OneHotEncoder(sparse_output=False) # 拟合和转换 one_hot_encoded = encoder.fit_transform(data) print(one_hot_encoded)

One-Hot 编码的优缺点

优点:

  • ✅ 简单直观,易于实现

  • ✅ 消除了类别间的虚假序数关系

  • ✅ 与大多数机器学习算法兼容

缺点:

  • 维度灾难:如果类别很多,会创建大量稀疏特征

  • 稀疏矩阵:存储效率低

  • ❌ 丢失了类别之间的关系信息(如果有的话)

替代方案

  1. 标签编码(Label Encoding):将类别转换为数字(0, 1, 2...)

  2. 目标编码(Target Encoding):用目标变量的统计量编码

  3. 二进制编码(Binary Encoding):将类别先转换为数字,再转换为二进制

实用建议

  • 当类别数量较少(如<15)时,使用 One-Hot 编码

  • 当类别数量很多时,考虑其他编码方式或特征选择

  • 在神经网络中,One-Hot 编码通常用于输入层

  • 对于有序类别,可以考虑使用有序编码而不是 One-Hot

示例:完整代码

python

import pandas as pd from sklearn.model_selection import train_test_split from sklearn.linear_model import LogisticRegression # 示例数据 data = pd.DataFrame({ '颜色': ['红', '绿', '蓝', '红', '绿', '蓝'], '尺寸': ['大', '中', '小', '中', '大', '小'], '价格': [100, 200, 150, 120, 180, 160] }) # One-Hot 编码分类变量 data_encoded = pd.get_dummies(data, columns=['颜色', '尺寸']) print("原始数据:") print(data) print("\nOne-Hot 编码后:") print(data_encoded)

One-Hot 编码是数据预处理中的基础且重要的技术,特别适用于处理名义型分类变量。

版权声明: 本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如若内容造成侵权/违法违规/事实不符,请联系邮箱:809451989@qq.com进行投诉反馈,一经查实,立即删除!
网站建设 2026/4/11 12:25:15

大学生必备!9个免费AI写论文神器,10分钟3万字+真实参考文献

如果你是面临延毕压力、被导师追着要稿的研究生&#xff1b;或是囊中羞涩、为知网查重费发愁的大学生&#xff1b;亦或是每日在海量文献中苦苦搜寻研究灵感的科研人员&#xff0c;那么这篇文章简直是为你量身打造的。在学术的道路上&#xff0c;撰写论文就像攀登高峰&#xff0…

作者头像 李华
网站建设 2026/4/13 17:41:44

C#之Modbus-RTU通讯-读取输出寄存器-整数

C#之Modbus-RTU通讯-读取输出寄存器-整数 Modbus Slave对应需要选择 数据类型 读取整数 case VarType.Int:switch (storeArea){case StoreArea.输出寄存器:res ReadKeepReg(stationAddress, startAddress, dataLength);break;case StoreArea.输入寄存器:break;default:break;}…

作者头像 李华
网站建设 2026/4/16 14:06:16

【计算机毕业设计案例】基于springboot的幼儿园管理系统的设计与实现活动信息、课程信息、菜谱信息(程序+文档+讲解+定制)

博主介绍&#xff1a;✌️码农一枚 &#xff0c;专注于大学生项目实战开发、讲解和毕业&#x1f6a2;文撰写修改等。全栈领域优质创作者&#xff0c;博客之星、掘金/华为云/阿里云/InfoQ等平台优质作者、专注于Java、小程序技术领域和毕业项目实战 ✌️技术范围&#xff1a;&am…

作者头像 李华
网站建设 2026/4/15 6:11:32

休闲无聊测试AI大模型生成

小兔子乖乖的起源与含义 《小兔子乖乖》是一首经典的中文儿歌&#xff0c;广泛流传于中国各地&#xff0c;常用于儿童启蒙教育。歌词通过小兔子与妈妈的对话&#xff0c;传递了安全教育的主题&#xff0c;教导孩子不要给陌生人开门。 歌词示例&#xff1a;小兔子乖乖&#xff0…

作者头像 李华
网站建设 2026/4/11 10:14:18

Java计算机毕设之基于springboot的物业报修系统的设计与实现住户信息管理、报修处理、费用收缴(完整前后端代码+说明文档+LW,调试定制等)

博主介绍&#xff1a;✌️码农一枚 &#xff0c;专注于大学生项目实战开发、讲解和毕业&#x1f6a2;文撰写修改等。全栈领域优质创作者&#xff0c;博客之星、掘金/华为云/阿里云/InfoQ等平台优质作者、专注于Java、小程序技术领域和毕业项目实战 ✌️技术范围&#xff1a;&am…

作者头像 李华