python - Pandas groupby 然后加入多列

Question

给定以下数据框:

A   B   C
1   2   3
1   9   8

df = df.groupby(['A'])['B'].apply(','.join).reset_index()

这产生了

A   B
1   2,9

不过，我也想用与 b 相同的方式将“C”列值与逗号连接在一起。

预期:

A   B   C
1   2,9 3,8

我试过:

df = df.groupby(['A'])['B','C'].apply(','.join).reset_index()

Answer 1

使用GroupBy.agg :

df = df.groupby(['A'])['B','C'].agg(','.join).reset_index()