def label_to_idx(labels, label): center_idx_bool = labels == label return np.asscalar(np.where(center_idx_bool)[0]), center_idx_bool