From 7441bd044f1efff01ed38bba12bd71d34fd71d95 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?S=C3=B6ren=20Henning?= <post@soeren-henning.de>
Date: Wed, 6 May 2020 15:42:19 +0200
Subject: [PATCH] Simplify plot creation

---
 execution/scalability-graph-finish.ipynb | 20 +++++++++-----------
 1 file changed, 9 insertions(+), 11 deletions(-)

diff --git a/execution/scalability-graph-finish.ipynb b/execution/scalability-graph-finish.ipynb
index e46f4b676..ffcf33b6b 100644
--- a/execution/scalability-graph-finish.ipynb
+++ b/execution/scalability-graph-finish.ipynb
@@ -7,9 +7,8 @@
    "outputs": [],
    "source": [
     "import os\n",
-    "import requests\n",
-    "from datetime import datetime, timedelta, timezone\n",
     "import pandas as pd\n",
+    "from functools import reduce\n",
     "import matplotlib.pyplot as plt"
    ]
   },
@@ -19,11 +18,12 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "exp_id = 1025\n",
-    "warmup_sec = 60\n",
-    "warmup_partitions_sec = 120\n",
-    "threshold = 2000 #slope\n",
-    "directory = '../results-inst'\n"
+    "directory = '../results-inst'\n",
+    "\n",
+    "experiments = {\n",
+    "    'exp1003': 'exp1003',\n",
+    "    'exp1025': 'exp1025',\n",
+    "}\n"
    ]
   },
   {
@@ -32,11 +32,9 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "df1 = pd.read_csv(os.path.join(directory, 'exp1025_min-suitable-instances.csv')).set_index('dim_value').rename(columns={\"instances\": \"1025\"})\n",
-    "df2 = pd.read_csv(os.path.join(directory, 'exp1003_min-suitable-instances.csv')).set_index('dim_value').rename(columns={\"instances\": \"1003\"})\n",
+    "dataframes = [pd.read_csv(os.path.join(directory, f'{v}_min-suitable-instances.csv')).set_index('dim_value').rename(columns={\"instances\": k}) for k, v in experiments.items()]\n",
     "\n",
-    "#df1.join(df2, on='instances')\n",
-    "df = df1.join(df2, how='outer')\n",
+    "df = reduce(lambda df1,df2: df1.join(df2,how='outer'), dataframes)\n",
     "\n",
     "df"
    ]
-- 
GitLab