본문 바로가기
머신러닝을 위한 파이썬 기초/판다스

[파이썬 판다스] 데이터프레임 열의 타입을 factor 로 변형

by bigpicture 2023. 9. 5.
반응형

데이터프레임의 특정 열의 타입을 factor로 바꾸는 방법을 알아봅시다. 특정 열의 타입을 범주형 타입으로 바꾸는 것입니다. 먼저 데이터프레임을 하나 정의합시다. 

import pandas as pd

L1=[[1,2,3],[4,5,6],[7,8,9],[10,11,12]]

df=pd.DataFrame(L1,columns=['C1','C2','C3'],index=['R1','R2','R3','R4'])


숫자 인덱스를 사용하여 열을 factor로 바꾸는 방법과 열 이름을 사용하여 바꾸는 방법이 있습니다. astype 메소드를 사용합니다. 

 

1. 숫자 인덱스 사용

세번째 열의 타입을 범

df.iloc[:,2]=df.iloc[:,2].astype('category')


결과는 아래와 같습니다. 3열이 범주형 타입으로 변경되었습니다. 

>>> df.dtypes  
C1       int64
C2       int64
C3    category
dtype: object

 

 

2. 열 이름 사용

df.loc[:,'C3']=df.loc[:,'C3'].astype('category')


결과는 동일합니다. 

반응형

댓글