membership_to_clusters

er_evaluation.membership_to_clusters(membership)

Transform membership vector into clusters dictionary.

Parameters:

membership (Series) – Membership vector.

Returns:

Cluters dictionary.

Examples

>>> import pandas as pd
>>> membership = pd.Series(index=[1,2,3,4,5,6,7,8], data=[1,1,2,3,2,4,4,4])
>>> membership_to_clusters(membership)
{1: array([1, 2]), 2: array([3, 5]), 3: array([4]), 4: array([6, 7, 8])}