Knn Classification Python Code