데이터 분석/Python

[Pandas] transform

eunki 2021. 10. 7. 09:45
728x90

transform

groupby 후 transform 함수를 사용하면 원래의 index를 유지한 상태로 통계 함수를 적용한다.

전체 데이터의 집계가 아닌 각 그룹에서의 집계를 계산한다.

따라서 새로 생성된 데이터를 원본 dataframe과 합치기 쉽다.

df.groupby('Pclass').transform(np.mean)

 

 

 

df['Age2'] = df.groupby('Pclass').transform(np.mean)['Age']
df

 

 

 

df['Age3'] = df.groupby(['Pclass', 'Sex']).transform(np.mean)['Age']
df

728x90