如何在特征为空的情况下插补均值？

Question

我想填写一个特征为空的平均值

我试过：

# Let's impute other NAs with mean value
for f in data:
   if data[f].isnull().sum() / data.shape[0] < threshold: np.isnan(data[f]) = data[f].mean()

…这会引发一个错误：

SyntaxError: can't assign to function call

我也试过了

data[f][np.isnan(data[f])] = data[f].mean()

但那更糟

我们将不胜感激

Answer 1

可以使用DataFrame.fillna直接填充nan。 它创建一个数据帧，用括号中指定的值替换nan

df.fillna(df.mean())将返回新的数据帧，因此必须写入df=df.fillna(df.mean())以保留它