qolmat.imputations.imputers.ImputerSimple

class qolmat.imputations.imputers.ImputerSimple(groups: Tuple[str, ...] = (), strategy='median')

Simple imputer.

Impute each column by its mean, its median or its mode (if its categorical).

Parameters

groups: Tuple[str, …]

List of column names to group by, by default []

Examples

>>> import numpy as np
>>> import pandas as pd
>>> from qolmat.imputations import imputers
>>> imputer = imputers.ImputerSimple()
>>> df = pd.DataFrame(
...     data=[
...         [1, 1, 1, 1],
...         [np.nan, np.nan, np.nan, np.nan],
...         [1, 2, 2, 5],
...         [2, 2, 2, 2],
...     ],
...     columns=["var1", "var2", "var3", "var4"],
... )
>>> imputer.fit_transform(df)
   var1  var2  var3  var4
0   1.0   1.0   1.0   1.0
1   1.0   2.0   2.0   2.0
2   1.0   2.0   2.0   5.0
3   2.0   2.0   2.0   2.0

__init__(groups: Tuple[str, ...] = (), strategy='median') → None

Examples using qolmat.imputations.imputers.ImputerSimple

Benchmark for categorical data

Benchmark for categorical data

Comparison of basic imputers

Comparison of basic imputers

Benchmark for time series

Benchmark for time series