The module pandas-wizard
was developed as an initiative to provide additional functionalities on top of pandas
. I've spent years in developing projects involving the use of pandas
and have always used snippets or redundant GitHub Gists to keep track of additional functionalities that can be used alongside and later, decided to compile some of the code snippets directly into one file and publish the same as a package in PyPI/pandas-wizard.
The consolidated functionalities of the modules are as follows:
- Statistical Functions to use alongside
pd.groupby(...).agg({})
as below:- 🎉
pandaswizard.quantile
: A simple function to calculate the quantile of a grouped data series, and - 🎉
pandaswizard.percentile
: A simple function to calculate the percentile of a grouped data series.
- 🎉
- The wrappers utility that prints useful information like memory usage and time taken that can be useful in function development and optimization,
- The window module which acts as a wrapper to the
pd.DataFrame(...).rolling(...)
method to apply additional user-defined functions.- Additionally, the submodule
pandas-wizard/functions
captures basic functions likeweightedMA
that can be applied directly in a rolling window without explicitly defining a custom function.
- Additionally, the submodule