refactoring

This commit is contained in:
frankknoll
2022-01-29 02:43:47 +01:00
parent c558c90513
commit 5805a8b240

View File

@@ -23,7 +23,10 @@
"source": [
"def _createDataFrame(vaersDescrs, manufacturer, dose):\n",
" def filter(df):\n",
" return df[(df[\"VAX_TYPE\"] == \"COVID19\") & (df[\"VAX_MANU\"] == manufacturer) & (df[\"VAX_DOSE_SERIES\"].str.contains(dose))]\n",
" return df[\n",
" (df[\"VAX_TYPE\"] == \"COVID19\") &\n",
" (df[\"VAX_MANU\"] == manufacturer) &\n",
" (df[\"VAX_DOSE_SERIES\"].str.contains(dose))]\n",
"\n",
" def vaersDescr2DataFrame(vaersDescr):\n",
" return pd.merge(\n",
@@ -45,7 +48,13 @@
"def createDataFrame(dataDir, manufacturer, dose):\n",
" def readVaersDescr(year):\n",
" def read_csv(file, usecols, dtype = {}):\n",
" return pd.read_csv(file, index_col = 'VAERS_ID', encoding = 'latin1', low_memory = False, usecols = usecols, dtype = dtype)\n",
" return pd.read_csv(\n",
" file,\n",
" index_col = 'VAERS_ID',\n",
" encoding = 'latin1',\n",
" low_memory = False,\n",
" usecols = usecols,\n",
" dtype = dtype)\n",
"\n",
" folder = dataDir + \"/\" + year + \"VAERSData/\"\n",
" return {\n",