Pandas groupby - 将唯一值计入每组单独的列

Question

我正在尝试进行分组，其中一列中有字符串分类数据：

ID   cat_1   cat_2
11   'OG'    'ASD'
11   'LOL'   'ASD' 
11   'OG'    'DFG' 
22   'LOL'   'DFG'
22   'OG'    'DFG'

我试图按 ID 分组，并将字符串数据聚合成一个数字特征，即每个类别的出现次数。 所以结果将是：

ID  OG  LOL  ASD  DFG
11   2    1    2    1
22   1    1    0    2

我怎样才能在 pandas 中实现这一点？ 谢谢你！

Answer 1

你可以堆叠/value_counts/unstack：

(df.set_index('ID')
   .stack()
   .groupby('ID')
   .value_counts()
   .unstack(fill_value=0)
)

注意。 如果你想要所有列，你可以添加.reset_index()

output：

    ASD  DFG  LOL  OG
ID                   
11    2    1    1   2
22    0    2    1   1

Answer 2

您可以将pd.get_dummies与groupby和stack一起使用：

>>> pd.get_dummies(df.set_index("ID").stack()).groupby("ID").sum()
    ASD  DFG  LOL  OG
ID                   
11    2    1    1   2
22    0    2    1   1