如何在python中將列數據類型int64轉換為分類列數據類型？

Question

我如何將 int 更改為分類

import pandas as pd
import numpy as np

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

現在有一個 int64 列損壞。 它顯示了不同的傷害組。 如何將此列轉換為分類列？ （背景是，有 4 個傷害組。1 不是真正的傷害，4 是完全傷害）。

我試過：

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

*data type 'categorical' not understood*

但它不起作用。 例如，這適用於 ('float') 或其他任何東西。 列損壞如何轉移到分類列？

Answer 1

根據DataFrame 文檔， DataFrame分類Series或列可以通過多種方式創建。 一種方法是將現有的Series或列轉換為category dtype。
所以改變：

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

到：

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