Pandas 分组和聚合自定义函数

Question

示例 Pandas 数据框:
姓名
丹尼
丹琰
乔恩
乔恩
乔恩

期望的决赛 table :
姓名
丹琰
乔恩

基本上我想检查是否有人至少赢过一次。我如何使用 group_by 和聚合来做到这一点？

Answer 1

自 'Y' > 'N'计算结果为 True你可以拿max名内。

df.groupby('Name')['Won'].max()

#Name
#dan    Y
#jon    N
#Name: Won, dtype: object

对于具有多个类别的情况，其中简单的字符串排序不起作用，请使用有序的 CategoricalDtype 并仍然取最大值。

import pandas as pd

my_cat = pd.CategoricalDtype(categories=['N', 'Y'], ordered=True)
df['Won'] = df['Won'].astype(my_cat)

df.groupby('Name')['Won'].max()