[python/파이썬] 데이터 전처리 - apply, map 을 활용한 데이터 변환

apply, map 을 활용한 데이터 변환

import pandas as pd

df = pd.DataFrame({'a' : [1, 2, 3, 4, 5]})

-문제 : a 가 2 보다 작으면 '2 미만', 4 보다 작으면 '4 미만', 4 보다 크면 '4 이상' 이 저장된 b 칼럼을 추가하기

df

df['b'] = 0

df

a = df[df['a] < 2]

a

df['b'][a.index] = '2 미만'

df

a = df[(df['a'] >= 2) & (df['a'] < 4)]

a

df['b'][a.index] = '4 미만'

pd.set_option('mode.chained_assignment',  None)

df['b'][a.index] = '4 미만'

df

a = df[df['a'] >= 4

df['b'][a.index] = '4 이상'

df

1. 함수 + apply 를 이용한 해결

• apply()

1. 데이터프레임을 input으로 받는 함수 생성
2. 반영하고 싶은 column 이름을 가져와 적용하고 싶은 함수 내용 작성
3. 원하는 df에 apply 함수로 적용하고 위에서 작성한 함수 적용

def case_function(x):
    if x < 2:
        return '2 미만'
    elif x < 4:
        return '4 미만'
    else:
        return '4 이상'

df['c'] = df['a'].apply(case_function)

df

-문제 : a 가 1 이면 'one', 2 이면 'two', 3 이면 'three', 4 이면 'four', 5 이면 'five' 를 출력하는 칼럼 d 를 만들기

df

• 사용자 정의함수를 이용한 해결방법

def function(x):
    if x == 1:
        return 'one'
    elif x == 2:
        return 'two'
    elif x == 3:
        return 'three'
    elif x == 4:
        return 'four'
    elif x == 5:
        return 'five'

df['d'] = df['a'].apply(function)

df

2. map을 이용한 해결 방법

• map() : 여러 개의 데이터를 한 번에 다른 형태로 변환하기 위해서 사용. 여러 개의 데이터를 담고 있는 list나 tuple을 대상으로 주로 사용하는 함수

a = { 1 : 'one', 2 : 'two', 3 : 'three', 4 : 'four', 5 : 'five'}

df['e'] = df['a'].map(a)

df