将分组结果附加到原始选项卡

Question

 col = full.groupby('state')[['demVote']].mean()

full是原始表的名称。如何将GROUPBY语句的结果作为名为'DEM'的列附加到原始表full上？你知道吗

Answer 1

使用transform：

full['DEM'] = full.groupby('state')['demVote'].transform('mean')

例如：

>>> full
   demVote state
0        1    ca
1        2    ca
2        3    ny
3        4    ny

full['DEM'] = full.groupby('state')['demVote'].transform('mean')

>>> full
   demVote state  DEM
0        1    ca  1.5
1        2    ca  1.5
2        3    ny  3.5
3        4    ny  3.5