diff --git a/HowBadIsMyBatch.ipynb b/HowBadIsMyBatch.ipynb index 936d2325370..f4b55ffb574 100644 --- a/HowBadIsMyBatch.ipynb +++ b/HowBadIsMyBatch.ipynb @@ -21,21 +21,23 @@ "metadata": {}, "outputs": [], "source": [ - "def createDataFrameFromDescrs(vaersDescrs, manufacturer, dose):\n", + "def createDataFrameFromDescr(vaersDescr, manufacturer, dose):\n", " def filter(df):\n", " return df[\n", " (df[\"VAX_TYPE\"] == \"COVID19\") &\n", " (df[\"VAX_MANU\"] == manufacturer) &\n", " (df[\"VAX_DOSE_SERIES\"].str.contains(dose))]\n", + " \n", + " return pd.merge(\n", + " vaersDescr['VAERSDATA'],\n", + " filter(vaersDescr['VAERSVAX']),\n", + " left_index = True,\n", + " right_index = True)\n", "\n", - " def vaersDescr2DataFrame(vaersDescr):\n", - " return pd.merge(\n", - " vaersDescr['VAERSDATA'],\n", - " filter(vaersDescr['VAERSVAX']),\n", - " left_index = True,\n", - " right_index = True)\n", - "\n", - " return pd.concat(map(vaersDescr2DataFrame, vaersDescrs))\n" + "def createDataFrameFromDescrs(vaersDescrs, manufacturer, dose):\n", + " _createDataFrameFromDescr = lambda vaersDescr: createDataFrameFromDescr(vaersDescr, manufacturer, dose)\n", + " dataFrames = map(_createDataFrameFromDescr, vaersDescrs)\n", + " return pd.concat(dataFrames)\n" ] }, {