python - df.fillna : Gives Error because of replace value for NaN not present in categorical attributes

Question

df.fillna('unknown',inplace=True)

我有一个数据框 df，其中一些属性是分类属性，一些属性是对象。使用上面的表达式，我尝试用“未知”填充任何这些列中的缺失值。但这样做时我收到一个错误，说“未知”不存在于分类数据列的类别中。

我该怎么做？

Answer 1

使用DataFrame.select_dtypes并循环遍历每个分类列，并使用 add_categories 添加“未知”作为类别.cat 访问器上的方法。

for col in df.select_dtypes('category'):
    df[col].cat.add_categories('unknown', inplace=True)

df.fillna('unknown',inplace=True)