pandas-返回包含每年最大值的月份

Question

我有一个数据框，如:

Year Month Value
2017  1     100
2017  2      1
2017  4      2
2018  3      88
2018  4      8
2019  5      87
2019  6      1

我让数据框返回值为最大值的每年的月份和值:

year  month  value
2017    1      100
2018    3      88
2019    5      87

我尝试过类似 df=df.groupby(["Year","Month"])['Value']).max()然而，它返回完整的数据集，因为每个年/月对都是独一无二的(我相信)。

Answer 1

您可以使用 .groupby(...).idxmax() 获取最高值出现的索引并使用它来索引原始数据帧:

In [28]: df.loc[df.groupby("Year")["Value"].idxmax()]
Out[28]:
   Year  Month  Value
0  2017      1    100
3  2018      3     88
5  2019      5     87