def assign_to_closest(X, centers, metric='euclidean'): return np.argmin(pairwise_distances(X, centers, metric=metric), axis=1)