在 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"]來指定列