Fix plotting: compatibility with 4th link port

2024-01-24 15:48:33 +01:00 · 2024-01-24 15:48:33 +01:00 · d8d3b2f2ab
commit d8d3b2f2ab
parent 2bee36be87
1 changed files with 4 additions and 1 deletions
--- a/scripts/plot_summary.py
+++ b/scripts/plot_summary.py
@ -282,7 +282,10 @@ def plot_balances():
        # remove trailing link ports
        df.index = [
            i[:-1]
-            if ((i not in ["co2", "NH3", "H2"]) and (i[-1:] in ["0", "1", "2", "3"]))
+            if (
+                (i not in ["co2", "NH3", "H2"])
+                and (i[-1:] in ["0", "1", "2", "3", "4"])
+            )
            else i
            for i in df.index
        ]