如何使用groupby获取与列的最大值对应的所有行

Question

对于给定的数据帧df，如下所示：

   Election Yr.  Party   States Votes
0     2000           A       a    50  
1     2000           A       b    30
2     2000           B       a    40
3     2000           B       b    50  
4     2000           C       a    30
5     2000           C       b    40
6     2005           A       a    50  
7     2005           A       b    30
8     2005           B       a    40
9     2005           B       b    50  
10    2005           C       a    30
11    2005           C       b    40

我想得到在相应年份获得最多选票的政党。我使用下面的代码对“选举年”和“政党”进行分组，然后使用.sum（）获得每个政党每年的总票数

df = df.groupby(['Election Yr.', 'Party']).sum()

现在，如何让该党获得每年最多的选票？我拿不到这个

非常感谢您的支持

Answer 1

尝试使用groupby和idxmax的组合：

gb = df.groupby(["Election Yr.", "Party"]).sum()
gb.loc[gb.groupby("Election Yr.")["Votes"].idxmax()].reset_index()
>>> gb
   Election Yr. Party  Votes
0          2000     B     90
1          2005     B     90