[Py3.5] Pandas.DataFrame의 Column별 연산

/*******************************************************************************************************************
-- Title : [Py3.5] Pandas.DataFrame의 Column별 연산
-- Reference : stackoverflow.com
-- Key word : pandas dataframe data frame 데이터프레임 데이터 프레임 max apply axis
*******************************************************************************************************************/

■ Script

import pandas as pd
 
# ******************************
# -- df 생성
# ******************************
data = {'name' : ['bill', 'joe', 'steve'],
        'test1' : [85, 75, 65],
        'test2' : [58, 45, 83],
        'test3' : [35, 65, 37],
        'test4' : [47, 61, 95]}
 
df = pd.DataFrame(data)
 
print (df)
print("[1] ", "-" * 100, "\n")
 
 
# ******************************
# -- df 컬럼간의 연산
# ******************************
 
# -- 숫자 컬럼 전체에서 max() 추출
df['HighScore'] = df.max(axis=1)
 
print (df)
print("[2] ", "*" * 100, "\n")
 
# -- 특정 숫자 컬럼서 max() 추출
df['HighScore'] = df[['test1','test2','test3']].max(axis=1)
print (df)
 
print("[3] ", "*" * 100, "\n")
 
# -- 특정 숫자 컬럼서 max() 추출
df['HighScore'] = df[['test2','test3']].apply(max, axis=1)
print (df)
 
print("[4] ", "*" * 100, "\n")