refactoring

src/HowBadIsMyBatch.ipynb

@@ -367,9 +367,8 @@
     "    \n",
     "    @staticmethod\n",
     "    def addSevereColumn(dataFrame):\n",
-    "        dataFrame['SEVERE'] = dataFrame.apply(\n",
-    "            lambda row: 1 if (row['DIED'] + row['L_THREAT'] + row['DISABLE']) > 0 else 0,\n",
-    "            axis = 'columns')\n",
+    "        dataFrame['SEVERE'] = (dataFrame['DIED'] + dataFrame['L_THREAT'] + dataFrame['DISABLE']) > 0\n",
+    "        dataFrame['SEVERE'].replace({True: 1, False: 0}, inplace = True)\n",
     "        return dataFrame\n"
    ]
   },
@@ -1387,14 +1386,6 @@
     "\n",
     "sns.boxplot(x = \"CONCENTRATION\", y = \"Total Number of Adverse Reaction Reports\", data = batchCodeTable, order = order)"
    ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": null,
-   "id": "c355356a",
-   "metadata": {},
-   "outputs": [],
-   "source": []
   }
  ],
  "metadata": {