我如何以及使用哪个库来使用knn插补进行缺失值分析

时间:2019-12-26 14:39:20

标签: python missing-data knn imputation

我正在尝试使用knn来插补缺失的值,但我无法使用 码: 来自fancyimpute进口KNN
还有knn插补的其他库吗?

1 个答案:

答案 0 :(得分:0)

备用库是sklearn

from sklearn.impute import KNNImputer

来自skleanr's documentation的示例:

import numpy as np
from sklearn.impute import KNNImputer

X = [[1, 2, np.nan], [3, 4, 3], [np.nan, 6, 5], [8, 8, 7]]
imputer = KNNImputer(n_neighbors=2)
imputer.fit_transform(X)

输出:

array([[1. , 2. , 4. ],
       [3. , 4. , 3. ],
       [5.5, 6. , 5. ],
       [8. , 8. , 7. ]])