df.groupby('cluster', group_keys=False).apply(lambda df: df.sample(1))