如何在python中将列数据类型int64转换为分类列数据类型？

Question

how can i change int to categorical

import pandas as pd
import numpy as np

data = pd.read_excel('data.xlsx',header=0)
data.info()

there is now a column damage which is int64. It shows different damage-groups. How can this column be convert to a categorical column? (background is, there are 4 damage groups. 1 not really damage, 4 is totally damage).

I tried:

data['damage']= data.damage.astype('categorical')

*data type 'categorical' not understood*

but it doesn´t work. This works for example with ('float') or anything else. How can the column damage transfer to a categorical column?

Answer 1

According to pandas documentation categorical Series or columns in a DataFrame can be created by several ways. One way is converting an existing Series or column to a category dtype.
So change:

data['damage']= data.damage.astype('categorical')

To:

data['damage'] = data.damage.astype('category')