Data Clustering In Python