gpt4 book ai didi

python - 根据另一列对前 n 条记录进行分组

转载 作者:行者123 更新时间:2023-11-30 22:22:12 25 4
gpt4 key购买 nike

下面是我正在使用的数据集的示例。我正在尝试根据 KG 仅针对前 3 个位置进行按位置分组。

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 指定前 n 个 KG 记录时遇到问题。

最佳答案

你可以做

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

关于python - 根据另一列对前 n 条记录进行分组,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/48405569/

25 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com