python - 如何根据组内实体找到最大金额？

Question

假设我的 DataFrame 看起来像这样:

Bank  Entity  Amount
JPM   NY      5000
JPM   NY      300
BOA   LA      10000
BOA   China   3000
MS    Japan   21000

我想根据顶级实体输出，同时记住 Bank 是不同的，所以 DataFrame 变成:

Bank  Entity  Amount
JPM   NY      5000
BOA   LA      10000
MS    Japan   21000

我将如何着手创建这样的东西？我知道如何sort_values 和group_by 但我确实做错了什么。

有什么想法吗？我相信这非常简单。

Answer 1

使用DataFrameGroupBy.idxmax获取最大值的索引，然后按 loc 选择:

df = df.loc[df.groupby('Bank')['Amount'].idxmax()]
print (df)
  Bank Entity  Amount
2  BOA     LA   10000
0  JPM     NY    5000
4   MS  Japan   21000

或sort_values先用 GroupBy.last :

df = df.sort_values('Amount').groupby('Bank', as_index=False).last()
print (df)
  Bank Entity  Amount
0  BOA     LA   10000
1  JPM     NY    5000
2   MS  Japan   21000