From da8eb1d5bfed6ba538b965cfcc0cde1b11ce81c5 Mon Sep 17 00:00:00 2001
From: Pascal <pnieters@uos.de>
Date: Wed, 2 Sep 2020 13:53:52 +0200
Subject: [PATCH] readd
This reverts commit df7dcc887e0d2a77195988472c1982ecc024ff26.
---
src/add_rki_to_map.py | 34 ++++++++++++++++++++++++++++++++++
1 file changed, 34 insertions(+)
create mode 100644 src/add_rki_to_map.py
diff --git a/src/add_rki_to_map.py b/src/add_rki_to_map.py
new file mode 100644
index 0000000..4a6a113
--- /dev/null
+++ b/src/add_rki_to_map.py
@@ -0,0 +1,34 @@
+import os
+import pickle as pkl
+import pandas as pd
+
+data = pd.read_csv('../data/diseases/covid19.csv', sep=',', encoding='iso-8859-1', index_col=0)
+data.index = [pd.Timestamp(date) for date in data.index]
+
+with open('../data/counties/counties.pkl', 'rb') as f:
+ counties = pkl.load(f)
+
+shared_assets = '/p/project/covid19dynstat/shared_assets/'
+figures = os.path.join(shared_assets, 'figures/')
+dates = next(os.walk(figures))[1]
+
+# errors if the RKI based csv is out of date with the available folders
+for date in dates:
+ map_csv_path = os.path.join(figures, date, 'map.csv')
+
+ map_rki = data.loc[date.replace('_', '-')].values.astype('float64')
+ for (i, (key, _)) in enumerate(counties.items()):
+ n_people = counties[key]['demographics'][('total', 2018)]
+ map_rki[i] = map_rki[i] / n_people * 100000
+
+ map_data = pd.read_csv(map_csv_path, index_col=0)
+ map_data["newInf100k_RKI"] = list(map_rki)
+
+
+
+
+
+
+
+
+
--
GitLab