在pandas中将前2列最大值分组

Question

我正在尝试将我的数据集分组为具有最高整体评价的年份。但是当我尝试使用groupby nlargest时，它只返回年份并检查整体分数。

例如：

style | year | review_score
a     | 1990 | 4.0
b     | 1990 | 5.0
c     | 1990 | 4.2

我的输出应该是

style | year | review_score
b     | 1990 | 5.0
c     | 1990 | 4.2

我尝试了以下方法：

v=style.groupby(by='review_year')['beer_style']['review_overall'].nlargest(3)

请帮帮我

Answer 1

我认为需要先保留所有列的set_index：

v = style.set_index('style').groupby(by='year')['review_score'].nlargest(2).reset_index()
print (v)
   year style  review_score
0  1990     b           5.0
1  1990     c           4.2

另一种方法是首先使用sort_values和GroupBy.head调用两个列进行排序：

v = (style.sort_values(['year','review_score'], ascending=(True, False))
          .groupby(by='year').head(2))
print (v)
  style  year  review_score
1     b  1990           5.0
2     c  1990           4.2