想让Pandas更快？试试Modin加速！

选自towardsdatascience

作者： Parul Pandey

机器之心编译

参与：Geek AI、Chita

「通过更改一行代码扩展你的 pandas 工作流。」

Pandas 是数据科学领域的专业人员所熟知的库。它提供了高性能且易于使用的结构和数据分析工具。然而，当处理超大数据集时，单核运行的 Pandas 性能会大打折扣，用户不得不转向不同的分布式系统来提升性能。然而，为了提升性能而进行的这种权衡会带来陡峭的学习曲线。

实际上，用户只希望 Pandas 运行得更快，而不是为特定的硬件设置调整他们的工作流程。这意味着人们希望在处理 10KB 和 10TB 的数据集时，可以使用相同的 Pandas 脚本。Modin 提供了一个优化 Pandas 的解决方案，使数据科学家能够专注于从数据中提取价值，而不是专注于处理数据的工具。

Modin

Modin 是加州大学伯克利分校 RISELab 的一个早期项目，旨在推动分布式计算在数据科学领域的应用。它是一个多进程的数据框架（Dataframe）库，具有与 Pandas 相同的应用程序接口（API），使用户能够加速他们的 Pandas 工作流程。

在一台 8 核的机器上，用户只需修改一行代码，Modin 就能将 Pandas 查询任务加速 4 倍。

该系统是为希望程序运行得更快、扩展性更强，而无需进行重大代码更改的 Pandas 用户设计的。这项工作的最终目标是能够在云环境中使用 Pandas。

安装

Modin 是完全开源的，可以通过下面的 GitHub 链接获得：

https://github.com/modin-project/modin

我们可以使用如下所示的 PyPi 指令来安装 Modin：

pip install modin

在 Windows 环境下，Ray 是安装 Modin 所需的依赖之一。Windows 本身并不支持 Ray，因此为了安装它，用户需要使用 WSL（适用于 Linux 的 Windows 子系统）。

Modin 如何加速数据处理过程

在笔记本上

在具有 4 个 CPU 核心的现代笔记本上处理适合该机器的数据框架时，Pandas 只使用了 1 个 CPU 核心，而 Modin 能够利用全部 4 个核心。

Pandas 和 Modin 对 CPU 核心的使用情况

从本质上讲，Modin 所做的只是增加了 CPU 所有核心的利用率，从而提供了更好的性能。

在大型机器上

在大型机器上，Modin 的作用就变得更加明显了。假设我们有一台服务器或一台非常强大的机器，Pandas 仍然只会利用一个核心，而 Modin 会使用所有的核心。下图显示了在一台拥有 144 核心的计算机上通过 Pandas 和 Modin 使用「read_csv」函数的性能对比情况：

Pandas 的运行时间会随着数据量的变化而线性增长，因为它仅仅使用 1 个核心。而从上图中可能很难看到绿色条形图的增长，因为 Modin 的运行时间实在太短了。

通常，Modin 使用「read_csv」函数读取 2G 数据需要 2 秒，而读取 18G 数据大约需要不到 18 秒。

架构

接下来，本文将解析 Modin 的架构。

数据框架分区

Modin 对数据框架的分区模式是沿着列和行同时进行划分的，因为这样为 Modin 在支持的列数和行数上都提供了灵活性和可扩展性。

系统架构

Modin 被分为不同的层：

Pandas API 在最顶层暴露给用户。下一层为查询编译器，它接收来自 Pandas API 层的查询并执行某些优化。最后一层为分区管理器（Partition Manager），负责数据布局并对发送到每个分区的任务进行重组、分区和序列化。modin 的一般架构

在 Modin 中实现 Pandas API

pandas 拥有大量的 API，这可能是它应用如此广泛的原因之一。

pandas API

由于 Pandas 具有这么多种操作，Modin 采用了一种数据驱动的策略。也就是说，Modin 的创造者找出了人们最常用的 Pandas 操作。他们研究了 Kaggle 平台上的 Pandas 使用数据，对上面所有的 notebook 和脚本进行了分析，最终总结出最受欢迎的 Pandas 方法如下：

「pd.read_CSV」是目前最常用的 Pandas 方法，其次是「pd.Dataframe」方法。因此，在 Modin 中，设计者们开始实现一些 Pandas 操作，并按照它们受欢迎程度从高到低的顺序进行优化：

