Python如何高效读取大型CSV文件?内存优化技巧

夏瑶姑娘_8342

夏瑶姑娘_8342

2025-07-25

993人浏览

原创

处理大型csv文件避免内存爆炸的方法有:1.分块读取,通过逐块处理降低内存占用;2.优化数据类型,如使用int8、float16或category类型减少内存消耗;3.选择合适工具,如dask实现并行计算或使用pandas的chunksize参数分块读取;4.处理缺失值时根据数据特性选择填充策略,如均值、中位数或前后值填充。使用dask可自动分块并行处理,而pandas则需手动循环处理每个chunk,同时结合dtype参数指定数据类型以进一步优化内存。

Python如何高效读取大型CSV文件?内存优化技巧

直接读取大型CSV文件,很容易遇到内存爆炸的问题。那么,如何才能在不耗尽内存的情况下,高效地处理这些庞然大物呢?关键在于分块读取、数据类型优化以及选择合适的工具。

Python如何高效读取大型CSV文件?内存优化技巧

分块读取,顾名思义,就是将大型CSV文件分割成小块,逐块读取并处理。数据类型优化则是尽可能使用更小的数据类型来存储数据,例如将字符串转换为数值类型,或者使用int8代替int64。选择合适的工具也很重要,例如pandas库虽然功能强大,但有时不如csv模块或Dask等工具更适合处理大型文件。

如何使用Dask加速CSV读取?

Python如何高效读取大型CSV文件?内存优化技巧

Dask是一个并行计算库,可以很好地处理大型数据集。它通过将数据分割成小块,并在多个核心上并行处理这些小块,从而实现加速。使用Dask读取CSV文件非常简单:

import dask.dataframe as dd

# 读取CSV文件
ddf = dd.read_csv('your_large_file.csv')

# 执行计算
result = ddf.groupby('column_name').mean().compute()

print(result)

dd.read_csv函数会创建一个Dask DataFrame,它类似于pandas DataFrame,但数据存储在磁盘上,而不是内存中。compute()函数会触发实际的计算,并在多个核心上并行执行。

Python如何高效读取大型CSV文件?内存优化技巧

如何使用Pandas分块读取CSV文件?

pandas也提供了分块读取CSV文件的功能,通过chunksize参数可以指定每次读取的行数:

macOS UI 自动化技能包 —— 用 Python 控制你的 Mac
macOS UI 自动化技能包 —— 用 Python 控制你的 Mac

让 macOS 自动化像说话一样简单 —— 用代码控制你的 Mac,无需学习 AppleScript 系统要求 macOS 14+ (Sonoma) Python 3.9+ 需要开启:辅助功能 + 屏幕录制权限

下载
import pandas as pd

# 分块读取CSV文件
chunksize = 10 ** 6  # 每次读取100万行
for chunk in pd.read_csv('your_large_file.csv', chunksize=chunksize):
    # 处理每个chunk
    print(chunk.head())
    # 进行数据清洗、转换等操作

这种方式可以避免一次性将整个文件加载到内存中,从而降低内存消耗。需要在循环中逐个处理每个chunk。

如何优化CSV文件的数据类型以减少内存占用?

优化数据类型是减少内存占用的关键步骤。pandas提供了dtype参数,可以在读取CSV文件时指定每一列的数据类型:

import pandas as pd

# 指定数据类型
dtype_dict = {
    'column_1': 'int32',
    'column_2': 'float16',
    'column_3': 'category'
}

# 读取CSV文件并指定数据类型
df = pd.read_csv('your_large_file.csv', dtype=dtype_dict)

print(df.info(memory_usage='deep'))

category类型特别适合存储重复的字符串数据,可以显著减少内存占用。使用df.info(memory_usage='deep')可以查看DataFrame的内存占用情况。根据实际数据情况,选择合适的数据类型,例如int8、int16、float16等。

如何处理CSV文件中的缺失值?

大型CSV文件中经常包含缺失值,处理缺失值也是一个重要的环节。常见的处理方法包括删除包含缺失值的行或列,或者使用合适的值填充缺失值。

import pandas as pd
import numpy as np

# 读取CSV文件
df = pd.read_csv('your_large_file.csv')

# 删除包含缺失值的行
df_dropna = df.dropna()

