python - 在 dask-ml 中估算单列的平均值

Question

在更改以下所有列时，使用 dask-ml 计算和估算平均值可以正常工作np.nan：

imputer = impute.SimpleImputer(strategy='mean')
data = [[100, 2], [np.nan, np.nan], [70, 7]]
df = pd.DataFrame(data, columns = ['Weight', 'Age']) 
x3 = imputer.fit_transform(df)
print(x3)

    Weight  Age
 0  100.0   2.0
 1  85.0    4.5
 2  70.0    7.0

但是，如果我需要Age保持原样怎么办？是否可以指定要估算的列？

Answer 1

您应该能够通过 df.Weight = imputer.fit_transform(df.Weight)或通过索引列来指定列df.loc["Weight"]