Etsy interview question

implement k means in vanilla python