From 5805a8b24036d4265773f900be87f121170f3063 Mon Sep 17 00:00:00 2001 From: frankknoll Date: Sat, 29 Jan 2022 02:43:47 +0100 Subject: [PATCH] refactoring --- HowBadIsMyBatch.ipynb | 13 +++++++++++-- 1 file changed, 11 insertions(+), 2 deletions(-) diff --git a/HowBadIsMyBatch.ipynb b/HowBadIsMyBatch.ipynb index 9bacecce7e8..085c3f40c28 100644 --- a/HowBadIsMyBatch.ipynb +++ b/HowBadIsMyBatch.ipynb @@ -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",