from util import get_customized_dataframe, paths_simuls_regimes df = get_customized_dataframe(paths_simuls_regimes.values()) df["letter"] = paths_simuls_regimes.keys() regimes = [ "Diffusive", "LAST", "Optimal", "Weakly stratified", "Passive scalar", ] regimes = {regime[0]: regime for regime in regimes} print(regimes) df["regime"] = [regimes[letter] for letter in df["letter"]] print(df)