Pandas 기초(11)

import pandas as pd
date_list = [{'yyyy-mm-dd': '2000-06-27'},
         {'yyyy-mm-dd': '2002-09-24'},
         {'yyyy-mm-dd': '2005-12-20'}]
df = pd.DataFrame(date_list, columns = ['yyyy-mm-dd'])
df

해당 DataFrame에서 year만 추출하고싶을 때

def extract_year(column):
    return column.split("-")[0]

df['year'] = df.['yyyy-mm-dd'].apply(extract_year)

Result

현재 age라는 Column을 새로만들고, 현재년도에서 뺀 숫자를 알고싶을 때

def get_age(year, current_year):
    return current_year - int(year)

df['age'] = df['year'].apply(get_age, current_year=2018)

Result

또는 여러개의 함수를 리턴해주고싶을 때

def get_introduce(age, prefix, suffix):
    return prefix + str(age) + suffix

df['introduce'] = df['age'].apply(get_introduce, prefix="I am ", suffix = " years old")

Result

하나의 Column만 가지고왔는데 여러개의 Column을 조합하고싶을 때

def get_introduce_2(row):
    return "I was born in " + str(row.year) + " my age is " + str(row.age)

df['introduce'] = df.apply(get_introduce_2, axis=1)

axis=1은 열을 뜻한다. 

Result

감사합니다. Thank you!