python - 计算大于 pandas groupby 中的值的项目

Question

我有 Yelp 数据集，我想计算所有超过 3 星的评论。我通过这样做得到评论的数量:

reviews.groupby('business_id')['stars'].count()

现在我想获得超过 3 星的评论数，所以我从 here 中获得灵感来尝试这个:

reviews.groupby('business_id')['stars'].agg({'greater':lambda val: (val > 3).count()})

但这只是像以前一样给我所有星星的计数。我不确定这是否是正确的方法？我在这里做错了什么。 lambda 表达式不遍历 stars 列的每个值吗？

编辑:好吧，我觉得很愚蠢。我应该使用 sum 函数而不是 count 来获取大于 3 的元素的值，如下所示:

reviews.groupby('business_id')['stars'].agg({'greater':lambda val: (val > 3).sum()})

Answer 1

你可以尝试做:

reviews[reviews['stars'] > 3].groupby('business_id')['stars'].count()