成功解决(机器学习填补数值型缺失值时报错):TypeError: init() got an unexpected keyword argument ‘axis’
报错问题
我的代码:
from sklearn.impute import SimpleImputer
import numpy as np
def si():
sip = SimpleImputer(missing_values=np.nan, strategy='mean', axis=0)
data = sip.fit_transform([[1, 2],
[np.nan, 3],
[7, 6]]
)
print(data)
si()
报错内容:
Traceback (most recent call last):
File "E:/Python/1.py", line 68, in <module>
si()
File "E:/Python/1.py", line 62, in si
si = SimpleImputer(missing_values='Nan',strategy='mean',axis=0)
TypeError: __init__() got an unexpected keyword argument 'axis'
报错原因
报错原因:0.20新版功能:SimpleImputer取代了以前的sklearn.preprocessing.Imputer;而SimpleImputer参数中没有axis
SimpleImputer类参数如下:
SimpleImputer(missing_values=np.nan,strategy="mean",fill_value=None,verbose=0,copy=True,add_indicator=False)
解决方法
去掉代码中的axis=0参数,修改后代码如下:
from sklearn.impute import SimpleImputer
import numpy as np
def si():
sip = SimpleImputer(missing_values=np.nan, strategy='mean')
data = sip.fit_transform([[1, 2],
[np.nan, 3],
[7, 6]]
)
print(data)
si()
再次运行成功:
[[1. 2.]
[4. 3.]
[7. 6.]]