在 groupby 之后查找单独列的最小/最大值

Question

我有以下要分组的 dataframe

TO    FROM    sort    toName    fromName    toLoc    fromLoc      max    min    
 A      B      0       test       test       test      test        10     8
 A      B      0       test       test       test      test        9      2

这个想法是按sort列分组，但保留最大/最小值。 最高的max和最低的min是我要填充的值。

TO    FROM    sort    toName    fromName    toLoc    fromLoc      max    min    
 A      B      0       test       test       test      test        10     2

我不知道如何同时进行比较和分组，有什么想法吗？

Answer 1

首先做groupby，然后使用aggregate ，可以得到两列的最小值和最大值：

import numpy as np

agg_dict = {"min": np.min, "max": np.max}
# OR YOU CAN DEFINE IT AS 
agg_dict = {"min": "min", "max": "max"}
df = df.groupby("sort").aggregate(agg_dict)