A flexible python implementation of the simple kmeans clustering algorithm