Scikit-learn - 输入特定列中的值

Question

是否可以估算特定列的值？

例如，如果我有 3 列：

Answer 1

您可以使用numpy.ravel：

from sklearn.preprocessing import Imputer
imp = Imputer(missing_values=0, strategy="mean", axis=0)
df["C"] = imp.fit_transform(df[["C"]]).ravel()

另外，对于未来的查看者，请使用 SimpleImputer，因为 Imputer 类已被弃用。检查https://github.com/scikit-learn/scikit-learn/blob/8d7e849428a4edd16c3e2a7dc8a088f108986a17/sklearn/preprocessing/imputation.py#L64 (2认同)