Knn Algorithm Python