qolmat.imputations.imputers.ImputerNOCB

class qolmat.imputations.imputers.ImputerNOCB(groups: Tuple[str, ...] = ())

NOCB imputer.

Impute by the next available value of the column. Relevent for time series. If the last observation is missing, it is imputed by a LOCF.

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.ImputerNOCB()
>>> 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   5.0
2   1.0   2.0   2.0   5.0
3   2.0   2.0   2.0   2.0

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