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

怎样在Python 3.11中实现多进程特征工程

老晨同学_4680

老晨同学_4680

发布时间:2026-10-04 06:13:36

|

466人浏览过

|

来源于php中文网

原创

multiprocessing.Pool在特征工程中易卡死或不生效,主因是:①缺失if __name__ == '__main__':保护致子进程递归启动;②特征函数含不可pickle对象(如lambda、未序列化模型);③全局状态(数据库连接、torch.device)无法跨进程传递;④DataFrame直接传入引发序列化开销与内存拷贝。

怎样在python 3.11中实现多进程特征工程

为什么 multiprocessing.Pool 在特征工程中容易卡死或不生效

Python 3.11 的 multiprocessing 默认启动方式已从 spawn 切换为 forkserver(Linux/macOS),但 Windows 仍用 spawn。若你在主模块中直接调用 Pool.map() 而没加 if __name__ == "__main__": 保护,子进程会重复导入并重新执行整个脚本——导致递归创建进程、内存暴涨甚至静默卡住。

特征工程函数若依赖全局状态(如未序列化的模型对象、数据库连接、torch.device)、或含不可 pickle 的对象(比如嵌套 lambda、某些 pandas accessor),Pool 会在序列化阶段失败,报错类似 AttributeError: Can't pickle local object 或直接 hang 住。

  • 确保所有传入 Pool.map() 的函数是模块顶层定义的,不是类方法或闭包
  • 避免在函数内初始化大型对象(如 sklearn.TfidfVectorizer)——应提前 fit 好并传入 fitted 实例,或改用 concurrent.futures.ProcessPoolExecutor 配合 initializer
  • Windows 用户必须加 if __name__ == "__main__":,否则子进程无法启动

怎样安全地并行处理 DataFrame 分块特征提取

直接对整张 pandas.DataFrame 调用 pool.map() 不现实:DataFrame 本身不可高效分片传输,且子进程间无法共享内存。正确做法是先切片成 list of DataFrame(每块约 1–5 万行),再并行处理,最后 pd.concat()。

注意:不要用 df.iloc[i:i+chunk_size] 在循环里反复切片——它每次返回新对象,开销大;改用 np.array_split(df, n_chunks) 更轻量。

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

# 正确示例:安全分块 + 并行
import pandas as pd
import numpy as np
from multiprocessing import Pool
<p>def extract_features_chunk(chunk_df):</p><div class="aritcle_card flexRow">
                                                        <div class="artcardd flexRow">
                                                                <a class="aritcle_card_img" href="/xiazai/skill6591" title="Li Python Sec Check"><img
                                                                                src="https://img.php.cn/upload/skill/000/000/081/179102166033725.jpg" alt="Li Python Sec Check"  onerror="this.onerror='';this.src='/static/lhimages/moren/morentu.png'" ></a>
                                                                <div class="aritcle_card_info flexColumn">
                                                                        <a href="/xiazai/skill6591" title="Li Python Sec Check">Li Python Sec Check</a>
                                                                        <p>Python 安全规范检查工具:基于 CloudBase 规范、腾讯安全指南,LLM 智能分析(默认禁用,优先本地执行)</p>
                                                                </div>
                                                                <a href="/xiazai/skill6591" title="Li Python Sec Check" class="aritcle_card_btn flexRow flexcenter"><b></b><span>下载</span> </a>
                                                        </div>
                                                </div><h1>所有操作必须只依赖 chunk_df 和内置/可 pickle 模块</h1><pre class="brush:php;toolbar:false;">chunk_df = chunk_df.copy()
chunk_df["text_len"] = chunk_df["text"].str.len()
chunk_df["word_count"] = chunk_df["text"].str.split().str.len()
return chunk_df

if name == "main": df = pd.read_parquet("data.parquet") chunks = np.array_split(df, 4) # 切成 4 块 with Pool(4) as p: result_chunks = p.map(extract_features_chunk, chunks) df_final = pd.concat(result_chunks, ignore_index=True)

ProcessPoolExecutor vs Pool:选哪个更稳

concurrent.futures.ProcessPoolExecutor 比原生 multiprocessing.Pool 更易控错、支持超时和资源管理,尤其适合特征工程这种可能偶发卡顿的任务。

关键差异点:

  • Pool.map() 是阻塞式,出错即中断;executor.map() 可捕获异常,executor.submit().result(timeout=30) 可设超时
  • ProcessPoolExecutor 自动处理子进程崩溃重试逻辑(需手动包装),Pool 需自己写 try/except + terminate()
  • Python 3.11 中,ProcessPoolExecutor 对 spawn 启动方式兼容性更好,初始化失败时错误提示更明确

如果特征函数偶尔因数据脏(如 NaN 文本)崩溃,优先用 ProcessPoolExecutor + as_completed 模式逐个取结果,避免整批失败。

哪些特征操作根本不该放进多进程

不是所有计算都适合并行:I/O 密集型(如读 CSV、查 Redis)、全局锁竞争型(如写同一文件、更新 shared memory array)、或单次耗时远低于进程启动开销(

  • 避免在子进程中调用 pd.read_csv() —— 改为父进程读好再分块传入
  • 不要用多进程做标准化(StandardScaler.fit_transform())—— fit 必须全量,transform 才可并行;应先 fit,再把 scaler 作为参数传给子进程
  • 慎用 multiprocessing.Manager 共享 dict/list:序列化开销大,且 Python 3.11 中 manager 进程默认不继承父进程环境变量,可能导致路径/配置丢失

真正值得并行的是 CPU 密集型纯计算,比如文本正则清洗、自定义 NLP 特征、数值列复杂表达式转换。其它环节,老老实实单线程更可靠。

热门AI工具

更多
UP简历
UP简历 Hot

一款AI办公效率工具,主要用于基于AI技术的免费在线简历制作工具,适合需要提升相关任务效率的用户。

超级简历WonderCV

一款AI办公效率工具,主要用于免费求职简历模版下载制作,应届生职场人必备简历制作神器,适合需要提升相关任务效率的用户。

DeepSeek

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

Laper
Laper Hot

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

Atoms
Atoms Hot

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

Loomy
Loomy Hot

一款AI工具,主要用于科大讯飞发布的桌面级 AI 助理,比 OpenClaw 更易用、更安全!,适合需要提升相关任务效率的用户。

UpDream
UpDream Hot

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

WorkBuddy

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

豆包大模型

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

相关专题

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

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

1651

2023.07.20

python能做什么
python能做什么

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

4084

2023.07.25

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

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

1649

2023.07.31

python教程
python教程

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

23577

2023.08.03

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

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

2887

2023.08.04

python eval
python eval

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

2927

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加载和测试用例编写流程。

60

2026.09.30

热门下载

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

精品课程

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

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