分组后查找单独列的最小值/最大值

Question

我有以下要分组的数据框

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

首先进行分组，然后使用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)