파이썬을 사용하여 데이터 프레임에 새로운 계산 된 열을 추가하는 방법

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의 행 1은 대상의 평균을 가져야합니다. 여기서 var1 == A, 즉 12 ((10.5 + 13.5)/2)

파이썬에서는 이것을 알아낼 수 없습니다. 도움이 될 것입니다.

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