将熊猫系列转换为分类

Question

我有一个只有唯一ID的熊猫系列“ ID”，这是对象的dtype。

data_df.id.dtype

返回dtype（'O'）

我正在尝试按照此处的示例从df创建一个稀疏矩阵：Efficiently create sparse pivot tables in pandas?

id_u= list(data_df.id.unique())
row = data_df.id.astype('category', categories=reviewer_u).cat.codes

我得到：

TypeError: data type "category" not understood

我不确定这个错误是什么意思，而且我找不到很多。

Answer 1

请尝试：

row = pd.Categorical(data_df['id'], categories=reviewer_u)

您可以使用以下代码获取代码：

row.codes