refactoring

This commit is contained in:
frankknoll
2022-02-04 09:18:59 +01:00
parent d61817957e
commit c41459908e
2 changed files with 8 additions and 5 deletions

View File

@@ -128,11 +128,7 @@
" \n",
" def __init__(self, dataFrame : pd.DataFrame):\n",
" self.dataFrame = dataFrame \n",
" self._convertColumnOfDataFrameToNumeric('DIED')\n",
" self._convertColumnOfDataFrameToNumeric('L_THREAT')\n",
" self._convertColumnOfDataFrameToNumeric('DISABLE')\n",
" self._convertColumnOfDataFrameToNumeric('HOSPITAL')\n",
" self._convertColumnOfDataFrameToNumeric('ER_VISIT')\n",
" self._convertColumnsOfDataFrameToNumerics(['DIED', 'L_THREAT', 'DISABLE', 'HOSPITAL', 'ER_VISIT'])\n",
"\n",
" def createBatchCodeTable(self):\n",
" batchCodeTable = self.dataFrame.groupby('VAX_LOT').agg(\n",
@@ -193,6 +189,10 @@
" manufacturerByBatchCodeTable = manufacturerByBatchCodeTable.drop_duplicates(subset = ['VAX_LOT'])\n",
" return manufacturerByBatchCodeTable.set_index('VAX_LOT')\n",
"\n",
" def _convertColumnsOfDataFrameToNumerics(self, columns):\n",
" for column in columns:\n",
" self._convertColumnOfDataFrameToNumeric(column)\n",
"\n",
" def _convertColumnOfDataFrameToNumeric(self, column):\n",
" self.dataFrame[column] = np.where(self.dataFrame[column] == 'Y', 1, 0)\n",
"\n",