python - 保留列中项目出现次数最多的 Pandas 数据框行

Question

我有一个 Pandas 数据框

import pandas as pd

df =pd.DataFrame({'name':['john','joe','bill','richard','sam'],
                  'cluster':['1','2','3','1','2']})

df['cluster'].value_counts() 将给出基于 cluster 列的项的出现次数。

是否可以只保留列 cluster 中出现次数最多的行？

预期的输出是

簇 1 和 2 的出现次数相同，因此需要保留簇 1 和 2 的所有行。

Answer 1

使用这个

# find the most common clusters then filter those clusters
df[df.cluster.isin(df.cluster.mode())]