# renaming so that class 0 will appear as setosa and class 1 as versicolor df_long.loc[df_long.Group==0, 'Group'] = classes_names[0] df_long.loc[df_long.Group==1, 'Group'] = classes_names[1]