728x90
반응형
Dataframe에서 상황에 따라 one-hot encoding을 label encoding으로 변환해서 사용해야하는 경우가 있다.
One hot encoding을 사용할 때는 get_dummies를 사용하면 쉽게 변환이 가능하다.
반대로 label encoding으로 변환하고 싶다면 다음과 같은 방식으로 손쉽게 변환이 가능하다!
onehot_encoding_list = [...] # one_hot encoding으로 표현되어 있는 feature 이름
df['label_encoding'] = np.argmax(np.asarray(df[onehot_encoding_list]), axis=1)728x90
반응형