diff --git a/scripts/history_stagging.py b/scripts/history_stagging.py index 49bcd2f..c16c812 100644 --- a/scripts/history_stagging.py +++ b/scripts/history_stagging.py @@ -23,9 +23,11 @@ def extract_cat(cat: pd.DataFrame): def lot_naming(value): - if str(value).isnumeric(): - return str(value).zfill(2) - return "PC" + try: + v = int(value) + except ValueError: + return "PC" + return str(v).zfill(2) def trans_2017_2021( @@ -99,7 +101,7 @@ def trans_2023( df = df.assign( Débit=df["Débit"].fillna(0), Crédit=df["Crédit"].fillna(0), - Lot=lot_naming(df["Porte"]), + Lot=df["Porte"].apply(lot_naming), Année=year, ) return df[stagging_columns]