Pandas 分类类型不起作用

Question

我厌倦了使用http://pandas.pydata.org/pandas-docs/stable/categorical.html中的文档将列更改为catgeory

df = pd.DataFrame({'A':[1,2,3,4,5], 'B':['a','b','c','d','e'], 'C':['A','B','A','B','A']})
df['C']=df['C'].astype('category')

如果我尝试通过类别

df['C']=df['C'].astype('category',categories=['A','B'])

说错了

TypeError: _astype() got an unexpected keyword argument 'categories'

将类别传递给 astype()的正确方法是什么？

Answer 1

现在，您现在需要通过CategorialDtype传递它，因为astype方法不再接受它们

from pandas.api.types import CategoricalDtype
df = pd.DataFrame({'A':[1,2,3,4,5], 'B':['a','b','c','d','e'], 'C':['A','B','A','B','A']})
df['C']=df['C'].astype(CategoricalDtype(categories=['A','B']))