Neighbors

KNeighborsClassifier

X = df['column'].values
X = df[['column1', 'column2']].values
knn = KNeighborsClassifier(n_neighbors=5)
knn.fit(X,y)