尝试对 pandas 中的两列进行分组并根据条件返回最大值

Question

我需要一些关于数据集的帮助，我正在尝试对其执行.groupby()以查找基于Entity和Year的.max() consumption值。

在此处输入图片描述导入

我用来执行最大值 function 的consumption列有时对于不同的Year具有相同的值。 发生这种情况时，我想返回该事件的max Year 。

df.groupby(['Entity','Year']).consumption.max().reset_index()

回报

在此处输入图像描述

最后，我想要一个DataFrame ，其中['Entity','Year','consumption']作为列，当特定Entity Year对的consumption相同时，返回两者中最高的Year 。

Answer 1

我制定了一个解决方案

df4 = df.groupby('Entity').consumption.max().reset_index()
df5 = df4.merge(df, left_on=['Entity', 'consumption'] , right_on=['Entity', 'consumption'])
df5 = df5.groupby(['Entity', 'consumption']).max().reset_index()
df5.head(10)

具有最大消费率的独特国家，如果消费率与去年同期相同，则返回最大年份。