swifter_parallel_apply_pandas_dataframe_quickstart.py

1import pandas as pd
2import swifter
3
4# Define a simple function to apply to the dataframe
5def power_function(x, n):
6    return x**n
7
8# Create a sample dataframe
9df = pd.DataFrame({'x': range(1000)})
10
11# Use swifter to apply the function to a column in parallel
12# Swifter automatically determines the fastest way to apply the function 
13# (e.g., vectorization, Dask, or simple pandas apply)
14df['x_squared'] = df['x'].swifter.apply(power_function, n=2)
15
16print(df.head())