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-02-01 09:13:22 +01:00
parent 4142e84bdb
commit 20915c14b8
1 changed files with 11 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
HowBadIsMyBatch.ipynb
View File

@@ -100,14 +100,19 @@
" self.dataFrame = dataFrame \n", " self.dataFrame = dataFrame \n",
"\n", "\n",
" def filterBy(self, manufacturer = None, dose = None):\n", " def filterBy(self, manufacturer = None, dose = None):\n",
" # FK-TODO: extract private instance methods\n", " return self.dataFrame[self._isCovid19() & self._isManufacturer(manufacturer) & self._isDose(dose)]\n",
" isCovid19 = self.dataFrame[\"VAX_TYPE\"] == \"COVID19\"\n",
" isManufacturer = self.dataFrame[\"VAX_MANU\"] == manufacturer if manufacturer is not None else True\n",
" isDose = self.dataFrame[\"VAX_DOSE_SERIES\"].str.contains(dose) if dose is not None else True\n",
" return self.dataFrame[isCovid19 & isManufacturer & isDose]\n",
"\n", "\n",
" def filterForSevereEffects(self, dose):\n", " def filterForSevereEffects(self, dose):\n",
" return self.filterBy(dose = dose)\n" " return self.filterBy(dose = dose)\n",
"\n",
" def _isCovid19(self):\n",
" return self.dataFrame[\"VAX_TYPE\"] == \"COVID19\"\n",
"\n",
" def _isManufacturer(self, manufacturer):\n",
" return self.dataFrame[\"VAX_MANU\"] == manufacturer if manufacturer is not None else True\n",
"\n",
" def _isDose(self, dose):\n",
" return self.dataFrame[\"VAX_DOSE_SERIES\"].str.contains(dose) if dose is not None else True\n"
] ]
}, },
{ {