francois/HowBadIsMyBatch
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

refactoring

Browse Source
This commit is contained in:
frankknoll
2022-03-03 13:41:54 +01:00
parent f452682208
commit 824be3fcad
1 changed files with 11 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
src/intensivstationen/Intensivstationen.ipynb
View File

@@ -227,23 +227,23 @@
"\n", "\n",
"\n", "\n",
"def getAndPersistIntensiveCareBeds(timeSeries, kreis = None):\n", "def getAndPersistIntensiveCareBeds(timeSeries, kreis = None):\n",
" dataFrame = getIntensiveCareBeds(timeSeries, kreis)\n", " intensiveCareBeds = getIntensiveCareBeds(timeSeries, kreis)\n",
" display(kreis)\n", " display(kreis)\n",
" _saveDataFrameAsJson(dataFrame, _getFilename(kreis))\n", " _saveAsJson(intensiveCareBeds, _getFilename(kreis))\n",
" return dataFrame\n", " return intensiveCareBeds\n",
"\n", "\n",
"\n", "\n",
"def _saveDataFrameAsJson(dataFrame, file):\n", "def _saveAsJson(intensiveCareBeds, file):\n",
" IOUtils.saveDictAsJson(\n", " IOUtils.saveDictAsJson(\n",
" {\n", " {\n",
" 'population': int(dataFrame.iloc[0]['Einwohnerzahl']),\n", " 'population': int(intensiveCareBeds.iloc[0]['Einwohnerzahl']),\n",
" 'data': _dataFrame2Dict(dataFrame),\n", " 'data': _intensiveCareBeds2Dict(intensiveCareBeds),\n",
" },\n", " },\n",
" file)\n", " file)\n",
"\n", "\n",
"\n", "\n",
"def _dataFrame2Dict(dataFrame):\n", "def _intensiveCareBeds2Dict(intensiveCareBeds):\n",
" df = dataFrame[['date', 'betten_belegt', 'betten_frei']]\n", " df = intensiveCareBeds[['date', 'betten_belegt', 'betten_frei']]\n",
" df['date'] = df['date'].dt.strftime('%Y-%m-%d')\n", " df['date'] = df['date'].dt.strftime('%Y-%m-%d')\n",
" return df.to_dict(orient = \"records\")\n", " return df.to_dict(orient = \"records\")\n",
"\n", "\n",