Python assignment 5 - k-cluster analysis