数据预处理-缺失值

def solve():
    import numpy as np
    from sklearn.impute import SimpleImputer
    from sklearn.datasets import load_iris
    iris = load_iris()
    X = iris.data.copy().astype(float)
    X[0, 0] = np.nan
    X[5, 1] = np.nan
    imputer = SimpleImputer(strategy='mean')
    X_imputed = imputer.fit_transform(X)
    print(np.isnan(X_imputed).sum())