Groupby基于另一列的前n条记录

Question

下面是我正在使用的数据集的一个示例。我试图做一个小组的位置只有前三名的位置基于公斤

Location KG Dollars
BKK     7   2
BKK     5   3
BKK     4   2
BKK     3   3
BKK     2   2
HKG     8   3
HKG     6   2
HKG     4   3
HKG     3   2
HKG     2   3

输出如下所示。按位置分组，将每个位置的前3公斤记录的公斤和美元相加

Location KG Dollars
BKK     16  7
HKG     18  8

我尝试过不同类型的groupby，只是在指定groupby的top n KG记录时遇到了一个问题

Answer 1

你能做到的

In [610]: df.groupby('Location').apply(lambda x: 
               x.nlargest(3, 'KG')[['KG', 'Dollars']].sum())
Out[610]:
          KG  Dollars
Location
BKK       16        7
HKG       18        8