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