Python | map, lambda 이용하여 파생변수 만들기

 

 

데이터프레임에서 기존에 있는 변수를 기반으로 파생변수를 만들고 싶은데 if문이나 for문을 이용해 복잡하게 구하기 싫을 때가 있다.

그럴 때 map과 lambda를 이용하면 아주 쉽게 해결이 가능하다.

 

import pandas as pd

df = pd.DataFrame([[1, 'a'], [2, 'b'], [3, 'c']], columns=['number', 'alphabet'])
df

 

이런 간단한 데이터프레임이 있다고 해보자.

여기서 'alphabet' 변수에 기반하여 새로운 파생변수를 만들어보겠다.

dict = {'a': '사과', 'b':'바나나', 'c':'전화'}

df['word'] = list(map(lambda x: dict[x], df.alphabet))
df

 

짠

각 a, b, c에 mapping되는 단어들을 나열한 새로운 변수가 생성되었다.

 

다들 알겠지만 lambda는 def로 긴 함수문을 만들지 않고도 비슷한 효과를 낼 수 있는 함수다.

map으로 뒤에서 input시킨 df의 alphabet의 값들을 하나씩 x로 대입하면 각 x에 dict의 key로 입력되고, 대응되는 value 값이 나온다.

이 과정을 df.alphabet의 row 길이 만큼 반복하고, 결과로 나온 값을 list화 시켰다.

 

정말 간단한건데 막상 내가 찾을 때 없길래 기록해둔다.

한 번 외워두면 아주 요긴하다.