Pandas 데이터에 일괄 함수 적용하기

 

Pandas DataFrame의 각 데이터들에 특정 함수를 일괄적으로 적용하는 방법을 알아보자.

 

get_category함수를 정의해주자. 이 함수는 정수의 나이값을 문자열의 연령대로 바꿔주는 함수다. 이 함수를 lambda 문법으로 활용해 titanic_df의 age컬럼의 데이터들을 일괄적으로 바꿔줄 수 있다.

 

def get_category(age):
    cat = ''
    if age <= -1: cat = 'Unknown'
    elif age <= 5: cat = 'Baby'
    elif age <= 12: cat = 'Child'
    elif age <= 18: cat = 'Teenager'
    elif age <= 25: cat = 'Student'
    elif age <= 35: cat = 'Young Adult'
    elif age <= 60: cat = 'Adult'
    else : cat = 'Elderly'

    return cat

titanic_df['Age_cat'] = titanic_df['Age'].apply(lambda x : get_category(x))