Canva interview question

implement k-means with native python