https://analyticsarora.com/k-means-for-beginners-how-to-build-from-scratch-in-python/