refactoring

This commit is contained in:
frankknoll
2022-02-07 21:18:33 +01:00
parent 4900629f42
commit c3b459d5f7

View File

@@ -109,7 +109,7 @@
" @staticmethod\n", " @staticmethod\n",
" def normalize(dataFrame):\n", " def normalize(dataFrame):\n",
" DataFrameNormalizer.convertVAX_LOTColumnToUpperCase(dataFrame)\n", " DataFrameNormalizer.convertVAX_LOTColumnToUpperCase(dataFrame)\n",
" DataFrameNormalizer._convertColumnsOfDataFrameToNumerics(\n", " DataFrameNormalizer._convertColumnsOfDataFrame_Y_to_1_else_0(\n",
" dataFrame,\n", " dataFrame,\n",
" ['DIED', 'L_THREAT', 'DISABLE', 'HOSPITAL', 'ER_VISIT'])\n", " ['DIED', 'L_THREAT', 'DISABLE', 'HOSPITAL', 'ER_VISIT'])\n",
"\n", "\n",
@@ -118,12 +118,12 @@
" dataFrame['VAX_LOT'] = dataFrame['VAX_LOT'].str.upper()\n", " dataFrame['VAX_LOT'] = dataFrame['VAX_LOT'].str.upper()\n",
"\n", "\n",
" @staticmethod\n", " @staticmethod\n",
" def _convertColumnsOfDataFrameToNumerics(dataFrame, columns):\n", " def _convertColumnsOfDataFrame_Y_to_1_else_0(dataFrame, columns):\n",
" for column in columns:\n", " for column in columns:\n",
" DataFrameNormalizer._convertColumnOfDataFrameToNumeric(dataFrame, column)\n", " DataFrameNormalizer._convertColumnOfDataFrame_Y_to_1_else_0(dataFrame, column)\n",
"\n", "\n",
" @staticmethod\n", " @staticmethod\n",
" def _convertColumnOfDataFrameToNumeric(dataFrame, column):\n", " def _convertColumnOfDataFrame_Y_to_1_else_0(dataFrame, column):\n",
" dataFrame[column] = DataFrameNormalizer._where(\n", " dataFrame[column] = DataFrameNormalizer._where(\n",
" condition = dataFrame[column] == 'Y',\n", " condition = dataFrame[column] == 'Y',\n",
" trueValue = 1,\n", " trueValue = 1,\n",