From f7291b75adabfa490add87ea4ea1d8a50412d1df Mon Sep 17 00:00:00 2001
From: Pascal <pnieters@uos.de>
Date: Wed, 2 Sep 2020 16:09:35 +0200
Subject: [PATCH] timedelta & check if map.csv exists
---
src/add_rki_to_map.py | 8 ++++++--
1 file changed, 6 insertions(+), 2 deletions(-)
diff --git a/src/add_rki_to_map.py b/src/add_rki_to_map.py
index 25af464..80141dd 100644
--- a/src/add_rki_to_map.py
+++ b/src/add_rki_to_map.py
@@ -15,8 +15,12 @@ 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')
+ if not os.path.exists(map_csv_path):
+ continue
+
+ day_0 = pd.Timestamp(date.replace('_','-')) + pd.Timedelta(days=26)
- map_rki = data.loc[date.replace('_', '-')].values.astype('float64')
+ map_rki = data.loc[day_0].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
@@ -24,4 +28,4 @@ for date in dates:
map_data = pd.read_csv(map_csv_path, index_col=0)
map_data["newInf100k_RKI"] = list(map_rki)
- map_data.to_csv(map_csv_path)
+ # map_data.to_csv(map_csv_path)
--
GitLab