如何使用python在数据框中添加新的计算列

Question

但是，这可能是一个新手问题，我花了几个小时弄清楚该如何做。 好的，我有一个包含150个变量和一个target的数据框。 示例如下：

train
Out[1]: 
        var1  var2    target
0          A      C      10.5   
1          A      D      13.5          
2          B      D      2.5     

我想用添加的具有目标均值的列创建新的数据框。 例如var1_new的row1应该具有目标的均值，其中var1 == A，即12（（10.5 + 13.5）/ 2）

我无法在python中弄清楚。 任何帮助，将不胜感激。

train_new
Out[2]: 
          var1   var2  target    var1_new   var2_new   
0          A      C      10.5       12       10.5             
1          A      D      13.5       12        8       
2          B      D      2.5        2.5       8      

Answer 1

variables = ['var1', 'var2']
for var in variables:
    df[var + "_new"] = None

for var in variables:
    for k, v in df.groupby(var).mean()['target'].iteritems():
        df.loc[df[var] == k, var + "_new"] = v