温馨提示×

温馨提示×

您好,登录后才能下订单哦!

密码登录×
登录注册×
其他方式登录
点击 登录注册 即表示同意《亿速云用户服务条款》

如何加快pandas计算速度

发布时间:2021-11-30 10:43:06 来源:亿速云 阅读:454 作者:小新 栏目:互联网科技

小编给大家分享一下如何加快pandas计算速度,相信大部分人都还不怎么了解,因此分享这篇文章给大家参考一下,希望大家阅读完这篇文章后大有收获,下面让我们一起去了解一下吧!

如何加快pandas计算速度

什么问题困扰我们?

使用pandas,当您运行以下行时:

# Standard apply

df.apply(func)

得到这个CPU使用率:

如何加快pandas计算速度

标准pandas适用 - 仅使用1个CPU

即使计算机有多个CPU,也只有一个完全专用于您的计算。

而不是下边这种CPU使用,想要一个简单的方法来得到这样的东西:

如何加快pandas计算速度

并行Pandas适用 - 使用所有CPU

Pandaral·lel 如何帮助解决这个问题?

Pandaral·lel 的想法是将pandas计算分布在计算机上所有可用的CPU上,以显着提高速度。

安装:

$ pip install pandarallel [--user]

导入和初始化:

# Import

from pandarallel import pandarallel



# Initialization

pandarallel.initialize()

用法:

使用带有pandas DataFrame的简单用例df和要应用的函数func,只需替换经典apply的parallel_apply。

# Standard pandas apply

df.apply(func)



# Parallel apply

df.parallel_apply(func)

做完了!

请注意如果不想并行化计算,仍然可以使用经典apply方法。

也可以通过将显示每个工作CPU一个进度条progress_bar=True的initialize功能。

如何加快pandas计算速度

并行应用进度条

并配有更复杂的情况下使用带有pandas DataFrame df,该数据帧的两列column1,column2和功能应用func:

# Standard pandas apply

df.groupby(column1).column2.rolling(4).apply(func)

# Parallel apply

df.groupby(column1).column2.rolling(4).parallel_apply(func)

基准

对于此处提供的四个示例,请执行以下配置:

https://github.com/nalepae/pandarallel/blob/master/docs/examples.ipynb

  • 操作系统:Linux Ubuntu 16.04

  • 硬件:Intel Core i7 @ 3.40 GHz - 4核

如何加快pandas计算速度

4核上的标准与并行(越低越好)

除了df.groupby.col_name.rolling.apply速度仅增加x3.2因子之外,平均速度增加约x4因子,即使用过的计算机上的核心数。

它是如何在引擎盖下工作的?

调用parallel_apply时,Pandaral·lel

  • 实例化一个Pyarrow Plasma共享内存

https://arrow.apache.org/docs/python/plasma.html

  • 为每个CPU创建一个子进程,然后要求每个CPU在DataFrame的子部分上工作

  • 将所有结果合并到父进程中

以上是“如何加快pandas计算速度”这篇文章的所有内容,感谢各位的阅读!相信大家都有了一定的了解,希望分享的内容对大家有所帮助,如果还想学习更多知识,欢迎关注亿速云行业资讯频道!

向AI问一下细节

免责声明:本站发布的内容(图片、视频和文字)以原创、转载和分享为主,文章观点不代表本网站立场,如果涉及侵权请联系站长邮箱:is@yisu.com进行举报,并提供相关证据,一经查实,将立刻删除涉嫌侵权内容。

AI