From 91dba5f4997ed3b918f7aab5523ee5a7ca09729b Mon Sep 17 00:00:00 2001
From: Manuel Santos <33625071+manel1874@users.noreply.github.com>
Date: Fri, 15 Jul 2022 16:49:50 +0100
Subject: [PATCH] Small fix in L3 Data Preparation

Country 0's data corresponds to the first column. Therefore, we should select raw_data[: , 0] instead of raw_data[0 , :]. Otherwise, we are selecting the first month (line) of every country.
---
 L3_DataPreparation.ipynb | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/L3_DataPreparation.ipynb b/L3_DataPreparation.ipynb
index bad438c..de684f8 100644
--- a/L3_DataPreparation.ipynb
+++ b/L3_DataPreparation.ipynb
@@ -1247,7 +1247,7 @@
     "import syft as sy\n",
     "\n",
     "# select all of Country 0's data\n",
-    "country0_data = raw_data[0, :]\n",
+    "country0_data = raw_data[:, 0]\n",
     "\n",
     "# Specify it to be a np.int32 dtype\n",
     "country0_data = country0_data.astype(np.int32)\n",