머신러닝 get_dummies()

데이터셋의 자료형은 대부분 수치형(정수나 실수 등)을 사용해 표현한다. 수치형이 데이터를 불러와 평균을 계산하거나 plot으로 표현하기에 편하기 때문이다.  하지만 어쩔 수 없이 여러 속성값을 이용해 표현해 놓은 데이터가 있기 마련이다. 이런 데이터들을 속성 값에 따라 수치형으로 바꿔 표현해주는 get_dummies() 함수가 있다.

 

위 데이터프레임에서 paymentMethod는 creditcard, storecredit, paypal 3가지의 속성값을 가진다. 이 속성값이 사용되었는지에 따라 0이나 1로 표현해주는 함수이다.

import pandas as pd

df = pd.read_csv('/Users/a111/ML/book-resources-master/chapter2/datasets/payment_fraud.csv')
df.info()
df = pd.get_dummies(df,columns=['paymentMethod'])   #
df.sample(3)

paymentMethod_creditcard 등으로 따로 열을 만들어서 0이나 1로 표현해준다.