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

86 votes135 views6 answers
Select random value from list python
Python
0 vote120 views1 answer
Pandas rearrange rows based on datetime index
Python
-2 votes153 views1 answer
Jwt authentication python flask
Python Flask
7 votes118 views2 answers
How to get the remainder in python
Python
-1 vote118 views1 answer
Map numpy array
Python
11 votes121 views4 answers
Create a 2d array in python
Python
0 vote143 views1 answer
Migrate using other database django
Python Django
0 vote121 views1 answer
Zeller year
Python
0 vote104 views2 answers
If something in something python example
Python
2 votes110 views1 answer
Pandas new df from groupby
Python