compress_memberships

er_evaluation.data_structures.compress_memberships(*memberships)

Compress membership vectors to int values, preserving index compatibility.

Parameters:

series (list) – list of membership vectors (Series) to compress

Returns:

List of Series with int codes for index and values. Index are compatible accross the Series.

Examples

>>> membership = pd.Series([None, "c1", "c1", "c2", "c2", "c3"], index=[0,1,2,3,4,5])
>>> compressed, = compress_memberships(membership)
>>> compressed
0    NaN
1    0.0
2    0.0
3    1.0
4    1.0
5    2.0
Name: 0, dtype: float64