K means clustering using python