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

Question

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

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"]