refactoring
This commit is contained in:
@@ -158,18 +158,8 @@
|
||||
"class SummationTableFactory:\n",
|
||||
"\n",
|
||||
" @staticmethod\n",
|
||||
" def createSummationTable(\n",
|
||||
" groupBy,\n",
|
||||
" columnNameMappingsDict = {\n",
|
||||
" \"DIED_size\": \"Adverse Reaction Reports\",\n",
|
||||
" \"DIED_sum\": \"Deaths\",\n",
|
||||
" \"L_THREAT_sum\": \"Life Threatening Illnesses\",\n",
|
||||
" \"DISABLE_sum\": \"Disabilities\",\n",
|
||||
" 'HOSPITAL_sum': 'Hospitalisations',\n",
|
||||
" 'ER_VISIT_sum': 'Emergency Room or Doctor Visits'\n",
|
||||
" }):\n",
|
||||
"\n",
|
||||
" summationTable = groupBy.agg({\n",
|
||||
" def createSummationTableHavingSevereReportsColumn(dataFrame):\n",
|
||||
" summationTable = dataFrame.agg({\n",
|
||||
" 'DIED': ['sum', 'size'],\n",
|
||||
" 'L_THREAT': 'sum',\n",
|
||||
" 'DISABLE': 'sum',\n",
|
||||
@@ -178,13 +168,7 @@
|
||||
" 'SEVERE': 'sum'\n",
|
||||
" })\n",
|
||||
" SummationTableFactory._flattenColumns(summationTable)\n",
|
||||
" return summationTable.rename(columns = columnNameMappingsDict)\n",
|
||||
"\n",
|
||||
" @staticmethod\n",
|
||||
" def createSummationTableHavingSevereReportsColumn(dataFrame):\n",
|
||||
" summationTable = SummationTableFactory.createSummationTable(\n",
|
||||
" dataFrame,\n",
|
||||
" columnNameMappingsDict = {\n",
|
||||
" summationTable = summationTable.rename(columns = {\n",
|
||||
" \"DIED_size\": \"Adverse Reaction Reports\",\n",
|
||||
" \"DIED_sum\": \"Deaths\",\n",
|
||||
" \"L_THREAT_sum\": \"Life Threatening Illnesses\",\n",
|
||||
@@ -362,8 +346,8 @@
|
||||
"\n",
|
||||
" @staticmethod\n",
|
||||
" def saveDataFrame(dataFrame, file):\n",
|
||||
" IOUtils.saveDataFrameAsExcelFile(dataFrame, file)\n",
|
||||
" IOUtils.saveDataFrameAsHtml(dataFrame, file)\n",
|
||||
" # IOUtils.saveDataFrameAsExcelFile(dataFrame, file)\n",
|
||||
" # IOUtils.saveDataFrameAsHtml(dataFrame, file)\n",
|
||||
" IOUtils.saveDataFrameAsJson(dataFrame, file)\n",
|
||||
"\n",
|
||||
" @staticmethod\n",
|
||||
|
||||
Reference in New Issue
Block a user