Pandas DataFrame多列groupby后Status列按规则聚合的实现方法

你可以直接在agg()里用以下两种方法实现需求，不用后续额外处理：

方法一：用any()判断是否存在OPEN #

通过判断分组内是否有至少一个'OPEN'值，直接返回对应结果：

df_agg = df.groupby(['col1', 'col2', 'col3'], as_index=False).agg(
    {'col4': 'sum', 
     'Status': lambda x: 'OPEN' if (x == 'OPEN').any() else 'CLOSED'}
)

原代码里的np.where会对分组内每个元素单独判断，所以返回列表；而any()是对整个分组做一次存在性检查，直接得到单个布尔值，再转成对应的状态字符串。

方法二：用max()简化逻辑 #

因为字符串比较中，'OPEN'的字母顺序比'CLOSED'靠后（'O' > 'C'），所以取分组内的最大值就能直接得到想要的结果：

df_agg = df.groupby(['col1', 'col2', 'col3'], as_index=False).agg(
    {'col4': 'sum', 
     'Status': 'max'}
)

这个方法更简洁，前提是Status列只有'OPEN'和'CLOSED'两种取值，完全符合你的场景。

内容的提问来源于stack exchange，提问作者Hotone