본문 바로가기
5-3. Pandas 제공 자료구조/DataFrame

[파이썬 판다스] 데이터프레임에서 매칭된 데이터 추가하기 (map)

by 만다린망고 2022. 1. 20.
반응형

 판다스를 불러오고 결측치가 포함된 데이터프레임을 만들어줍니다. 

import pandas as pd

df1=pd.DataFrame({'C1':['미국','미국','한국','일본','한국'],'C2':['Javis','John','김','사나','정']},
    index=['R1','R2','R3','R4','R5'])

>>> df1
    C1     C2
R1  미국  Javis
R2  미국   John
R3  한국      김
R4  일본     사나
R5  한국      정


세번째 열에 국적을 영어로 추가하고 싶습니다. 물론 ['USA','USE','KOREA','JAPAN',KOREA'] 를 직접 정의하여 추가해줄 수도 있지만, 행이 많은 경우는 불편합니다. 편하게 추가하는 방법은 아래와 같습니다. 

먼저 매칭을 정의합니다. 

ENG_NAME={'미국':'USA','한국':'KOREA','일본':'JAPAN'}


1열에 map 메소드를 사용하면 아래와 같습니다. 

>>> df1['C1'].map(ENG_NAME) 
R1      USA
R2      USA
R3      NaN
R4    JAPAN
R5    KOREA
Name: C1, dtype: object


위 결과를 3열에 추가합시다. 

>>> df1['C3']=df1['C1'].map(ENG_NAME)
>>> df1
    C1     C2     C3
R1  미국  Javis    USA
R2  미국   John    USA
R3  힌국      김    NaN
R4  일본     사나  JAPAN
R5  한국      정  KOREA
반응형

댓글