[Python] One Hot Encoding을 활용하여 범주형 변수 변형하기

One Hot Encoding을 사용하여 범주형 변수를 논리값이 포함된 열로 나누어 줄 수 있다.

 

import pandas as pd

#1
pd.get_dummies(data['열 이름'], prefix = '원하는 추가 이름')

#2
pd.get_dummies(data = 데이터, columns = ['열 이름'], prefix = '원하는 추가 이름')

 

#2를 활용하여 기존의 데이터프레임에 추가할 수 있다.

 

prefix는 열에 속한 데이터 값 앞에 추가로 붙는 이름을 넣어준다.