reset_index on measurements_amount() dataframe, fixes #64

Deltares · Mar 11, 2024 · 2f2f594 · 2f2f594
1 parent fad611e
commit 2f2f594
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/ddlpy/ddlpy.py b/ddlpy/ddlpy.py
@@ -178,16 +178,18 @@ def measurements_amount(location, start_date, end_date, period="Jaar"):
             # combine columns to a period string
             df["Groeperingsperiode"] = df["Groeperingsperiode.Jaarnummer"].apply(lambda x: f"{x:04d}")
             if period in ["Maand", "Dag"]:
-                df["Groeperingsperiode"] = df["Groeperingsperiode"] + "-" + df["Groeperingsperiode.Maandnummer"].apply(lambda x: f"{x:02d}")
+                df["Groeperingsperiode"] = (df["Groeperingsperiode"] + "-" + 
+                                            df["Groeperingsperiode.Maandnummer"].apply(lambda x: f"{x:02d}"))
             if period in ["Dag"]:
-                df["Groeperingsperiode"] = df["Groeperingsperiode"] + "-" + df["Groeperingsperiode.Dag"].apply(lambda x: f"{x:02d}")
+                df["Groeperingsperiode"] = (df["Groeperingsperiode"] + "-" + 
+                                            df["Groeperingsperiode.Dag"].apply(lambda x: f"{x:02d}"))
 
             # select columns from dataframe and append to list
             df = df[["Groeperingsperiode","AantalMetingen"]]
             df_list.append(df)
 
         # concatenate
-        amount_all = pd.concat(df_list).sort_values("Groeperingsperiode")
+        amount_all = pd.concat(df_list).sort_values("Groeperingsperiode").reset_index(drop=True)
         return amount_all