본문 바로가기

언어 꿀Tip/Python 꿀tip!

07_01_45. groupby 컬럼명 level 합치기

groupby 했을 때,

qty에 대해서 합계, 평균, 분산 등 여러가지를 aggregate 하면, 컬럼명이 깔끔하지 않음!

 

qty_sum, qty_mean과 같이 나타내기 위해서 아래 코드 이용!

df_ag = df_1m_s.groupby(g_col, as_index=False).agg({"qty" : ["sum","mean","var","std","max","skew"]}).reset_index(drop=True)

# 1 방법
df_ag.columns = ['_'.join(tup).rstrip('_') for tup in df_ag.columns.values]

# 2 방법
df_ag.columns = list(map('_'.join, df_new_month.columns))