K Mean Clustering Python Code