Add unique index to label in fmuobs when converting from ResInsight f…

…ormat to be ERT compatible
equinor · Dec 6, 2023 · 83a321a · 83a321a
1 parent d0f04a1
commit 83a321a
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 1 deletion.
diff --git a/src/subscript/fmuobs/parsers.py b/src/subscript/fmuobs/parsers.py
@@ -462,7 +462,11 @@ def resinsight_df2df(ri_dframe: pd.DataFrame) -> pd.DataFrame:
 
     dframe = ri_dframe.copy()
     dframe.rename({"VECTOR": "KEY"}, axis="columns", inplace=True)
-    dframe["LABEL"] = dframe["KEY"].astype(str)  # + "-" + dframe["DATE"].astype(str)
+    dframe["LABEL"] = (
+        dframe["KEY"].astype(str)
+        + "-"
+        + (dframe.groupby("KEY").cumcount() + 1).astype(str)
+    )
     dframe["CLASS"] = "SUMMARY_OBSERVATION"
     if "DATE" in dframe:
         dframe["DATE"] = pd.to_datetime(dframe["DATE"])

diff --git a/tests/testdata_fmuobs/ri-obs.csv b/tests/testdata_fmuobs/ri-obs.csv
@@ -1,2 +1,4 @@
 DATE;VECTOR;VALUE;ERROR
 2005-08-21;GOPR:BRENT;100.0;5.0
+2005-08-21;GWPR:BRENT;50.0;5.0
+2006-08-21;GOPR:BRENT;200.0;5.0