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

深度学习基础教程:使用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)

数据预处理技巧总结

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

实践建议

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

扩展练习

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

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

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

项目优选

收起
kernelkernel
deepin linux kernel
C
27
11
docsdocs
OpenHarmony documentation | OpenHarmony开发者文档
Dockerfile
466
3.47 K
nop-entropynop-entropy
Nop Platform 2.0是基于可逆计算理论实现的采用面向语言编程范式的新一代低代码开发平台,包含基于全新原理从零开始研发的GraphQL引擎、ORM引擎、工作流引擎、报表引擎、规则引擎、批处理引引擎等完整设计。nop-entropy是它的后端部分,采用java语言实现,可选择集成Spring框架或者Quarkus框架。中小企业可以免费商用
Java
10
1
leetcodeleetcode
🔥LeetCode solutions in any programming language | 多种编程语言实现 LeetCode、《剑指 Offer(第 2 版)》、《程序员面试金典(第 6 版)》题解
Java
65
19
flutter_flutterflutter_flutter
暂无简介
Dart
715
172
giteagitea
喝着茶写代码!最易用的自托管一站式代码托管平台,包含Git托管,代码审查,团队协作,软件包和CI/CD。
Go
23
0
kernelkernel
openEuler内核是openEuler操作系统的核心,既是系统性能与稳定性的基石,也是连接处理器、设备与服务的桥梁。
C
203
82
RuoYi-Vue3RuoYi-Vue3
🎉 (RuoYi)官方仓库 基于SpringBoot,Spring Security,JWT,Vue3 & Vite、Element Plus 的前后端分离权限管理系统
Vue
1.27 K
695
rainbondrainbond
无需学习 Kubernetes 的容器平台,在 Kubernetes 上构建、部署、组装和管理应用,无需 K8s 专业知识,全流程图形化管理
Go
15
1
apintoapinto
基于golang开发的网关。具有各种插件,可以自行扩展,即插即用。此外,它可以快速帮助企业管理API服务,提高API服务的稳定性和安全性。
Go
22
1