python - 缺少数据的 Pandas 分类变量

Question

假设我有这个数据框：

dfdic = {"col1": ['azul', 'amarillo', 'amarillo', np.nan], "col2": [4, 5, 8, 10]}
df = pd.DataFrame(dfdic)

我想将该col1字段转换为虚拟变量。我可以这样做：

pd.get_dummies(df, columns=['col1']).head()

这使

    col2    col1_amarillo   col1_azul
0   4.0     0               1
1   5.0     1               0
2   8.0     1               0
3   10      0               0

虚拟变量中的 NaNcol1已替换为两个零。这是有道理的，因为它表示该实例不属于任何类别。但是，我怎样才能用 NaN 替换那些零，所以我可以

    col2    col1_amarillo   col1_azul
0   4.0     0               1
1   5.0     1               0
2   8.0     1               0
3   10      NaN             NaN

标签： pythonpandasmissing-data

Answer 1

mask+isnull

您可以使用mask使选定的列 null 依赖于另一个系列。

df.iloc[:, 1:] = df.iloc[:, 1:].mask(df['col2'].isnull())

print(df)

   col2  col1_amarillo  col1_azul
0   4.0            0.0        1.0
1   5.0            1.0        0.0
2   8.0            1.0        0.0
3   NaN            NaN        NaN