Knn Algorithm Example Python