map(lambda x: ~~)와 dictionary

Jungwoo Kim·2023년 11월 21일

np/pd

목록 보기
7/12

데이터프레임에 기존 변수를 이용해 변수를 만들 때, dictionary를 먼저 생성한 후 map() 메서드를 사용하면 가시성이 좋다.

예를 들어, 아래는 Income_Category의 카테고리를 map 함수를 이용하여 다음과 같이 변경하여 newIncome 컬럼에 mapping하는 코드이다.

dic = {
    'Unknown' : 'N',
    'Less than $40K' : 'a',
    '$40K - $60K' : 'b',
    '$60K - $80K' : 'c',
    '$80K - $120K' : 'd',
    '$120K +' : 'e'
	}
    
df['newIncome'] = df['Income_Category'].map(lambda x: dic[x])

0개의 댓글