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

71 votes267 views4 answers
List loop python
Python
11 votes170 views2 answers
Append 1 colimn in pandas df
Python
0 vote172 views1 answer
Loop through list, find specific number and output element's index
Python
9 votes196 views5 answers
Join to dataframes pandas
Python
11 votes227 views3 answers
Python print list with newline
Python
10 votes180 views2 answers
Python re.search()
Python
0 vote199 views1 answer
Fibonacci logic in pthon in for loop
Python
0 vote165 views1 answer
Pandas normalize rows to max value
Python
0 vote229 views2 answers
Django ajax body to json
Python Django
0 vote215 views1 answer
Class inside class python
Python