k-Nearest Neighbor (kNN) algorithm in Python with a sample dataset