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

Question

以下是我正在使用的数据集的示例。我正在尝试通过位置仅针对基于KG的前三个位置进行分组。

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

输出如下所示。将该地点分组，将KG和美元汇总为每个地点的3个KG记录。

Location KG Dollars
BKK     16  7
HKG     18  8

我尝试过不同类型的群组，只是在为群组指定前n个KG记录时出现问题。

Answer 1

你可以做到

c = np.arange(12, dtype=np.float_).reshape(4,3)
c[2:4, 1] = np.nan

nansumwrapper(c, axis=1)
# array([  3.,  12.,  14.,  20.])