assign clusters

def assign_clusters(X, clusters):

    cluster_labels = np.zeros(X.shape[0])
    for i in range(X.shape[0]):
        min_dist = np.inf
        for j in range(clusters.shape[0]):
            if np.linalg.norm(X[i] - clusters[j]) < min_dist:
                label = j
                min_dist = np.linalg.norm(X[i] - clusters[j])
        cluster_labels[i] = label
    return cluster_labels