深度学习基础教程:使用Pandas进行数据预处理
2025-06-04 20:00:44作者:魏侃纯Zoe
前言
在深度学习项目中,数据预处理是一个至关重要的环节。原始数据往往存在缺失值、格式不统一等问题,需要经过适当处理才能输入到神经网络中。本文将介绍如何使用Python的Pandas库进行数据预处理,为后续的深度学习模型训练做好准备。
Pandas简介
Pandas是Python生态系统中广受欢迎的数据分析工具,它提供了高效的数据结构和数据分析工具。在深度学习中,我们经常需要将Pandas处理后的数据转换为张量格式(如NumPy数组或PyTorch/TensorFlow张量)以供模型使用。
创建示例数据集
让我们从一个简单的CSV文件开始,模拟房屋价格数据集:
import os
import pandas as pd
# 创建数据目录和文件
os.makedirs(os.path.join('..', 'data'), exist_ok=True)
data_file = os.path.join('..', 'data', 'house_tiny.csv')
# 写入示例数据
with open(data_file, 'w') as f:
f.write('NumRooms,Alley,Price\n') # 列名
f.write('NA,Pave,127500\n') # 每条记录代表一个数据样本
f.write('2,NA,106000\n')
f.write('4,NA,178100\n')
f.write('NA,NA,140000\n')
这个数据集包含4条记录和3个特征:
- NumRooms:房间数量(数值型)
- Alley:巷子类型(类别型)
- Price:房屋价格(目标值)
读取数据
使用Pandas的read_csv函数读取CSV文件:
data = pd.read_csv(data_file)
print(data)
输出将显示包含NaN(缺失值)的数据表格。
处理缺失数据
数值型缺失值处理
对于数值型特征(如NumRooms),常见的处理方式是用均值填充缺失值:
inputs, outputs = data.iloc[:, 0:2], data.iloc[:, 2]
inputs['NumRooms'] = inputs['NumRooms'].fillna(inputs['NumRooms'].mean())
类别型缺失值处理
对于类别型特征(如Alley),我们将NaN视为一个独立的类别:
inputs = pd.get_dummies(inputs, dummy_na=True)
get_dummies函数会自动将类别型变量转换为独热编码(one-hot encoding),并为缺失值创建额外的列。
转换为张量格式
处理后的数据可以转换为各种深度学习框架支持的张量格式:
NumPy数组(MXNet等框架使用)
import numpy as np
X, y = np.array(inputs.values), np.array(outputs.values)
PyTorch张量
import torch
X, y = torch.tensor(inputs.values), torch.tensor(outputs.values)
TensorFlow常量
import tensorflow as tf
X, y = tf.constant(inputs.values), tf.constant(outputs.values)
数据预处理技巧总结
- 数值型数据:对于缺失值,可以采用均值、中位数或特定值填充
- 类别型数据:使用独热编码或嵌入表示,将缺失值视为独立类别
- 数据分割:通常将特征和目标值分开处理
- 张量转换:最后将数据转换为框架支持的张量格式
实践建议
- 对于大型数据集,考虑使用Pandas的块处理(chunk)功能
- 在填充缺失值时,记录填充的值,以便在预测时对新数据使用相同的填充策略
- 类别型变量转换时,保存转换规则,确保训练和预测时使用相同的编码方式
扩展练习
- 创建一个包含更多行和列的数据集
- 尝试删除缺失值最多的列而不是填充
- 探索Pandas的其他数据清洗功能,如异常值处理
通过掌握这些数据预处理技巧,您将能够为深度学习模型准备更高质量的数据,从而提高模型性能。
登录后查看全文
热门项目推荐
GLM-5智谱 AI 正式发布 GLM-5,旨在应对复杂系统工程和长时域智能体任务。Jinja00
GLM-5-w4a8GLM-5-w4a8基于混合专家架构,专为复杂系统工程与长周期智能体任务设计。支持单/多节点部署,适配Atlas 800T A3,采用w4a8量化技术,结合vLLM推理优化,高效平衡性能与精度,助力智能应用开发Jinja00
jiuwenclawJiuwenClaw 是一款基于openJiuwen开发的智能AI Agent,它能够将大语言模型的强大能力,通过你日常使用的各类通讯应用,直接延伸至你的指尖。Python0203- QQwen3.5-397B-A17BQwen3.5 实现了重大飞跃,整合了多模态学习、架构效率、强化学习规模以及全球可访问性等方面的突破性进展,旨在为开发者和企业赋予前所未有的能力与效率。Jinja00
AtomGit城市坐标计划AtomGit 城市坐标计划开启!让开源有坐标,让城市有星火。致力于与城市合伙人共同构建并长期运营一个健康、活跃的本地开发者生态。01
awesome-zig一个关于 Zig 优秀库及资源的协作列表。Makefile00
项目优选
收起
deepin linux kernel
C
27
12
OpenHarmony documentation | OpenHarmony开发者文档
Dockerfile
606
4.05 K
🔥LeetCode solutions in any programming language | 多种编程语言实现 LeetCode、《剑指 Offer(第 2 版)》、《程序员面试金典(第 6 版)》题解
Java
69
21
暂无简介
Dart
848
205
🎉 (RuoYi)官方仓库 基于SpringBoot,Spring Security,JWT,Vue3 & Vite、Element Plus 的前后端分离权限管理系统
Vue
1.47 K
829
Nop Platform 2.0是基于可逆计算理论实现的采用面向语言编程范式的新一代低代码开发平台,包含基于全新原理从零开始研发的GraphQL引擎、ORM引擎、工作流引擎、报表引擎、规则引擎、批处理引引擎等完整设计。nop-entropy是它的后端部分,采用java语言实现,可选择集成Spring框架或者Quarkus框架。中小企业可以免费商用
Java
12
1
喝着茶写代码!最易用的自托管一站式代码托管平台,包含Git托管,代码审查,团队协作,软件包和CI/CD。
Go
24
0
本项目是CANN提供的数学类基础计算算子库,实现网络在NPU上加速计算。
C++
923
772
🎉 基于Spring Boot、Spring Cloud & Alibaba、Vue3 & Vite、Element Plus的分布式前后端分离微服务架构权限管理系统
Vue
235
152
昇腾LLM分布式训练框架
Python
131
157