Add workpackage-meta-indicator to workflow (#3)

iiasa · Feb 19, 2024 · ab23a8f · ab23a8f
1 parent 95c5902
commit ab23a8f
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/workflow.py b/workflow.py
@@ -16,4 +16,13 @@ def main(df: pyam.IamDataFrame) -> pyam.IamDataFrame:
     processor = RegionProcessor.from_directory(path=here / "mappings", dsd=dsd)
     df = process(df, dsd, processor=processor)
 
+    # assign meta indicator for scenario "work package" category
+    for model, scenario in df.index:
+        if "DIAG" in scenario:
+            df.meta.loc[(model, scenario), "Work package"] = "Diagnostic"
+        elif scenario.startswith("WP"):
+            df.meta.loc[(model, scenario), "Work package"] = (
+                "Work Package " + scenario[2:4]
+            )
+
     return df