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

如何在Python中使用多进程加速海量数据批处理?

秋静姑娘_8243

秋静姑娘_8243

发布时间:2026-10-06 08:21:29

|

790人浏览过

|

来源于php中文网

原创

multiprocessing.Pool.map卡住或报错的根本原因是忽略默认pickle序列化和资源隔离特性:参数须可pickle,子进程需独立初始化数据库连接等I/O资源,大数据量需用starmap+imap分批处理。

如何在python中使用多进程加速海量数据批处理?

多进程能显著加速CPU密集型批处理,但直接套用 Pool.map 很容易卡在数据传输、共享资源或子进程崩溃上——尤其处理上千万条记录时,错误常出现在参数传递、数据库连接复用和结果收集环节。

为什么 multiprocessing.Pool.map 会卡住或报错

常见现象是程序无响应、子进程静默退出,或抛出 BrokenPipeError / PicklingError。根本原因不是并发本身,而是默认序列化机制和资源隔离特性被忽略:

  • map 要求所有参数可被 pickle(比如不能传未绑定的类方法、lambda、数据库连接对象)
  • 每个子进程启动时重新导入模块,若主模块里有全局数据库连接(如 mysql = ConnectMySQL(Config)),子进程会尝试重复初始化,可能触发连接池耗尽或线程不安全错误
  • 当数据量极大(如单次传入百万级列表),pickle/unpickle 开销飙升,甚至触发内存溢出
  • Windows/macOS 下若没加 if __name__ == '__main__':,会反复 fork 主进程,导致无限递归创建子进程

用 starmap 处理多参数且避免闭包陷阱

当你需要为每条数据传多个参数(如 (id, image_base64, config)),别用 map + 匿名函数或闭包,而要用 starmap 配合元组解包。它明确要求参数是可迭代的元组序列,规避了作用域污染问题:

def process_item(item_id, img_b64, cfg):
    img_data = base64.b64decode(img_b64)
    md5 = get_md5_value(np.frombuffer(img_data, dtype=np.uint8))
    # 注意:这里不能复用主进程的 mysql 实例
    db = ConnectMySQL(cfg)  # 每个子进程独立建连
    db.update_md5(md5, item_id)
    return md5
<p>if <strong>name</strong> == '<strong>main</strong>':
mysql = ConnectMySQL(Config)
data_rows = mysql.select_data()  # 假设返回 [{'id': 1, 'upload_image': '...'}, ...]</p><pre class="brush:php;toolbar:false;"># 构造参数元组列表:每个元素是 (id, upload_image, Config)
args_list = [(r['id'], r['upload_image'], Config) for r in data_rows]

with Pool(4) as p:
    results = p.starmap(process_item, args_list)  # 安全、清晰、可预测

立即学习“Python免费学习笔记(深入)”;

python-script-generator
python-script-generator

快速生成专业的 Python 脚本和应用代码。一键创建完整项目结构,支持CLI、API、爬虫、Bot、Django等多种项目类型,包含完整的项目结构、配置文件、依赖管理、测试、README和文档。

下载

用 imap + chunksize 控制内存与吞吐平衡

处理千万级数据时,map 会一次性把全部参数序列发给子进程,内存峰值高;imap 是惰性迭代器,配合 chunksize 可分批调度,既降低内存压力,又避免小任务频繁 IPC 开销:

  • 设 chunksize=1000:每 1000 条数据打包成一个子任务,减少进程间通信次数
  • 用 for res in p.imap(...) 边计算边消费,适合需实时日志或提前终止的场景
  • 注意:若函数执行时间极不均衡(比如部分图片超大),小 chunksize 反而增加调度负担,建议先采样估算平均耗时再设值

示例中若要带进度条,直接用 tqdm(p.imap(...)) 即可,无需额外同步逻辑。

子进程中绝不能复用主进程的数据库连接或文件句柄

这是最隐蔽也最致命的坑。multiprocessing 的 fork(Linux/macOS)或 spawn(Windows)机制下,子进程继承的是主进程的文件描述符副本,但底层 TCP 连接、锁状态、缓冲区等无法安全共享。表现是:

  • 随机出现 MySQL Server has gone away
  • 部分进程写入成功,部分静默失败
  • 日志里看到重复的连接数暴涨

正确做法是:所有 I/O 资源(数据库连接、文件打开、HTTP session)必须在子进程内部按需创建、使用后立即关闭。不要试图用 Manager 或 Queue 传连接对象——它们不可 pickle,且即使能传也会引发竞态。

真正麻烦的从来不是“怎么开多进程”,而是“哪些东西不能跨进程传”——每次新增一个外部依赖(新库、新配置、新日志 handler),都得重审它的线程/进程安全性。这点比语法细节重要得多。

热门AI工具

更多
讯飞绘文

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

WorkBuddy

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

墨刀AI
墨刀AI Hot

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

豆包大模型

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

UpDream
UpDream Hot

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

蛙蛙写作

一款AI论文写作工具,主要用于超级AI智能写作助手,适合需要提升相关任务效率的用户。

立刻MV
立刻MV Hot

立刻MV是一款AI文本写作工具,AI 音乐视频(MV)创作工具。

Laper
Laper Hot

Laper是专为编剧、导演和制片人推出的 AI 原生剧本创作工具。

DeepSeek

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

相关专题

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

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

1651

2023.07.20

python能做什么
python能做什么

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

4124

2023.07.25

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

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

1669

2023.07.31

python教程
python教程

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

23897

2023.08.03

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

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

2927

2023.08.04

python eval
python eval

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

2967

2023.08.04

scratch和python区别
scratch和python区别

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

1143

2023.08.11

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

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

596

2023.08.10

LLVM自定义Pass怎么写
LLVM自定义Pass怎么写

本专题聚焦LLVM自定义Pass开发,整理Pass类结构、run()方法、PreservedAnalyses、CMake构建、插件注册、-load-pass-plugin加载和测试用例编写流程。

80

2026.09.30

热门下载

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

精品课程

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

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