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

37 votes330 views6 answers
Drop columns with nan pandas
Python
-2 votes354 views1 answer
Calculate sum in 2d list python
Python
0 vote327 views1 answer
Flask remove file after send_file
Python
3 votes362 views2 answers
Df.loc
Python
1 vote331 views1 answer
Frequency
Python
12 votes309 views1 answer
Python regex cheat sheet
Python
2 votes345 views3 answers
Fibonacci series python using function
Python
0 vote316 views1 answer
Python tkinter get image size
Python
2 votes312 views1 answer
How to replace a word in csv file using python
Python
1 vote323 views1 answer
Pandas change dtype to timestamp
Python