python两列数据相加_python:如何基于另一列分别对数据帧列的唯一元素求和

例如,我有一个包含两列的df。

输入

df = pd.DataFrame({'user_id':list('aaabbbccc'),'label':[0,0,1,0,0,2,0,1,2]})

print('df\n',df)

产量

df

label user_id

0 0 a

1 0 a

2 1 a

3 0 b

4 0 b

5 2 b

6 0 c

7 1 c

8 2 c

我想把元素算进去

label

按用户ID分组。

预期输出如下所示。

预期

df

label user_id label_0 label_1 label_2

0 0 a 2 1 0

1 0 a 2 1 0

2 1 a 2 1 0

3 0 b 2 0 1

4 0 b 2 0 1

5 2 b 2 0 1

6 0 c 1 1 1

7 1 c 1 1 1

8 2 c 1 1 1

简言之,列中

label_0

,我数一数

0

编队

标签

基于列

user_id

.

希望能得到帮助!