目前，Modin 支持大约 71% 的 Pandas API。根据研究，这代表了 93% 的使用场景。Ray

Modin 利用 Ray 以无缝的方式加速 Pandas 的 notebook、脚本和库。Ray 是一个针对大规模机器学习和强化学习应用的高性能分布式执行框架。相同的代码可以在单台机器上运行以实现高效的多进程，也可以在集群上用于大型计算。你可以通过下面的 GitHub 链接获取 Ray：http://github.com/ray-project/ray。

使用方法

导入

Modin 封装了 Pandas，并透明地分发数据和计算任务，它通过修改一行代码就加速了 Pandas 的工作流程。用户可以继续使用以前的 Pandas notebook，同时体验 Modin 带来的显著加速，甚至在一台机器上。用户需要做的只是修改导入程序包的声明，引入「modin.pandas」而不是「pandas」。

import numpy as np
import modin.pandas as pd

我们将使用 Numpy 构建一个由随机整数组成的简单数据集。请注意，我们并不需要在这里指定分区。

data = np.random.randint(0,100,size = (2**16, 2**4))
df = pd.DataFrame(data)
df = df.add_prefix("Col:")

当我们将数据的类型打印在屏幕上时，会显示出「Modin 数据框架」。

type(df)
modin.pandas.dataframe.DataFrame

如果我们使用「head」命令打印出前五行数据，它会像 Pandas 一样显示出 HTML 表单。

df.head()

对比实验

Modin 会管理数据分区和重组，从而使用户能够将注意力集中于从数据中提取出价值。下面的代码是在一台 2013 年的拥有 4 个 CPU 核心和 32 GB RAM 内存的 iMac 机器上运行的。

pd.read_csv

「read_csv」是目前为止最常用的 Pandas 操作。接下来，本文将对分别在 Pandas 和 Modin 环境下使用「read_csv」函数的性能进行一个简单的对比。

pandas

%%time
import pandas
pandas_csv_data = pandas.read_csv("../800MB.csv")
-----------------------------------------------------------------
CPU times: user 26.3 s, sys: 3.14 s, total: 29.4s
Wall time: 29.5 s

Modin

%%time
modin_csv_data = pd.read_csv("../750MB.csv")
-----------------------------------------------------------------
CPU times: user 76.7 ms, sys: 5.08 ms, total: 81.8 ms
Wall time: 7.6 s

通过使用 Modin，只需要修改导入声明就可以在一台 4 核机器上以普通 Pandas 4 倍的速度执行「read_csv」操作。

df.groupby

Pandas 的「groupby」聚合函数底层编写得非常好，运行速度非常快。但是即使如此，Modin 的性能也比 Pandas 要好。

Pandas

%%time
import pandas
_ = pandas_csv_data.groupby(by=pandas_csv_data.col_1).sum()
-----------------------------------------------------------------
CPU times: user 5.98 s, sys: 1.77 s, total: 7.75 s
Wall time: 7.74 s

modin

%%time
results = modin_csv_data.groupby(by=modin_csv_data.col_1).sum()
-----------------------------------------------------------------
CPU times: user 3.18 s, sys: 42.2 ms, total: 3.23 s
Wall time: 7.3 s

Pandas 实现的默认设置

如果想要使用尚未实现或优化的 Pandas API，实际上可以使用默认的 Pandas API。这使得该系统可以用于使用 Modin 中尚未实现操作的 notebook 中（尽管由于即将使用 Pandas API，性能会有所下降）。当使用默认的 Pandas API 时，你将看到一个警告：

dot_df = df.dot(df.T)

当计算完成后，该操作会返回一个分布式的 Modin 数据框架。

type(dot_df)
-----------------
modin.pandas.dataframe.DataFrame

结语

Modin 项目仍处于早期阶段，但对 Pandas 来说是一个非常有发展前景的补充。Modin 为用户处理所有的数据分区和重组任务，这样我们就可以集中精力处理工作流程。Modin 的基本目标是让用户能够在小数据和大数据上使用相同的工具，而不用考虑改变 API 来适应不同的数据规模。

原文链接：https://towardsdatascience.com/get-faster-pandas-with-modin-even-on-your-laptops-b527a2eeda74