[python] 판다스에서 특정 컬럼에 속한 값 원하는 값으로 바꾸기

[python] 판다스에서 특정 컬럼에 속한 값 원하는 값으로 바꾸기

참고링크

• 판다스 iloc, loc 차이
• 판다스에서 특정 컬럼에 속하는 값 원하는 값으로 바꾸기
• 판다스 데이터 parquet 파일을 DB로 넣기

판다스에서 데이터 불러오기

import pandas as pd
df = pd.read_csv("./data/stella/star_classification_origin.csv")
df

해당 컬럼에서 유니크 값 확인

df['class'].unique()

array(['GALAXY', 'QSO', 'STAR'], dtype=object)

위 데이터에서 class 변수에 있는 값은 총 세가지로 문자열 데이터 입니다. 문자형 데이터인 위 값을 숫자로 바꾸겠습니다.

해당 컬럼에서 조건에 맞는 값 바꾸기

df.loc[(df['class']=='GALAXY'), 'class']=0
df.loc[(df['class']=='QSO'), 'class']=1
df.loc[(df['class']=='STAR'), 'class']=2
df

바뀐 값 확인

df['class'].unique()

array([0, 1, 2], dtype=object)