Shorter kmeans init function using lodash