深度学习基础教程:使用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的其他数据清洗功能,如异常值处理
通过掌握这些数据预处理技巧,您将能够为深度学习模型准备更高质量的数据,从而提高模型性能。
登录后查看全文
热门项目推荐
atomcodeClaude Code 的开源替代方案。连接任意大模型,编辑代码,运行命令,自动验证 — 全自动执行。用 Rust 构建,极致性能。 | An open-source alternative to Claude Code. Connect any LLM, edit code, run commands, and verify changes — autonomously. Built in Rust for speed. Get StartedRust0214
cann-learning-hubCANN 学习中心仓,支持在线互动运行、边学边练,提供教程、示例与优化方案,一站式助力昇腾开发者快速上手。Jupyter Notebook0138
uni-appA cross-platform framework using Vue.jsJavaScript08
GLM-5.2智谱开源 GLM-5.2,这是针对长文本任务的最新旗舰模型。相较于前代产品 GLM-5.1,它在长文本任务处理能力上实现了显著飞跃,并且首次在稳定的 100 万 token 上下文中提供这一能力。Jinja00
SwanLab⚡️SwanLab - an open-source, modern-design AI training tracking and visualization tool. Supports Cloud / Self-hosted use. Integrated with PyTorch / Transformers / LLaMA Factory / veRL/ Swift / Ultralytics / MMEngine / Keras etc.Python00
tiny-universe《大模型白盒子构建指南》:一个全手搓的Tiny-UniverseJupyter Notebook03
项目优选
收起
deepin linux kernel
C
32
16
openEuler内核是openEuler操作系统的核心,既是系统性能与稳定性的基石,也是连接处理器、设备与服务的桥梁。
C
469
465
暂无描述
Dockerfile
778
5.08 K
本项目是CANN提供的transformer类大模型算子库,实现网络在NPU上加速计算。
C++
877
2.03 K
Ascend Extension for PyTorch
Python
758
968
本项目是CANN提供的神经网络类计算算子库,实现网络在NPU上加速计算。
C++
697
1.4 K
昇腾LLM分布式训练框架
Python
185
231
本项目是CANN提供的数学类基础计算算子库,实现网络在NPU上加速计算。
C++
1.1 K
1.14 K
本仓库是 Flutter SDK 与 Flutter Engine 的 OpenHarmony 适配版本,由 CPF-Flutter 团队维护。开发者可使用熟悉的 Flutter 技术栈开发 OpenHarmony 应用,3.35.7 及以后的适配版本可基于本仓库源码构建支持 OpenHarmony 的 Flutter Engine。
Dart
1.04 K
271
JiuwenSwarm 是一款基于openJiuwen开发的智能AI Agent,它能够将大语言模型的强大能力,通过你日常使用的各类通讯应用,直接延伸至你的指尖。
Python
2.25 K
677