refactoring
This commit is contained in:
@@ -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",
|
||||
|
||||
Reference in New Issue
Block a user