python - 具有 Series 输入的类别列的 fillna 无法按预期工作

Question

我有一个类别列，我想用一个系列来填充。我试过这个：

df = pd.DataFrame({'key': ['a', 'b'], 'value': ['c', np.nan]})
df['value'] = df['value'].astype("category")
df['value'] = df['value'].cat.add_categories(df['key'].unique())
print(df['value'].cat.categories)
df['value'] = df['value'].fillna(df['key'])
print(df)

预期输出：

Index(['c', 'a', 'b'], dtype='object')
  key value
0   a     c
1   b     b

实际输出：

Index(['c', 'a', 'b'], dtype='object')
  key value
0   a     a
1   b     b

Answer 1

这似乎是一个错误，但幸运的是，解决方法非常简单。填充时，您必须将“值”视为字符串列。

df['value'] = pd.Categorical(
    df.value.astype(object).fillna(df.key), categories=df.stack().unique())
df

  key value
0   a     c
1   b     b

Answer 2

从文档中，分类数据将接受标量而不是系列，因此您可能需要将其转换回系列

df.value.astype('object').fillna(df.key) # then convert to category again
Out[248]: 
0    c
1    b
Name: value, dtype: object

value : 用于填充孔的标量值（例如 0）