如何在sklearn中修复此自定义转换器？

Question

我已经编写了这个简单的自定义转换器，该转换器在特定的列中用0填充na。当我在数据集上执行fit_transform时，它没有在指定的列中填充nas。我在代码中看不到问题。

<label>{{currentMonth}}</label>
<button (click)="nextMonth()">Next</button>

我希望返回的数据帧是一个指定的列，其中nas用0填充了nas，但是我得到了一个具有相同空值的数据帧。

Answer 1

您从未将fillna操作分配给新变量。使用

X[col].fillna(0)

不会就地发生。而是使用：

X.loc[:,col] = X[col].fillna(0)

Answer 2

一个班轮解决方案是

cols = ['PavedDrive', 'GarageQual', 'GarageFinish', 'FireplaceQu', 'KitchenQual', 'CentralAir', 'HeatingQC', 'BsmtExposure', 'BsmtCond', 'BsmtQual', 'ExterCond', 'ExterQual', 'Street']

X.loc[:,cols] = X[cols].fillna(0)