SimpleImputer不会删除熊猫数据框中的NaN值

时间:2020-09-14 16:35:35

标签: python pandas numpy dataframe scikit-learn

我一直在处理Titanic数据集,并尝试使用SimpleImputer类删除NaN值。不幸的是,它似乎没有做任何事情。我发现了一个与此相关的较旧的文章,但似乎没有答案。这是我的代码:

import numpy as np
from sklearn.impute import SimpleImputer

imp = SimpleImputer( missing_values = np.nan, strategy='median')

imp.fit_transform(X_train)
imp.fit_transform(y_train)

X_train是具有属性Sex(作为整数),Age,SibSp,Parch,Pclass和Fare的数据框。 y_train包含生存信息。

0 个答案:

没有答案