Scikit-learn Imputer缩小尺寸

Question

我有一个332栏的数据框。我想要推测价值，以便能够使用scikit-learn的决策树分类器。我的问题是从imputer函数所得到的数据的列仅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) 

Answer 1

按照documentation of sklearn.preprocessing.Imputer：

当轴= 0，这仅包含缺失在拟合值的列被丢弃经过改造。

因此，这是删除所有缺少值列。