SimpleImputer 如何替换 dataframe 中的 nan 值？

Question

我有一个形状（10,3）的ndarray，其中包含一些缺失值（nan）。 我想用相应列的平均值替换缺失值。 为此，我使用了 SimpleImpueter。 我的代码如下：

from sklearn.impute import SimpleImputer
imputer = SimpleImputer(missing_values = 'nan',strategy='mean')

此代码不起作用并显示以下值错误：

ValueError: Input contains NaN, infinity or a value too large for dtype('float64').

有什么建议可以解决这个问题吗？