讲师中心 微信公众号
AI工具推荐 视频效率加速

如何通过 生成器函数(Generator) 实现对 GB 级别海量数据集的“流式分片迭代”

风婷姑娘_1262

风婷姑娘_1262

发布时间:2026-05-14 18:19:01

|

512人浏览过

|

来源于php中文网

原创

用生成器函数实现GB级数据流式分片迭代,核心是不加载全量到内存,通过逐行/按块读取文件、fetchmany查数据库、mmap处理二进制等方式对齐yield节奏,按需产出固定大小批次。

如何通过 生成器函数(generator) 实现对 gb 级别海量数据集的“流式分片迭代”

用生成器函数实现 GB 级数据的流式分片迭代,核心是“不加载全量到内存,边读边切、边产边用”。关键不在生成器语法本身,而在如何把数据源(如大文件、数据库游标、网络流)与 yield 节奏对齐,按需吐出固定大小的批次。

一、从大文件按行/按块分片:避免 readlines() 全载入

GB 文本文件(如日志、CSV)最常见。直接 f.readlines() 会 OOM;应逐行读取 + 批量 yield:

  • 用 for line in f: 迭代文件对象,内存只存单行
  • 内部累积列表达指定数量(如 1000 行)后 yield batch,再清空继续
  • 最后一组不足批量时,别漏掉 if batch: yield batch

示例片段:

  def file_chunker(path, batch_size=1000):
    with open(path, encoding='utf-8') as f:
      batch = []
      for line in f:
        batch.append(line.rstrip('\n'))
        if len(batch) == batch_size:
          yield batch
          batch = []
      if batch:
        yield batch

二、处理二进制大文件:按字节块读取 + 解析边界

如大型 Protocol Buffer、自定义二进制格式,需控制每次读取字节数,但注意不能截断记录:

  • 用 f.read(chunk_bytes) 每次读固定字节(如 64KB),比逐行更底层
  • 若记录有长度头(如前4字节是 payload 长度),需缓冲未完成记录,跨块拼接
  • 推荐先用 mmap 映射文件,再配合生成器按逻辑记录切片,更高效

三、对接数据库游标:用 fetchmany 替代 fetchall

查千万级表时,cursor.fetchall() 仍会把全部结果集加载进 Python 内存。正确方式:

  • 执行查询后,用 cursor.fetchmany(size) 循环取批,每次只拿一批(如 5000 行)
  • 每批 yield 后,Python GC 可及时回收该批引用,内存保持恒定
  • 注意:需确保数据库驱动支持服务器端游标(如 PostgreSQL 的 name 参数,MySQL 的 SSCursor),否则 fetchmany 仍是客户端缓存

四、组合与增强:添加进度、过滤、转换逻辑

真实场景常需在分片中嵌入处理链。生成器天然适合管道化:

  • 用装饰器或嵌套生成器封装:一个生成器负责读取分片,另一个负责解析/清洗/类型转换
  • 加简单进度提示:在每次 yield 前打印当前偏移或批次序号(注意别影响性能)
  • 支持提前终止:调用方用 break 退出 for 循环,生成器自动停止,底层文件/连接可被 finally 或上下文管理器安全关闭

不复杂但容易忽略。

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

热门AI工具

更多
WorkBuddy

一款AI办公效率工具,主要用于腾讯云推出的AI原生桌面智能体工作台,适合需要提升相关任务效率的用户。

VibeKnow
VibeKnow Hot

一款AI视频创作工具,主要用于全球首个AI知识视频创作平台,文档、文章、网页,一键生成视频,适合需要提升相关任务效率的用户。

讯飞绘文

讯飞绘文是一款由科大讯飞推出的一站式 AIGC 内容运营平台。

DeepSeek

DeepSeek是一款面向对话、写作、编程和推理场景的AI大模型工具。

UpDream
UpDream Hot

一款AI视频创作工具,主要用于哔哩哔哩推出的自研AI视频创作工具,适合需要提升相关任务效率的用户。

墨刀AI
墨刀AI Hot

一款AI图像与设计工具,主要用于产品经理的专属智能体,适合需要提升相关任务效率的用户。

豆包大模型

豆包大模型是一款由字节跳动推出的企业级大语言模型服务平台。

讯飞智作

