Pythonを使用してデータフレームに新しい計算カラムを追加する方法

Question

これは初心者の質問かもしれませんが、これを行う方法を理解するのに数時間を費やしました。 Okie、私は150の変数とターゲットを持つデータフレームを持っています。以下のようにサンプル：

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