feat: add option to keep or remove uncertainties when reading catalog

swiss-seismological-service · Aug 16, 2023 · f7b4c3c · f7b4c3c
1 parent 6f1c79e
commit f7b4c3c
Showing 1 changed file with 8 additions and 5 deletions.
diff --git a/catalog_tools/io/client.py b/catalog_tools/io/client.py
@@ -85,9 +85,12 @@ def get_events(self, start_time: Optional[datetime] = None,
         parser.parse(r.raw)
 
         df = Catalog.from_dict(catalog)
-        # if not include_uncertainty:
-        # wrong way round
-        # rgx = "(_uncertainty|_lowerUncertainty|" \
-        #       "_upperUncertainty|_confidenceLevel)$"
-        # df = df.filter(regex=rgx)
+
+        if not include_uncertainty:
+            rgx = "(_uncertainty|_lowerUncertainty|" \
+                "_upperUncertainty|_confidenceLevel)$"
+            # df = df.filter(regex=rgx)
+            cols = df.filter(regex=rgx).columns
+            df = df.drop(columns=cols)
+
         return df