[Python] Pandas 다루기 - 카테고리 데이터

import pandas as pd

 

# 데이터 생성

ages = [0, 2, 10, 21, 23, 37, 31, 61, 20, 41, 32]
# 구간 설정 : 맨 앞을 0으로 하면 0이라는 데이터가 결측값으로 출력되기 때문에 -1
bins = [-1, 15, 25, 35, 60, 99]
labels = ['미성년자', '청년', '중년', '장년', '노년']

cats = pd.cut(ages, bins, labels = labels)

print(cats)
print('')
print(type(cats))

# 데이터 프레임 생성

ages_df = pd.DataFrame(ages, columns = ['ages'])
ages_df

# 데이터 프레임에 카테고리 적용

ages_df['cat'] = cats
ages_df