티스토리 뷰

pandas

group by 후 custom된 집계함수 사용

데이터조이 2022. 6. 16. 13:48 
import numpy as np
import pandas as pd
from sklearn.metrics import r2_score, mean_squared_error

def r2_rmse( g ):
    r2 = r2_score( g['Actual'], g['Predicted'] )
    rmse = np.sqrt( mean_squared_error( g['Actual'], g['Predicted'] ) )
    return pd.Series( dict(  r2 = r2, rmse = rmse ) )

your_df.groupby( 'Type' ).apply( r2_rmse ).reset_index()

data frame에서 group by 후 custom된 집계함수 사용

저작자표시

'pandas' 카테고리의 다른 글

영업일(bdate_range, is_busday, weekmask)  (0) 2022.07.15
pd.to_datetime, strptime, strftime,  (0) 2022.07.15
차집합, 교집합, drop_duplicates, duplicated, indicator  (0) 2022.03.07
drop id max, datetime, 요일, one-hot encoding, lag, mms  (0) 2022.03.06
group/sort by, index, series, array, rename, concat, merge, type  (0) 2022.03.06
댓글
Total
Today
Yesterday
공지사항
최근에 올라온 글
글 보관함