From 0c105a2f8e96d8d6cfd199e721c036c0f536ce6d Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Rok=20Ro=C5=A1kar?= <rok.roskar@sdsc.ethz.ch>
Date: Sun, 22 Nov 2020 23:56:11 +0000
Subject: [PATCH] chore: update italy notebook
---
notebooks/covid-19-italy.ipynb | 8 +++++---
1 file changed, 5 insertions(+), 3 deletions(-)
diff --git a/notebooks/covid-19-italy.ipynb b/notebooks/covid-19-italy.ipynb
index acc41431d..3be3bd78a 100644
--- a/notebooks/covid-19-italy.ipynb
+++ b/notebooks/covid-19-italy.ipynb
@@ -157,8 +157,10 @@
"outputs": [],
"source": [
"# Total cases and deaths per 100k population\n",
+ "from datetime import datetime\n",
+ "start_date = datetime.fromisoformat('2020-08-01')\n",
"\n",
- "base = alt.Chart(df_regions).encode(alt.X('date'))\n",
+ "base = alt.Chart(df_regions.loc[(df_regions.date > start_date)]).encode(alt.X('date'))\n",
"death = base.mark_line(color='black').encode(alt.Y('deceased', title='Deaths'))\n",
"total = base.mark_bar(color='orange', size=5).encode(alt.Y('positive', title=\"Total cases\")).properties(\n",
" width=150,\n",
@@ -184,7 +186,7 @@
"source": [
"# Total cases and deaths per 100k population\n",
"\n",
- "base = alt.Chart(df_regions).encode(alt.X('date'))\n",
+ "base = alt.Chart(df_regions.loc[(df_regions.date > start_date)]).encode(alt.X('date'))\n",
"death = base.mark_line(color='black').encode(alt.Y('deceased_100k', title='Deaths'))\n",
"total = base.mark_bar(color='orange', size=5).encode(alt.Y('positive_100k', title=\"Total cases\")).properties(\n",
" width=150,\n",
@@ -208,7 +210,7 @@
"outputs": [],
"source": [
"# compute daily differences\n",
- "tdf = df_regions.sort_values(['region_label', 'date'], ascending=[True, False]).set_index(['region_label', 'date'])\n",
+ "tdf = df_regions.loc[(df_regions.date > start_date)].sort_values(['region_label', 'date'], ascending=[True, False]).set_index(['region_label', 'date'])\n",
"diffs_df = tdf[['positive', 'deceased']].groupby(level='region_label').diff(periods=-1).dropna(how='all')\n",
"tdf_diff=tdf.join(diffs_df, rsuffix='_diff').reset_index()\n"
]
--
GitLab