分组熊猫后仅返回最大值

时间:2020-09-08 18:44:27

标签: python pandas

我要解决的问题是找到运输速度最慢的产品。

我要做的第一步是使用以下代码查找产品的平均运输速度

westData.groupby('Product ID')['Shipping Speed'].agg(lambda x: x.mean())

我得到的结果采用以下格式

mean shipping speed per product

现在,我正在尝试过滤结果,以仅获得速度最慢的结果。换句话说,仅返回等于“运送速度”列中最大平均值的值。

在我看来,最慢的运送速度是

westData.groupby('Product ID')['Shipping Speed'].agg(lambda x: x.mean()).max()

Timedelta('7天00:00:00')

所以我只需要退回发货速度为('7天00:00:00')的产品

应类似于以下内容

required output

1 个答案:

答案 0 :(得分:1)

获得mean

s = westData.groupby('Product ID')['Shipping Speed'].mean()

使用max

进行过滤
s = s[s==s.max()]
items = s.index.tolist()