refactoring
This commit is contained in:
@@ -23,7 +23,10 @@
|
|||||||
"source": [
|
"source": [
|
||||||
"def _createDataFrame(vaersDescrs, manufacturer, dose):\n",
|
"def _createDataFrame(vaersDescrs, manufacturer, dose):\n",
|
||||||
" def filter(df):\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",
|
"\n",
|
||||||
" def vaersDescr2DataFrame(vaersDescr):\n",
|
" def vaersDescr2DataFrame(vaersDescr):\n",
|
||||||
" return pd.merge(\n",
|
" return pd.merge(\n",
|
||||||
@@ -45,7 +48,13 @@
|
|||||||
"def createDataFrame(dataDir, manufacturer, dose):\n",
|
"def createDataFrame(dataDir, manufacturer, dose):\n",
|
||||||
" def readVaersDescr(year):\n",
|
" def readVaersDescr(year):\n",
|
||||||
" def read_csv(file, usecols, dtype = {}):\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",
|
"\n",
|
||||||
" folder = dataDir + \"/\" + year + \"VAERSData/\"\n",
|
" folder = dataDir + \"/\" + year + \"VAERSData/\"\n",
|
||||||
" return {\n",
|
" return {\n",
|
||||||
|
|||||||
Reference in New Issue
Block a user