Pandaral·lel
| Without parallelization |  |
|---|---|
| With parallelization |  |
Installation
$ pip install pandarallel [--upgrade] [--user]
Requirements
On Windows, Pandaral·lel will works only if the Python session (python, ipython, jupyter notebook, jupyter lab, ...) is executed from Windows Subsystem for Linux (WSL).
On Linux & macOS, nothing special has to be done.
Warning
- Parallelization has a cost (instantiating new processes, sending data via shared memory, ...), so parallelization is efficient only if the amount of calculation to parallelize is high enough. For very little amount of data, using parallelization is not always worth it.
Examples
An example of each API is available here.
Benchmark
For some examples, here is the comparative benchmark with and without using Pandaral·lel.
Computer used for this benchmark:
- OS: Linux Ubuntu 16.04
- Hardware: Intel Core i7 @ 3.40 GHz - 4 cores
For those given examples, parallel operations run approximately 4x faster than the standard operations (except for series.map which runs only 3.2x faster).
API
First, you have to import pandarallel:
from pandarallel import pandarallel
Then, you have to initialize it.
pandarallel.initialize()
This method takes 5 optional parameters:
shm_size_mb: Deprecated.nb_workers: Number of workers used for parallelization. (int) If not set, all available CPUs will be used.progress_bar: Display progress bars if set toTrue. (bool,Falseby default)verbose: The verbosity level (int,2by default)- 0 - don't display any logs
- 1 - display only warning logs
- 2 - display all logs
use_memory_fs: (bool,Noneby default)- If set to None and if memory file system is available, Pandarallel will use it to transfer data between the main process and workers. If memory file system is not available, Pandarallel will default on multiprocessing data transfer (pipe).
- If set to True, Pandarallel will use memory file system to transfer data between the main process and workers and will raise a
SystemErrorif memory file system is not available. - If set to False, Pandarallel will use multiprocessing data transfer (pipe) to transfer data between the main process and workers.
Using memory file system reduces data transfer time between the main process and workers, especially for big data.
Memory file system is considered as available only if the directory /dev/shm exists and if the user has read and write rights on it.
Basically, memory file system is only available on some Linux distributions (including Ubuntu).
With df a pandas DataFrame, series a pandas Series, func a function to apply/map, args, args1, args2 some arguments, and col_name a column name:
| Without parallelization | With parallelization |
|---|---|
df.apply(func) |
df.parallel_apply(func) |
df.applymap(func) |
df.parallel_applymap(func) |
df.groupby(args).apply(func) |
df.groupby(args).parallel_apply(func) |
df.groupby(args1).col_name.rolling(args2).apply(func) |
df.groupby(args1).col_name.rolling(args2).parallel_apply(func) |
df.groupby(args1).col_name.expanding(args2).apply(func) |
df.groupby(args1).col_name.expanding(args2).parallel_apply(func) |
series.map(func) |
series.parallel_map(func) |
series.apply(func) |
series.parallel_apply(func) |
series.rolling(args).apply(func) |
series.rolling(args).parallel_apply(func) |
You will find a complete example here for each row in this table.
Troubleshooting
I have 8 CPUs but parallel_apply speeds up computation only about x4. Why?
Actually Pandarallel can only speed up computation until about the number of cores your computer has. The majority of recent CPUs (like Intel Core i7) uses hyperthreading. For example, a 4-core hyperthreaded CPU will show 8 CPUs to the operating system, but will really have only 4 physical computation units.
On Ubuntu, you can get the number of cores with $ grep -m 1 'cpu cores' /proc/cpuinfo.
I use Jupyter Lab and instead of progress bars, I see these kind of things:
VBox(children=(HBox(children=(IntProgress(value=0, description='0.00%', max=625000), Label(value='0 / 625000')…
Run the following 3 lines, and you should be able to see the progress bars:
$ pip install ipywidgets
$ jupyter nbextension enable --py widgetsnbextension
$ jupyter labextension install @jupyter-widgets/jupyterlab-manager
(You may also have to install nodejs if asked)
0 Jan 20, 2022
179 Dec 21, 2022
791 Jan 04, 2023
50 Jan 05, 2023
94 Jan 04, 2023
8k Dec 29, 2022
3.7k Jan 03, 2023
2.2k Dec 25, 2022
0 Jul 07, 2021
167 Dec 13, 2022
4 Mar 06, 2022
1 Nov 17, 2021
6 Oct 25, 2022
2 Dec 04, 2021
5 Nov 13, 2022
38 Dec 03, 2022
1 Nov 18, 2021
1 Jan 24, 2022
1 Jan 12, 2022
32 Apr 25, 2022