python - 格式化 Pandas groupby() 以进行合并

Question

我希望有人能帮助我做一些我一直在尝试做的相对简单的事情。最近，我用了groupby总结我的df中的不同群体，如下所示:

                      MSA.name ifr.industry
0   Abilene, TX Metro Area           90
1   Abilene, TX Metro Area            P
2   Abilene, TX Metro Area           90
3   Abilene, TX Metro Area            F
4   Abilene, TX Metro Area           90
5   Abilene, TX Metro Area           90
6   Abilene, TX Metro Area           90
7   Abilene, TX Metro Area           90

然后，我将各个行业的员 worker 数相加为 df.groupby(['MSA.name', 'ifr.industry'])['employees'].sum()

MSA.name               ifr.industry        employees  
Abilene, TX Metro Area 10-12               720
                       13-15                70
                       16                   87
                       17-18                54
                       20-21                77
                       22                  403
                       23                  324
                       24                  383
                       25                  701
                       28                  310
                       29                  198
                       30                  233

但是，我想将其合并回原来的 df ，但我不知道如何轻松做到这一点。我想知道是否有办法 groupby()将列出MSA.name对于我执行 groupby() 后的每一次观察.

Answer 1

首先尝试一下df.groupby(['MSA.name', 'ifr.industry'])['employees'].sum().reset_index()

此外，您不需要在这里merge，使用transform

df['sum']=df.groupby(['MSA.name', 'ifr.industry'])['employees'].transform(sum)