An implementation of Wilkinson formulas.
Project description
Formulaic is a high-performance implementation of Wilkinson formulas for Python.
Note: This project, while largely complete, is still a work in progress, and the API is subject to change between major versions (0.<major>.<minor>).
- Documentation: https://matthewwardrop.github.io/formulaic
- Source Code: https://github.com/matthewwardrop/formulaic
- Issue tracker: https://github.com/matthewwardrop/formulaic/issues
It provides:
- high-performance dataframe to model-matrix conversions.
- support for reusing the encoding choices made during conversion of one data-set on other datasets.
- extensible formula parsing.
- extensible data input/output plugins, with implementations for:
- input:
pandas.DataFrame
pyarrow.Table
- output:
pandas.DataFrame
numpy.ndarray
scipy.sparse.CSCMatrix
- input:
- support for symbolic differentiation of formulas (and hence model matrices).
Example code
import pandas
from formulaic import Formula
df = pandas.DataFrame({
'y': [0,1,2],
'x': ['A', 'B', 'C'],
'z': [0.3, 0.1, 0.2],
})
y, X = Formula('y ~ x + z').get_model_matrix(df)
y =
y | |
---|---|
0 | 0 |
1 | 1 |
2 | 2 |
X =
Intercept | x[T.B] | x[T.C] | z | |
---|---|---|---|---|
0 | 1.0 | 0 | 0 | 0.3 |
1 | 1.0 | 1 | 0 | 0.1 |
2 | 1.0 | 0 | 1 | 0.2 |
Benchmarks
Formulaic typically outperforms R for both dense and sparse model matrices, and vastly outperforms patsy
(the existing implementation for Python) for dense matrices (patsy
does not support sparse model matrix output).
For more details, see here.
Related projects and prior art
- Patsy: a prior implementation of Wilkinson formulas for Python, which is widely used (e.g. in statsmodels). It has fantastic documentation (which helped bootstrap this project), and a rich array of features.
- StatsModels.jl
@formula
: The implementation of Wilkinson formulas for Julia. - R Formulas: The implementation of Wilkinson formulas for R, which is thoroughly introduced here. [R itself is an implementation of S, in which formulas were first made popular].
- The work that started it all: Wilkinson, G. N., and C. E. Rogers. Symbolic description of factorial models for analysis of variance. J. Royal Statistics Society 22, pp. 392–399, 1973.
Project details
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
formulaic-0.6.4.tar.gz
(421.0 kB
view hashes)
Built Distribution
formulaic-0.6.4-py3-none-any.whl
(88.9 kB
view hashes)
Close
Hashes for formulaic-0.6.4-py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 570b3c7ab15fe8095e0dc86a4f71217102b23ce7e715053751a6cdcc580d7767 |
|
MD5 | 606f1af7866e00641b945f6ce5608cdf |
|
BLAKE2b-256 | db972ac97273f77138c4248ab63cdc4799ea3c87ea2be2d28bb726562e0d0827 |