Python如何实现数据的分块计算_Pandas中的chunksize设置

风枫酱_9496

风枫酱_9496

2026-04-27

550人浏览

原创

chunksize参数只属于pandas.read_csv等少数io函数,需作为顶层参数传入,返回textfilereader对象;设为1000表示每次读1000行,须用for循环或get_chunk()遍历,且应显式指定dtype和na_values以防类型不一致。

python如何实现数据的分块计算_pandas中的chunksize设置

chunksize参数到底传给谁?不是read_csv的兄弟函数

它只属于 pandas.read_csv(以及少数几个 IO 函数如 read_json),不是 pd.DataFrame 方法,也不是 groupby 或 apply 的参数。传错地方会直接报错:TypeError: read_csv() got an unexpected keyword argument 'chunksize'——说明你可能写在了 pd.read_csv(...).chunksize 后面,或者误加在 pd.concat() 里。

正确姿势是:把 chunksize 当作 read_csv 的顶层参数传进去,返回的是一个 TextFileReader 对象,不是 DataFrame。

  • chunksize=1000 表示每次读 1000 行,生成一个可迭代对象
  • 不设 chunksize 就是一次全读进内存,哪怕文件有 20GB
  • 设了但没用 for 循环或 get_chunk() 拿数据,等于白设

怎么安全地遍历chunk?别直接 for df in reader

看似最直觉的写法 for df in pd.read_csv("big.csv", chunksize=5000) 确实能跑,但容易掩盖两个隐患:类型转换失效、缺失值处理不一致。因为每块独立解析,如果某列前 5000 行全是整数,pandas 会推断为 int64;下一块出现 NaN,就强制转成 float64,后续 concat 时列类型可能错乱。

更稳妥的做法是显式统一 dtypes 和 na_values:

reader = pd.read_csv(
    "data.csv",
    chunksize=5000,
    dtype={"user_id": "string", "score": "float32"},
    na_values=["NULL", "", "N/A"]
)
  • 提前指定 dtype 可避免类型抖动,也省内存
  • 若原始文件无表头,必须加 header=None,否则第一块会当列名,第二块开始错位
  • 不要在循环里反复调 df.fillna().groupby().agg(),先做轻量过滤或映射,重计算留到合并后

chunksize设多少才不卡不死也不太慢?看内存和磁盘IO

没有万能值。设太小(比如 chunksize=100)会导致频繁磁盘寻址、Python 对象创建开销压倒计算本身;设太大(比如 chunksize=500000)又可能触发 OOM,尤其在 pandas 1.5+ 默认启用 copy_on_write=False 后,某些链式操作隐式复制变多。

361SCAN security check your skill python scripts
361SCAN security check your skill python scripts

扫描并分析已安装技能,用于查看指定技能目录的名称、描述和详情,或列出所有已安装的技能。

下载

经验参考(8GB 内存机器):

  • CSV 列少(chunksize=10000
  • 含长文本或嵌套 JSON 字符串:降到 5000 甚至 2000
  • SSD + 压缩 CSV(.csv.gz):可比 HDD 时提高 30%~50%,但解压耗 CPU,别盲目拉高

真正卡顿往往不是 chunksize 本身,而是你在每块里调了 df.merge() 或 df.apply(lambda x: requests.get(...)) ——这种 I/O 或网络阻塞跟分块逻辑无关,得单独拆出来。

合并结果时 concat容易丢索引或重复计数

默认 pd.concat([df1, df2]) 会保留各 chunk 的原始行号(0, 1, 2… 0, 1, 2…),导致最终 DataFrame 出现大量重复索引。虽然不影响计算,但调 df.loc[0] 会返回多行,df.drop_duplicates() 也可能误删。

解决很简单,但常被跳过:

  • 加 ignore_index=True:重排全局索引,最常用
  • 需要保留原始位置信息?改用 df.assign(chunk_id=i, row_in_chunk=df.index) 显式记录
  • 如果最后要写回 CSV,且原文件有索引列,别依赖 index_col=0 自动识别——分块读时它会被当普通列,得手动 df.set_index("id", drop=True)

真正麻烦的是聚合类任务(比如求全量均值)。不能对每个 chunk 算一次 mean 再平均——得保存每块的 sum 和 count,最后总和除以总数。这个逻辑不在 chunksize 能自动解决的范围里。

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

相关专题

更多
python打包成可执行文件
python打包成可执行文件

本专题为大家带来python打包成可执行文件相关的文章,大家可以免费的下载体验。

2023.07.20

1771

4

python能做什么
python能做什么

python能做的有:可用于开发基于控制台的应用程序、多媒体部分开发、用于开发基于Web的应用程序、使用python处理数据、系统编程等等。本专题为大家提供python相关的各种文章、以及下载和课程。

2023.07.25

4584

7

format在python中的用法
format在python中的用法

Python中的format是一种字符串格式化方法,用于将变量或值插入到字符串中的占位符位置。通过format方法,我们可以动态地构建字符串,使其包含不同值。php中文网给大家带来了相关的教程以及文章,欢迎大家前来阅读学习。

2023.07.31

1729

3

python教程
python教程

Python已成为一门网红语言,即使是在非编程开发者当中,也掀起了一股学习的热潮。本专题为大家带来python教程的相关文章,大家可以免费体验学习。

2023.08.03

27117

23

python环境变量的配置
python环境变量的配置

Python是一种流行的编程语言,被广泛用于软件开发、数据分析和科学计算等领域。在安装Python之后,我们需要配置环境变量,以便在任何位置都能够访问Python的可执行文件。php中文网给大家带来了相关的教程以及文章,欢迎大家前来学习阅读。

2023.08.04

3287

5

python eval
python eval

eval函数是Python中一个非常强大的函数,它可以将字符串作为Python代码进行执行,实现动态编程的效果。然而,由于其潜在的安全风险和性能问题,需要谨慎使用。php中文网给大家带来了相关的教程以及文章,欢迎大家前来学习阅读。

2023.08.04

3287

5

scratch和python区别
scratch和python区别

scratch和python的区别:1、scratch是一种专为初学者设计的图形化编程语言,python是一种文本编程语言;2、scratch使用的是基于积木的编程语法,python采用更加传统的文本编程语法等等。本专题为大家提供scratch和python相关的文章、下载、课程内容,供大家免费下载体验。

2023.08.11

1203

5

python合并两个列表
python合并两个列表

Python是一种强大的编程语言,具有许多方便的功能和工具。在Python中,有多种方法可以合并两个列表。php中文网给大家带来了相关的教程以及文章,欢迎大家前来学习阅读。

2023.08.10

596

4

python是前端还是后端
python是前端还是后端

Python属于前端也属于后端,其灵活性和丰富的生态系统使得开发人员能够在不同的领域中灵活运用。本专题为大家提供python相关的文章、下载、课程内容,供大家免费下载体验。

2023.08.11

2543

5

热门下载

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

精品课程

更多
相关推荐
/
热门推荐
/
最新课程
Uvicorn官方安装与入门手册
Uvicorn官方安装与入门手册

共0课时 | 0人学习

Python uv官方项目管理指南
Python uv官方项目管理指南

共0课时 | 0人学习

Python uv官方安装与入门文档
Python uv官方安装与入门文档

共0课时 | 0人学习