Scikit learn k means python example