如何从Pandas DataFrame中移除行数少于3的Token分组

嘿，我来帮你搞定这个问题！你已经成功找出了那些分组行数不足3的token，现在只需要把这些分组从主DataFrame里移除就行，这里有两种简单的方法：

方法1：直接使用groupby.filter()（最简洁） #

Pandas的groupby.filter()方法可以直接帮你筛选出满足条件的分组，一行代码就能搞定：

filtered_df = df.groupby('token').filter(lambda x: len(x) >= 3)

这个方法会自动保留所有分组行数≥3的行，直接排除掉像58、97这种行数不足3的分组的全部数据。

方法2：结合你已有的结果进行过滤 #

如果你想基于已经得到的sc_df来操作，可以先提取出需要排除的token列表，再用isin()取反过滤主DataFrame：

# 从sc_df中提取要排除的token
bad_tokens = sc_df.index.tolist()
# 过滤主DataFrame，保留不在bad_tokens里的行
filtered_df = df[~df['token'].isin(bad_tokens)]

这里的~符号表示取反，意思是"保留token不在bad_tokens列表中的所有行"。

验证结果 #

执行完上面任意一种方法后，输出filtered_df就能看到结果：

token  active
2     63       5
3     63       9
4     63       0

刚好剩下分组行数为3的63的所有数据，完全符合你的需求！

内容的提问来源于stack exchange，提问作者yaliga