Nearest neighbors imputation

# Nearest neighbors imputation

import numpy as np
from sklearn.impute import KNNImputer
nan = np.nan
X = [[1, 2, nan], [3, 4, 3], [nan, 6, 5], [8, 8, 7]]
imputer = KNNImputer(n_neighbors=2, weights="uniform")
imputer.fit_transform(X)
# array([[1. , 2. , 4. ],
#        [3. , 4. , 3. ],
#        [5.5, 6. , 5. ],
#        [8. , 8. , 7. ]])

Source: scikit-learn.org

Similar codes

0 vote267 views1 answer
Drawing rectangle with border only in matplotlib
Python
0 vote266 views1 answer
How to make text bold in tkinter
Python
2 votes274 views1 answer
Load python file in jupyter notebook
Python
0 vote271 views1 answer
Matrix of matrices python grepper
Python
0 vote267 views1 answer
Data type of none in python
Python
0 vote250 views1 answer
Data dictionary python into numpy
Python
1 vote262 views2 answers
Factors addition in pyhone
Python
0 vote273 views1 answer
Log log grid python
Python
0 vote270 views1 answer
Xlabel not showing matplotlib
Python
21 votes302 views4 answers
Append to pandas dataframe
Python