讯飞智作是一款AI视频创作工具,AI文本配音工具,数字人课程、营销视频制作。

Atoms
Atoms Hot

Atoms是一款AI智能体工具,第一支自动构建真实业务的 AI 团队。

相关专题

更多
js获取数组长度的方法
js获取数组长度的方法

在js中,可以利用array对象的length属性来获取数组长度,该属性可设置或返回数组中元素的数目,只需要使用“array.length”语句即可返回表示数组对象的元素个数的数值,也就是长度值。php中文网还提供JavaScript数组的相关下载、相关课程等内容,供大家免费下载使用。

5126

2023.06.20

js刷新当前页面
js刷新当前页面

js刷新当前页面的方法:1、reload方法,该方法强迫浏览器刷新当前页面,语法为“location.reload([bForceGet]) ”;2、replace方法,该方法通过指定URL替换当前缓存在历史里(客户端)的项目,因此当使用replace方法之后,不能通过“前进”和“后退”来访问已经被替换的URL,语法为“location.replace(URL) ”。php中文网为大家带来了js刷新当前页面的相关知识、以及相关文章等内容

1249

2023.07.04

js四舍五入
js四舍五入

js四舍五入的方法:1、tofixed方法,可把 Number 四舍五入为指定小数位数的数字;2、round() 方法,可把一个数字舍入为最接近的整数。php中文网为大家带来了js四舍五入的相关知识、以及相关文章等内容

5044

2023.07.04

js删除节点的方法
js删除节点的方法

js删除节点的方法有:1、removeChild()方法,用于从父节点中移除指定的子节点,它需要两个参数,第一个参数是要删除的子节点,第二个参数是父节点;2、parentNode.removeChild()方法,可以直接通过父节点调用来删除子节点;3、remove()方法,可以直接删除节点,而无需指定父节点;4、innerHTML属性,用于删除节点的内容。

980

2023.09.01

JavaScript转义字符
JavaScript转义字符

JavaScript中的转义字符是反斜杠和引号,可以在字符串中表示特殊字符或改变字符的含义。本专题为大家提供转义字符相关的文章、下载、课程内容,供大家免费下载体验。

1916

2023.09.04

js生成随机数的方法
js生成随机数的方法

js生成随机数的方法有:1、使用random函数生成0-1之间的随机数;2、使用random函数和特定范围来生成随机整数;3、使用random函数和round函数生成0-99之间的随机整数;4、使用random函数和其他函数生成更复杂的随机数;5、使用random函数和其他函数生成范围内的随机小数;6、使用random函数和其他函数生成范围内的随机整数或小数。

3605

2023.09.04

如何启用JavaScript
如何启用JavaScript

JavaScript启用方法有内联脚本、内部脚本、外部脚本和异步加载。详细介绍:1、内联脚本是将JavaScript代码直接嵌入到HTML标签中;2、内部脚本是将JavaScript代码放置在HTML文件的`<script>`标签中;3、外部脚本是将JavaScript代码放置在一个独立的文件;4、外部脚本是将JavaScript代码放置在一个独立的文件。

4673

2023.09.12

Js中Symbol类详解
Js中Symbol类详解

javascript中的Symbol数据类型是一种基本数据类型,用于表示独一无二的值。Symbol的特点:1、独一无二,每个Symbol值都是唯一的,不会与其他任何值相等;2、不可变性,Symbol值一旦创建,就不能修改或者重新赋值;3、隐藏性,Symbol值不会被隐式转换为其他类型;4、无法枚举,Symbol值作为对象的属性名时,默认是不可枚举的。

3060

2023.09.20

Gradle插件开发入门教程
Gradle插件开发入门教程

本专题介绍Gradle插件开发的基本流程,涵盖插件类编写、自定义任务注册、扩展参数设计和插件应用方法,帮助你理解插件如何组织构建逻辑,减少多个项目之间的重复配置。

0

2026.10.22

热门下载

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

精品课程

更多
热门推荐
/
最新课程
关于我们 免责申明 举报中心 意见反馈 讲师合作 广告合作 最新更新
php中文网:公益在线php培训,帮助PHP学习者快速成长!
关注服务号
PHP中文网订阅号
每天精选资源文章推送

Copyright 2014-2026 https://www.php.cn/ All Rights Reserved | php.cn