refactoring

This commit is contained in:
frankknoll
2022-01-29 03:00:22 +01:00
parent 6702d8456d
commit ae7cb41dac

View File

@@ -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"
]
},
{