Machine Learning -- K-Nearest Neighbors algorithm with Python