Pandas 数据框 groupby 按相反顺序求和字符串

Question

我有一个像这样的 DataFrame：

colA    colB 
1       aaa
1       rrr
1       www
2       bbb
2       ccc
2       sss
...

我想按如下方式转换 DataFrame

colA    Sum
1       wwwrrraaa
2       ssscccbbb
...

我尝试过

df.groupby(['colA'])['colB'].sum().reset_index()

，但字符串的总和被颠倒了。有没有一种优雅的方法来做到这一点？

Answer 1

不要使用 sum 来连接字符串。它看起来很花哨，但它是二次的，应该被认为是不好的做法。在Python中使用函数join

df = df[::-1].groupby('colA')['colB'].agg(''.join).reset_index()

Answer 2

反转数据帧；然后groupby + sum：

out = df[::-1].groupby('colA', as_index=False)['colB'].sum()

输出：

   colA       colB
0     1  wwwrrraaa
1     2  ssscccbbb