Scikit-learn Imputer 缩减维度

Question

Scikit-learn Imputer 缩减维度

4

我有一个数据框，其中有332列。我想进行值填充以便使用scikit-learn的决策树分类器。我的问题是，填充函数生成的数据列只有330列。

from sklearn.preprocessing import Imputer
imp = Imputer(missing_values='NaN', strategy='mean', axis=0)
cols = data.columns
new = imp.fit_transform(data)

print(data.shape,new.shape)
(34132, 332) (34132, 330)

- jrlund

你有没有所有列都是NaN的情况？ - Ami Tavory

是的，那是我的答案。谢谢。 - jrlund

1个回答

网页内容由stack overflow 提供, 点击上面的

可以查看英文原文，
原文链接

- Ami Tavory · Accepted Answer

根据sklearn.preprocessing.Imputer的文档：

当axis=0时，在拟合时只包含缺失值的列在转换时被丢弃。

因此，这将删除所有缺失值列。