Canva Interview Question

implement k-means with native python