Pandas 多进程处理数据,速度快了不少!
前言
python 有自己的多进包 multiprocessing 去实现并行计算,但在 pandas 处理数据中,使用 multiprocessing 并不好使,只听见风扇转啊转,就不见运行完毕。
为了提高一点数据清洗的速度,找到一个 pandas 多进程的方法,pandarallel 库,做了一下测试。
小数据集(先试过了 1w)可能多进程还没单进程快,因为进程开启关闭也要一点时间;于是我弄了 100w 数据来测试:
利用以上数据做以下处理:
-
剔除 titile,comment 两列文本中的表情符号
-
title,comment 两列做一个分词处理,覆盖原来的列
单进程
在单进程的情况下,可以看到用时 294s,接近 5min 了。
多进程
multiprocessing 多进程写法,这种写法网上一搜一大把,代码没有错,多进程任务可以执行。
例如 run_task 函数中的任务是爬虫代码时,没有什么问题,但如果是数据清洗的代码,我测试就很久都跑不出来:
接下来换成 pandas 多进程 pandarallel 的写法就可以:
可以看到改写后时间用时 154s,2min30s,比单进程快了一倍。
关于 pandarallel 可以查看文档:
https://github.com/nalepae/pandarallel/tree/v1.5.2
对应的多进程写法函数对照表,pandas 中的 apply,applymap,map 三个函数的区别,写对应的代码:
未闻 Code·知识星球开放啦!
一对一答疑爬虫相关问题
职业生涯咨询
面试经验分享
每周直播分享
......
未闻 Code·知识星球期待与你相见~
一二线大厂在职员工
十多年码龄的编程老鸟
国内外高校在读学生
中小学刚刚入门的新人
在 “未闻 Code技术交流群” 等你来!
入群方式:添加微信“mekingname”,备注“粉丝群”(谢绝广告党,非诚勿扰!)
- 长见识,让大家看看什么是垃圾代码
- 一日一技:用一个奇技淫巧把字符串转成特定类型
- 最适合小白的Python学习神器!
- 【粉丝投稿】机器马大佬的微软面经
- 统计千行代码Bug率,有没有意义?
- 一日一技:二分偏左,二分搜索在分布式系统里面也有用?
- 一日一技:使用Python翻译HTML中的文本字符串
- 一日一技:如何让自己的工具函数在Python全局可用?
- 一日一技:Any与TypeVar,让IDE的自动补全更好用
- 一日一技:用Python做游戏有多简单
- 一日一技:如何批量给PDF添加水印?
- 一日一技:抛掉JavaScript,用HTML和Python做网站
- 一个让我感到 "细思极恐" 的开源项目!
- 一日一技:FastAPI 接口限流
- 5 分钟,使用内网穿透快速实现远程手机桌面!
- Python Delorean 优秀的时间格式智能转换工具
- 写在公众号粉丝2w时
- 一日一技:协程与多进程的完美结合
- 一个 "丧心病狂" 的开源项目
- python中如何优雅的实现代码与敏感信息分离?