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

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

2025-06-04 06:57:16作者:魏侃纯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的其他数据清洗功能,如异常值处理

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

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