# 使用0填充缺失值
df_fillna_0 = df.fillna(0)

# 使用平均值填充缺失值
df_fillna_mean = df.fillna(df.mean())

# 使用中位数填充缺失值
df_fillna_median = df.fillna(df.median())

# 使用前一个值填充缺失值
df_fillna_ffill = df.fillna(method='ffill')

# 使用后一个值填充缺失值
df_fillna_bfill = df.fillna(method='bfill')

print(df.isnull().sum()) # 查看每列缺失值数量

选择哪种填充方法取决于数据的特性和业务需求。例如,时间序列数据可以使用前一个或后一个值填充,数值型数据可以使用平均值或中位数填充。需要注意的是,在分块读取CSV文件时,需要考虑缺失值在不同块之间的分布情况,选择合适的处理策略。

Python免费学习笔记(深入):立即使用
在学习笔记中,你将探索 Python 的核心概念和高级技巧!

相关文章

PHP速学视频免费教程(入门到精通)
PHP速学视频免费教程(入门到精通)

PHP怎么学习?PHP怎么入门?PHP在哪学?PHP怎么学才快?不用担心,这里为大家提供了PHP速学教程(入门到精通),有需要的小伙伴保存下载就能学习啦!

下载

本站声明:本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系admin@php.cn

相关专题

更多
Python 时间序列分析与预测
Python 时间序列分析与预测

本专题专注讲解 Python 在时间序列数据处理与预测建模中的实战技巧,涵盖时间索引处理、周期性与趋势分解、平稳性检测、ARIMA/SARIMA 模型构建、预测误差评估,以及基于实际业务场景的时间序列项目实操,帮助学习者掌握从数据预处理到模型预测的完整时序分析能力。

2025.12.04

340

15

Python 数据清洗与预处理实战
Python 数据清洗与预处理实战

本专题系统讲解 Python 在数据清洗与预处理中的核心技术,包括使用 Pandas 进行缺失值处理、异常值检测、数据格式化、特征工程与数据转换,结合 NumPy 高效处理大规模数据。通过实战案例,帮助学习者掌握 如何处理混乱、不完整数据,为后续数据分析与机器学习模型训练打下坚实基础。

2026.01.31

272

21

Python数据分析与Pandas高级实战
Python数据分析与Pandas高级实战

本专题围绕 Python 数据分析展开,系统讲解 Pandas 的高级用法,包括数据清洗、透视表、时间序列分析以及多表合并与分组操作。通过实战案例,帮助开发者掌握高效处理与分析数据的方法,提高数据处理效率与分析能力。

2026.04.13

751

25

数据类型有哪几种
数据类型有哪几种

数据类型有整型、浮点型、字符型、字符串型、布尔型、数组、结构体和枚举等。本专题为大家提供相关的文章、下载、课程内容,供大家免费下载体验。

2023.10.31

3271

5

php数据类型
php数据类型

本专题整合了php数据类型相关内容,阅读专题下面的文章了解更多详细内容。

2025.10.31

554

9

c语言 数据类型
c语言 数据类型

本专题整合了c语言数据类型相关内容,阅读专题下面的文章了解更多详细内容。

2026.02.12

542

19

js 字符串转数组
js 字符串转数组

js字符串转数组的方法:1、使用“split()”方法;2、使用“Array.from()”方法;3、使用for循环遍历;4、使用“Array.split()”方法。本专题为大家提供js字符串转数组的相关的文章、下载、课程内容,供大家免费下载体验。

2023.08.03

1838

5

js截取字符串的方法
js截取字符串的方法

js截取字符串的方法有substring()方法、substr()方法、slice()方法、split()方法和slice()方法。本专题为大家提供字符串相关的文章、下载、课程内容,供大家免费下载体验。

2023.09.04

2944

5

java基础知识汇总
java基础知识汇总

java基础知识有Java的历史和特点、Java的开发环境、Java的基本数据类型、变量和常量、运算符和表达式、控制语句、数组和字符串等等知识点。想要知道更多关于java基础知识的朋友,请阅读本专题下面的的有关文章,欢迎大家来php中文网学习。

2023.10.24

6244

49

热门下载

更多
网站特效
/
网站源码
/
网站素材
/
前端模板

精品课程

更多
相关推荐
/
热门推荐
/
最新课程