dask-ml 中单列的插补平均值

问题描述

在更改所有 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 不变怎么办？是否可以指定要估算的列？

解决方法

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