想要熊猫数据框分组，但要保留数据结构

Question

我有一个如下所示的熊猫数据框。

code1  code2 sum
 a1     z1    2 
 a1     z2    4
 a1     z3    6
 b1     y1    5
 b1     y2    5
 c1     w1    3
 c1     w2    6
 c1     w3    9

在此数据框中，我想添加其他mean（）列，但保持完整 数据帧结构如下。

code1  code2 sum  mean
 a1     z1    2   (2+4+6)/3
 a1     z2    4   (2+4+6)/3
 a1     z3    6   (2+4+6)/3
 b1     y1    5   (5+5)/2
 b1     y2    5   (5+5)/2
 c1     w1    3   (3+6+9)/3
 c1     w2    6   (3+6+9)/3
 c1     w3    9   (3+6+9)/3

据我所知，在groupby和mean之后，code2列将如下合并。

但是想要code1和code2列以及数据结构（a1，a1，a1，b1，b1，c1，c1，c1）带仅在每个code1列上添加均值列。

我该如何实现？

谢谢