熊猫掉落稀有条目

Question

我是熊猫新手。 为了简化，我有一个包含两列的数据框：product_id 和 rating。每个条目都是给定产品的新评论。 现在我想获得一个新的数据框，其中删除了与收到少于 20 条评论（即在原始数据框中出现少于 20 次）的产品相对应的行。 我可以计算出现次数：

a = data.groupby('product_id').count()
b = a.loc[a['rating']>20]

但这给了我一个一维数据框。显示时，每个 product_id 都有其计数，但我无法访问实际的 product_id 以使用它们来过滤原始表。例如，

b.values

返回计数的一维数组，但不返回 product_ids。

Answer 1

你想filter:

a = data.groupby('product_id').filter(lambda x: len(x) > 20)