首页
/ 深度学习基础教程:使用Pandas进行数据预处理

深度学习基础教程:使用Pandas进行数据预处理

2025-06-04 09:05:53作者:魏侃纯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)

数据预处理技巧总结

  1. 数值型数据:对于缺失值,可以采用均值、中位数或特定值填充
  2. 类别型数据:使用独热编码或嵌入表示,将缺失值视为独立类别
  3. 数据分割:通常将特征和目标值分开处理
  4. 张量转换:最后将数据转换为框架支持的张量格式

实践建议

  1. 对于大型数据集,考虑使用Pandas的块处理(chunk)功能
  2. 在填充缺失值时,记录填充的值,以便在预测时对新数据使用相同的填充策略
  3. 类别型变量转换时,保存转换规则,确保训练和预测时使用相同的编码方式

扩展练习

  1. 创建一个包含更多行和列的数据集
  2. 尝试删除缺失值最多的列而不是填充
  3. 探索Pandas的其他数据清洗功能,如异常值处理

通过掌握这些数据预处理技巧,您将能够为深度学习模型准备更高质量的数据,从而提高模型性能。

登录后查看全文
热门项目推荐

热门内容推荐

最新内容推荐

项目优选

收起
ohos_react_nativeohos_react_native
React Native鸿蒙化仓库
C++
176
260
RuoYi-Vue3RuoYi-Vue3
🎉 (RuoYi)官方仓库 基于SpringBoot,Spring Security,JWT,Vue3 & Vite、Element Plus 的前后端分离权限管理系统
Vue
858
507
openGauss-serveropenGauss-server
openGauss kernel ~ openGauss is an open source relational database management system
C++
129
182
openHiTLSopenHiTLS
旨在打造算法先进、性能卓越、高效敏捷、安全可靠的密码套件,通过轻量级、可剪裁的软件技术架构满足各行业不同场景的多样化要求,让密码技术应用更简单,同时探索后量子等先进算法创新实践,构建密码前沿技术底座!
C
255
299
ShopXO开源商城ShopXO开源商城
🔥🔥🔥ShopXO企业级免费开源商城系统,可视化DIY拖拽装修、包含PC、H5、多端小程序(微信+支付宝+百度+头条&抖音+QQ+快手)、APP、多仓库、多商户、多门店、IM客服、进销存,遵循MIT开源协议发布、基于ThinkPHP8框架研发
JavaScript
93
15
Cangjie-ExamplesCangjie-Examples
本仓将收集和展示高质量的仓颉示例代码,欢迎大家投稿,让全世界看到您的妙趣设计,也让更多人通过您的编码理解和喜爱仓颉语言。
Cangjie
331
1.08 K
HarmonyOS-ExamplesHarmonyOS-Examples
本仓将收集和展示仓颉鸿蒙应用示例代码,欢迎大家投稿,在仓颉鸿蒙社区展现你的妙趣设计!
Cangjie
397
370
note-gennote-gen
一款跨平台的 Markdown AI 笔记软件,致力于使用 AI 建立记录和写作的桥梁。
TSX
83
4
CangjieCommunityCangjieCommunity
为仓颉编程语言开发者打造活跃、开放、高质量的社区环境
Markdown
1.07 K
0
kernelkernel
deepin linux kernel
C
21
5