ModeloPIB.ipynb

{
 "cells": [
  {
   "cell_type": "code",
   "execution_count": 1,
   "metadata": {},
   "outputs": [],
   "source": [
    "from pymongo import MongoClient\n",
    "import pandas as pd\n",
    "import numpy as np\n",
    "from seaborn import scatterplot\n",
    "from seaborn import regplot\n",
    "from sklearn.linear_model import LinearRegression\n",
    "from sklearn.linear_model import ElasticNet\n",
    "from sklearn.model_selection import train_test_split\n",
    "from sklearn.ensemble import RandomForestClassifier\n",
    "from sklearn.ensemble import RandomForestRegressor \n",
    "from sklearn.ensemble import GradientBoostingClassifier\n",
    "from sklearn.model_selection import KFold\n",
    "from sklearn.model_selection import TimeSeriesSplit\n",
    "from xgboost import XGBRegressor"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 3,
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/html": [
       "<div>\n",
       "<style scoped>\n",
       "    .dataframe tbody tr th:only-of-type {\n",
       "        vertical-align: middle;\n",
       "    }\n",
       "\n",
       "    .dataframe tbody tr th {\n",
       "        vertical-align: top;\n",
       "    }\n",
       "\n",
       "    .dataframe thead th {\n",
       "        text-align: right;\n",
       "    }\n",
       "</style>\n",
       "<table border=\"1\" class=\"dataframe\">\n",
       "  <thead>\n",
       "    <tr style=\"text-align: right;\">\n",
       "      <th></th>\n",
       "      <th>Periodo</th>\n",
       "      <th>num_casos</th>\n",
       "      <th>Total Paro</th>\n",
       "    </tr>\n",
       "  </thead>\n",
       "  <tbody>\n",
       "    <tr>\n",
       "      <th>0</th>\n",
       "      <td>201001</td>\n",
       "      <td>0</td>\n",
       "      <td>4617.700000</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>1</th>\n",
       "      <td>201002</td>\n",
       "      <td>0</td>\n",
       "      <td>4654.900000</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>2</th>\n",
       "      <td>201003</td>\n",
       "      <td>0</td>\n",
       "      <td>4585.700000</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>3</th>\n",
       "      <td>201004</td>\n",
       "      <td>0</td>\n",
       "      <td>4701.800000</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>4</th>\n",
       "      <td>201101</td>\n",
       "      <td>0</td>\n",
       "      <td>4921.400000</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>5</th>\n",
       "      <td>201102</td>\n",
       "      <td>0</td>\n",
       "      <td>4844.300000</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>6</th>\n",
       "      <td>201103</td>\n",
       "      <td>0</td>\n",
       "      <td>4998.500000</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>7</th>\n",
       "      <td>201104</td>\n",
       "      <td>0</td>\n",
       "      <td>5286.700000</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>8</th>\n",
       "      <td>201201</td>\n",
       "      <td>0</td>\n",
       "      <td>5668.400000</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>9</th>\n",
       "      <td>201202</td>\n",
       "      <td>0</td>\n",
       "      <td>5731.400000</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>10</th>\n",
       "      <td>201203</td>\n",
       "      <td>0</td>\n",
       "      <td>5823.900000</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>11</th>\n",
       "      <td>201204</td>\n",
       "      <td>0</td>\n",
       "      <td>6021.600000</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>12</th>\n",
       "      <td>201301</td>\n",
       "      <td>0</td>\n",
       "      <td>6278.000000</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>13</th>\n",
       "      <td>201302</td>\n",
       "      <td>0</td>\n",
       "      <td>6047.600000</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>14</th>\n",
       "      <td>201303</td>\n",
       "      <td>0</td>\n",
       "      <td>5942.400000</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>15</th>\n",
       "      <td>201304</td>\n",
       "      <td>0</td>\n",
       "      <td>5936.000000</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>16</th>\n",
       "      <td>201401</td>\n",
       "      <td>0</td>\n",
       "      <td>5932.500000</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>17</th>\n",
       "      <td>201402</td>\n",
       "      <td>0</td>\n",
       "      <td>5623.000000</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>18</th>\n",
       "      <td>201403</td>\n",
       "      <td>0</td>\n",
       "      <td>5427.300000</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>19</th>\n",
       "      <td>201404</td>\n",
       "      <td>0</td>\n",
       "      <td>5457.500000</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>20</th>\n",
       "      <td>201501</td>\n",
       "      <td>0</td>\n",
       "      <td>5444.900000</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>21</th>\n",
       "      <td>201502</td>\n",
       "      <td>0</td>\n",
       "      <td>5149.200000</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>22</th>\n",
       "      <td>201503</td>\n",
       "      <td>0</td>\n",
       "      <td>4850.300000</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>23</th>\n",
       "      <td>201504</td>\n",
       "      <td>0</td>\n",
       "      <td>4779.200000</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>24</th>\n",
       "      <td>201601</td>\n",
       "      <td>0</td>\n",
       "      <td>4791.600000</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>25</th>\n",
       "      <td>201602</td>\n",
       "      <td>0</td>\n",
       "      <td>4574.800000</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>26</th>\n",
       "      <td>201603</td>\n",
       "      <td>0</td>\n",
       "      <td>4321.300000</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>27</th>\n",
       "      <td>201604</td>\n",
       "      <td>0</td>\n",
       "      <td>4237.900000</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>28</th>\n",
       "      <td>201701</td>\n",
       "      <td>0</td>\n",
       "      <td>4255.800000</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>29</th>\n",
       "      <td>201702</td>\n",
       "      <td>0</td>\n",
       "      <td>3913.700000</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>30</th>\n",
       "      <td>201703</td>\n",
       "      <td>0</td>\n",
       "      <td>3731.600000</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>31</th>\n",
       "      <td>201704</td>\n",
       "      <td>0</td>\n",
       "      <td>3766.300000</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>32</th>\n",
       "      <td>201801</td>\n",
       "      <td>0</td>\n",
       "      <td>3795.900000</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>33</th>\n",
       "      <td>201802</td>\n",
       "      <td>0</td>\n",
       "      <td>3490.600000</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>34</th>\n",
       "      <td>201803</td>\n",
       "      <td>0</td>\n",
       "      <td>3326.300000</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>35</th>\n",
       "      <td>201804</td>\n",
       "      <td>0</td>\n",
       "      <td>3304.800000</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>36</th>\n",
       "      <td>201901</td>\n",
       "      <td>0</td>\n",
       "      <td>3354.400000</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>37</th>\n",
       "      <td>201902</td>\n",
       "      <td>0</td>\n",
       "      <td>3230.200000</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>38</th>\n",
       "      <td>201903</td>\n",
       "      <td>0</td>\n",
       "      <td>3214.000000</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>39</th>\n",
       "      <td>201904</td>\n",
       "      <td>0</td>\n",
       "      <td>3192.000000</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>40</th>\n",
       "      <td>202001</td>\n",
       "      <td>934961</td>\n",
       "      <td>3313.900000</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>41</th>\n",
       "      <td>202002</td>\n",
       "      <td>1103579</td>\n",
       "      <td>3368.400000</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>42</th>\n",
       "      <td>202003</td>\n",
       "      <td>6850799</td>\n",
       "      <td>3723.600000</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>43</th>\n",
       "      <td>202004</td>\n",
       "      <td>14041451</td>\n",
       "      <td>3719.400000</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>44</th>\n",
       "      <td>202101</td>\n",
       "      <td>15790846</td>\n",
       "      <td>3653.700000</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>45</th>\n",
       "      <td>202102</td>\n",
       "      <td>6411176</td>\n",
       "      <td>3543.500000</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>46</th>\n",
       "      <td>202103</td>\n",
       "      <td>13285322</td>\n",
       "      <td>3416.400000</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>47</th>\n",
       "      <td>202104</td>\n",
       "      <td>21827643</td>\n",
       "      <td>3104.100000</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>48</th>\n",
       "      <td>202201</td>\n",
       "      <td>20660184</td>\n",
       "      <td>3743.771448</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>49</th>\n",
       "      <td>202202</td>\n",
       "      <td>142812</td>\n",
       "      <td>3739.411991</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>50</th>\n",
       "      <td>202203</td>\n",
       "      <td>107760</td>\n",
       "      <td>3739.411991</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>51</th>\n",
       "      <td>202204</td>\n",
       "      <td>141024</td>\n",
       "      <td>3739.504638</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>52</th>\n",
       "      <td>202301</td>\n",
       "      <td>32664</td>\n",
       "      <td>3739.425529</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>53</th>\n",
       "      <td>202302</td>\n",
       "      <td>37416</td>\n",
       "      <td>3739.425529</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>54</th>\n",
       "      <td>202303</td>\n",
       "      <td>201372</td>\n",
       "      <td>3739.425529</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>55</th>\n",
       "      <td>202304</td>\n",
       "      <td>194700</td>\n",
       "      <td>3739.825110</td>\n",
       "    </tr>\n",
       "  </tbody>\n",
       "</table>\n",
       "</div>"
      ],
      "text/plain": [
       "    Periodo  num_casos   Total Paro\n",
       "0    201001          0  4617.700000\n",
       "1    201002          0  4654.900000\n",
       "2    201003          0  4585.700000\n",
       "3    201004          0  4701.800000\n",
       "4    201101          0  4921.400000\n",
       "5    201102          0  4844.300000\n",
       "6    201103          0  4998.500000\n",
       "7    201104          0  5286.700000\n",
       "8    201201          0  5668.400000\n",
       "9    201202          0  5731.400000\n",
       "10   201203          0  5823.900000\n",
       "11   201204          0  6021.600000\n",
       "12   201301          0  6278.000000\n",
       "13   201302          0  6047.600000\n",
       "14   201303          0  5942.400000\n",
       "15   201304          0  5936.000000\n",
       "16   201401          0  5932.500000\n",
       "17   201402          0  5623.000000\n",
       "18   201403          0  5427.300000\n",
       "19   201404          0  5457.500000\n",
       "20   201501          0  5444.900000\n",
       "21   201502          0  5149.200000\n",
       "22   201503          0  4850.300000\n",
       "23   201504          0  4779.200000\n",
       "24   201601          0  4791.600000\n",
       "25   201602          0  4574.800000\n",
       "26   201603          0  4321.300000\n",
       "27   201604          0  4237.900000\n",
       "28   201701          0  4255.800000\n",
       "29   201702          0  3913.700000\n",
       "30   201703          0  3731.600000\n",
       "31   201704          0  3766.300000\n",
       "32   201801          0  3795.900000\n",
       "33   201802          0  3490.600000\n",
       "34   201803          0  3326.300000\n",
       "35   201804          0  3304.800000\n",
       "36   201901          0  3354.400000\n",
       "37   201902          0  3230.200000\n",
       "38   201903          0  3214.000000\n",
       "39   201904          0  3192.000000\n",
       "40   202001     934961  3313.900000\n",
       "41   202002    1103579  3368.400000\n",
       "42   202003    6850799  3723.600000\n",
       "43   202004   14041451  3719.400000\n",
       "44   202101   15790846  3653.700000\n",
       "45   202102    6411176  3543.500000\n",
       "46   202103   13285322  3416.400000\n",
       "47   202104   21827643  3104.100000\n",
       "48   202201   20660184  3743.771448\n",
       "49   202202     142812  3739.411991\n",
       "50   202203     107760  3739.411991\n",
       "51   202204     141024  3739.504638\n",
       "52   202301      32664  3739.425529\n",
       "53   202302      37416  3739.425529\n",
       "54   202303     201372  3739.425529\n",
       "55   202304     194700  3739.825110"
      ]
     },
     "execution_count": 3,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "dataParaPIB = pd.read_csv('csvTableau/ParoFinalPorCCAAIntegrado.csv')\n",
    "dataParaPIB = dataParaPIB[['Periodo','num_casos','Total Paro']].groupby(by=\"Periodo\", as_index=False).sum()\n",
    "dataParaPIB"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 4,
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/html": [
       "<div>\n",
       "<style scoped>\n",
       "    .dataframe tbody tr th:only-of-type {\n",
       "        vertical-align: middle;\n",
       "    }\n",
       "\n",
       "    .dataframe tbody tr th {\n",
       "        vertical-align: top;\n",
       "    }\n",
       "\n",
       "    .dataframe thead th {\n",
       "        text-align: right;\n",
       "    }\n",
       "</style>\n",
       "<table border=\"1\" class=\"dataframe\">\n",
       "  <thead>\n",
       "    <tr style=\"text-align: right;\">\n",
       "      <th></th>\n",
       "      <th>Periodo</th>\n",
       "      <th>PIB Trimestral</th>\n",
       "      <th>num_casos</th>\n",
       "      <th>Total Paro</th>\n",
       "    </tr>\n",
       "  </thead>\n",
       "  <tbody>\n",
       "    <tr>\n",
       "      <th>22</th>\n",
       "      <td>201001</td>\n",
       "      <td>268.031</td>\n",
       "      <td>0</td>\n",
       "      <td>4617.7</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>46</th>\n",
       "      <td>201002</td>\n",
       "      <td>267.355</td>\n",
       "      <td>0</td>\n",
       "      <td>4654.9</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>35</th>\n",
       "      <td>201003</td>\n",
       "      <td>267.959</td>\n",
       "      <td>0</td>\n",
       "      <td>4585.7</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>45</th>\n",
       "      <td>201004</td>\n",
       "      <td>269.364</td>\n",
       "      <td>0</td>\n",
       "      <td>4701.8</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>44</th>\n",
       "      <td>201101</td>\n",
       "      <td>267.991</td>\n",
       "      <td>0</td>\n",
       "      <td>4921.4</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>21</th>\n",
       "      <td>201102</td>\n",
       "      <td>267.784</td>\n",
       "      <td>0</td>\n",
       "      <td>4844.3</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>20</th>\n",
       "      <td>201103</td>\n",
       "      <td>265.547</td>\n",
       "      <td>0</td>\n",
       "      <td>4998.5</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>43</th>\n",
       "      <td>201104</td>\n",
       "      <td>262.441</td>\n",
       "      <td>0</td>\n",
       "      <td>5286.7</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>34</th>\n",
       "      <td>201201</td>\n",
       "      <td>261.538</td>\n",
       "      <td>0</td>\n",
       "      <td>5668.4</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>42</th>\n",
       "      <td>201202</td>\n",
       "      <td>258.877</td>\n",
       "      <td>0</td>\n",
       "      <td>5731.4</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>8</th>\n",
       "      <td>201203</td>\n",
       "      <td>257.492</td>\n",
       "      <td>0</td>\n",
       "      <td>5823.9</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>19</th>\n",
       "      <td>201204</td>\n",
       "      <td>253.192</td>\n",
       "      <td>0</td>\n",
       "      <td>6021.6</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>41</th>\n",
       "      <td>201301</td>\n",
       "      <td>255.914</td>\n",
       "      <td>0</td>\n",
       "      <td>6278.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>18</th>\n",
       "      <td>201302</td>\n",
       "      <td>255.183</td>\n",
       "      <td>0</td>\n",
       "      <td>6047.6</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>17</th>\n",
       "      <td>201303</td>\n",
       "      <td>254.815</td>\n",
       "      <td>0</td>\n",
       "      <td>5942.4</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>40</th>\n",
       "      <td>201304</td>\n",
       "      <td>254.436</td>\n",
       "      <td>0</td>\n",
       "      <td>5936.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>39</th>\n",
       "      <td>201401</td>\n",
       "      <td>256.308</td>\n",
       "      <td>0</td>\n",
       "      <td>5932.5</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>38</th>\n",
       "      <td>201402</td>\n",
       "      <td>256.980</td>\n",
       "      <td>0</td>\n",
       "      <td>5623.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>7</th>\n",
       "      <td>201403</td>\n",
       "      <td>258.252</td>\n",
       "      <td>0</td>\n",
       "      <td>5427.3</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>16</th>\n",
       "      <td>201404</td>\n",
       "      <td>260.618</td>\n",
       "      <td>0</td>\n",
       "      <td>5457.5</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>37</th>\n",
       "      <td>201501</td>\n",
       "      <td>265.848</td>\n",
       "      <td>0</td>\n",
       "      <td>5444.9</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>33</th>\n",
       "      <td>201502</td>\n",
       "      <td>268.181</td>\n",
       "      <td>0</td>\n",
       "      <td>5149.2</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>32</th>\n",
       "      <td>201503</td>\n",
       "      <td>270.633</td>\n",
       "      <td>0</td>\n",
       "      <td>4850.3</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>15</th>\n",
       "      <td>201504</td>\n",
       "      <td>272.928</td>\n",
       "      <td>0</td>\n",
       "      <td>4779.2</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>6</th>\n",
       "      <td>201601</td>\n",
       "      <td>275.039</td>\n",
       "      <td>0</td>\n",
       "      <td>4791.6</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>14</th>\n",
       "      <td>201602</td>\n",
       "      <td>276.760</td>\n",
       "      <td>0</td>\n",
       "      <td>4574.8</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>13</th>\n",
       "      <td>201603</td>\n",
       "      <td>280.145</td>\n",
       "      <td>0</td>\n",
       "      <td>4321.3</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>5</th>\n",
       "      <td>201604</td>\n",
       "      <td>281.896</td>\n",
       "      <td>0</td>\n",
       "      <td>4237.9</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>12</th>\n",
       "      <td>201701</td>\n",
       "      <td>285.723</td>\n",
       "      <td>0</td>\n",
       "      <td>4255.8</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>11</th>\n",
       "      <td>201702</td>\n",
       "      <td>289.461</td>\n",
       "      <td>0</td>\n",
       "      <td>3913.7</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>31</th>\n",
       "      <td>201703</td>\n",
       "      <td>291.234</td>\n",
       "      <td>0</td>\n",
       "      <td>3731.6</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>36</th>\n",
       "      <td>201704</td>\n",
       "      <td>295.449</td>\n",
       "      <td>0</td>\n",
       "      <td>3766.3</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>30</th>\n",
       "      <td>201801</td>\n",
       "      <td>296.217</td>\n",
       "      <td>0</td>\n",
       "      <td>3795.9</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>4</th>\n",
       "      <td>201802</td>\n",
       "      <td>299.384</td>\n",
       "      <td>0</td>\n",
       "      <td>3490.6</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>24</th>\n",
       "      <td>201803</td>\n",
       "      <td>301.953</td>\n",
       "      <td>0</td>\n",
       "      <td>3326.3</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>29</th>\n",
       "      <td>201804</td>\n",
       "      <td>305.705</td>\n",
       "      <td>0</td>\n",
       "      <td>3304.8</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>28</th>\n",
       "      <td>201901</td>\n",
       "      <td>307.704</td>\n",
       "      <td>0</td>\n",
       "      <td>3354.4</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>27</th>\n",
       "      <td>201902</td>\n",
       "      <td>309.942</td>\n",
       "      <td>0</td>\n",
       "      <td>3230.2</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>3</th>\n",
       "      <td>201903</td>\n",
       "      <td>311.706</td>\n",
       "      <td>0</td>\n",
       "      <td>3214.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>2</th>\n",
       "      <td>201904</td>\n",
       "      <td>315.023</td>\n",
       "      <td>0</td>\n",
       "      <td>3192.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>26</th>\n",
       "      <td>202001</td>\n",
       "      <td>296.636</td>\n",
       "      <td>934961</td>\n",
       "      <td>3313.9</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>10</th>\n",
       "      <td>202002</td>\n",
       "      <td>245.851</td>\n",
       "      <td>1103579</td>\n",
       "      <td>3368.4</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>25</th>\n",
       "      <td>202003</td>\n",
       "      <td>289.165</td>\n",
       "      <td>6850799</td>\n",
       "      <td>3723.6</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>9</th>\n",
       "      <td>202004</td>\n",
       "      <td>290.296</td>\n",
       "      <td>14041451</td>\n",
       "      <td>3719.4</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>1</th>\n",
       "      <td>202101</td>\n",
       "      <td>288.038</td>\n",
       "      <td>15790846</td>\n",
       "      <td>3653.7</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>23</th>\n",
       "      <td>202102</td>\n",
       "      <td>292.209</td>\n",
       "      <td>6411176</td>\n",
       "      <td>3543.5</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>0</th>\n",
       "      <td>202103</td>\n",
       "      <td>303.717</td>\n",
       "      <td>13285322</td>\n",
       "      <td>3416.4</td>\n",
       "    </tr>\n",
       "  </tbody>\n",
       "</table>\n",
       "</div>"
      ],
      "text/plain": [
       "    Periodo  PIB Trimestral  num_casos  Total Paro\n",
       "22   201001         268.031          0      4617.7\n",
       "46   201002         267.355          0      4654.9\n",
       "35   201003         267.959          0      4585.7\n",
       "45   201004         269.364          0      4701.8\n",
       "44   201101         267.991          0      4921.4\n",
       "21   201102         267.784          0      4844.3\n",
       "20   201103         265.547          0      4998.5\n",
       "43   201104         262.441          0      5286.7\n",
       "34   201201         261.538          0      5668.4\n",
       "42   201202         258.877          0      5731.4\n",
       "8    201203         257.492          0      5823.9\n",
       "19   201204         253.192          0      6021.6\n",
       "41   201301         255.914          0      6278.0\n",
       "18   201302         255.183          0      6047.6\n",
       "17   201303         254.815          0      5942.4\n",
       "40   201304         254.436          0      5936.0\n",
       "39   201401         256.308          0      5932.5\n",
       "38   201402         256.980          0      5623.0\n",
       "7    201403         258.252          0      5427.3\n",
       "16   201404         260.618          0      5457.5\n",
       "37   201501         265.848          0      5444.9\n",
       "33   201502         268.181          0      5149.2\n",
       "32   201503         270.633          0      4850.3\n",
       "15   201504         272.928          0      4779.2\n",
       "6    201601         275.039          0      4791.6\n",
       "14   201602         276.760          0      4574.8\n",
       "13   201603         280.145          0      4321.3\n",
       "5    201604         281.896          0      4237.9\n",
       "12   201701         285.723          0      4255.8\n",
       "11   201702         289.461          0      3913.7\n",
       "31   201703         291.234          0      3731.6\n",
       "36   201704         295.449          0      3766.3\n",
       "30   201801         296.217          0      3795.9\n",
       "4    201802         299.384          0      3490.6\n",
       "24   201803         301.953          0      3326.3\n",
       "29   201804         305.705          0      3304.8\n",
       "28   201901         307.704          0      3354.4\n",
       "27   201902         309.942          0      3230.2\n",
       "3    201903         311.706          0      3214.0\n",
       "2    201904         315.023          0      3192.0\n",
       "26   202001         296.636     934961      3313.9\n",
       "10   202002         245.851    1103579      3368.4\n",
       "25   202003         289.165    6850799      3723.6\n",
       "9    202004         290.296   14041451      3719.4\n",
       "1    202101         288.038   15790846      3653.7\n",
       "23   202102         292.209    6411176      3543.5\n",
       "0    202103         303.717   13285322      3416.4"
      ]
     },
     "execution_count": 4,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "#Conexion a MongoDb\n",
    "client = MongoClient('mongodb+srv://tfmeaecovid:eaemongo@cluster0.qwy07.mongodb.net/test')\n",
    "#Seleccion de la Base de Datos\n",
    "db = client.TFMCOVID\n",
    "#Conversion de Collection a Dataframe de datos de los dataset\n",
    "dataPIB = pd.DataFrame.from_records(db.PIB.find())\n",
    "dataPIB = dataPIB.rename(columns={dataPIB.columns[1]: 'Periodo'})\n",
    "#Cambio de llave a utilizar 0 para futuras referencias\n",
    "dataPIB['Periodo'] = dataPIB['Periodo'].str.replace('T','0')\n",
    "#dataCasos = dataCasos.drop('Columna1',axis=1)\n",
    "dataPIB['PIB Trimestral'] = dataPIB['PIB Trimestral'].str.replace(',','.').astype(float)\n",
    "dataPIB['Periodo'] = pd.to_numeric(dataPIB['Periodo'], downcast='integer')\n",
    "dataPIB = dataPIB[['Periodo','PIB Trimestral']]\n",
    "dataPIBConCasosYParo = dataPIB.merge(dataParaPIB, on=\"Periodo\", how=\"left\").sort_values(by='Periodo')\n",
    "dataPIBConCasosYParo"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 6,
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/html": [
       "<div>\n",
       "<style scoped>\n",
       "    .dataframe tbody tr th:only-of-type {\n",
       "        vertical-align: middle;\n",
       "    }\n",
       "\n",
       "    .dataframe tbody tr th {\n",
       "        vertical-align: top;\n",
       "    }\n",
       "\n",
       "    .dataframe thead th {\n",
       "        text-align: right;\n",
       "    }\n",
       "</style>\n",
       "<table border=\"1\" class=\"dataframe\">\n",
       "  <thead>\n",
       "    <tr style=\"text-align: right;\">\n",
       "      <th></th>\n",
       "      <th>Periodo</th>\n",
       "      <th>num_casos</th>\n",
       "      <th>Total Paro</th>\n",
       "      <th>PIB Trimestral</th>\n",
       "    </tr>\n",
       "  </thead>\n",
       "  <tbody>\n",
       "    <tr>\n",
       "      <th>0</th>\n",
       "      <td>202104</td>\n",
       "      <td>71203626</td>\n",
       "      <td>3104.200000</td>\n",
       "      <td>0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>1</th>\n",
       "      <td>202201</td>\n",
       "      <td>69084800</td>\n",
       "      <td>3246.194386</td>\n",
       "      <td>0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>2</th>\n",
       "      <td>202202</td>\n",
       "      <td>421960</td>\n",
       "      <td>3373.258353</td>\n",
       "      <td>0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>3</th>\n",
       "      <td>202203</td>\n",
       "      <td>73600</td>\n",
       "      <td>3373.258353</td>\n",
       "      <td>0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>4</th>\n",
       "      <td>202204</td>\n",
       "      <td>149680</td>\n",
       "      <td>3373.258353</td>\n",
       "      <td>0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>5</th>\n",
       "      <td>202301</td>\n",
       "      <td>1350520</td>\n",
       "      <td>3373.258353</td>\n",
       "      <td>0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>6</th>\n",
       "      <td>202302</td>\n",
       "      <td>422000</td>\n",
       "      <td>3373.258353</td>\n",
       "      <td>0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>7</th>\n",
       "      <td>202303</td>\n",
       "      <td>74200</td>\n",
       "      <td>3373.258353</td>\n",
       "      <td>0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>8</th>\n",
       "      <td>202304</td>\n",
       "      <td>149680</td>\n",
       "      <td>3373.258353</td>\n",
       "      <td>0</td>\n",
       "    </tr>\n",
       "  </tbody>\n",
       "</table>\n",
       "</div>"
      ],
      "text/plain": [
       "   Periodo  num_casos   Total Paro  PIB Trimestral\n",
       "0   202104   71203626  3104.200000               0\n",
       "1   202201   69084800  3246.194386               0\n",
       "2   202202     421960  3373.258353               0\n",
       "3   202203      73600  3373.258353               0\n",
       "4   202204     149680  3373.258353               0\n",
       "5   202301    1350520  3373.258353               0\n",
       "6   202302     422000  3373.258353               0\n",
       "7   202303      74200  3373.258353               0\n",
       "8   202304     149680  3373.258353               0"
      ]
     },
     "execution_count": 6,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "PIBpredic = pd.read_csv('DatosPIBPrediccion.csv',sep=';', encoding=\"latin-1\")\n",
    "PIBpredic = PIBpredic.rename(columns={\"Total\":\"Total Paro\"})\n",
    "PIBpredic"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 8,
   "metadata": {},
   "outputs": [],
   "source": [
    "X = dataPIBConCasosYParo[['Periodo','num_casos','Total Paro']]\n",
    "y = dataPIBConCasosYParo[['PIB Trimestral']]\n",
    "\n",
    "X_predict = PIBpredic[['Periodo','num_casos','Total Paro']]\n",
    "y_predict = PIBpredic[['PIB Trimestral']]"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 9,
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/html": [
       "<div>\n",
       "<style scoped>\n",
       "    .dataframe tbody tr th:only-of-type {\n",
       "        vertical-align: middle;\n",
       "    }\n",
       "\n",
       "    .dataframe tbody tr th {\n",
       "        vertical-align: top;\n",
       "    }\n",
       "\n",
       "    .dataframe thead th {\n",
       "        text-align: right;\n",
       "    }\n",
       "</style>\n",
       "<table border=\"1\" class=\"dataframe\">\n",
       "  <thead>\n",
       "    <tr style=\"text-align: right;\">\n",
       "      <th></th>\n",
       "      <th>Periodo</th>\n",
       "      <th>num_casos</th>\n",
       "      <th>Total Paro</th>\n",
       "      <th>PIB Trimestral</th>\n",
       "    </tr>\n",
       "  </thead>\n",
       "  <tbody>\n",
       "    <tr>\n",
       "      <th>0</th>\n",
       "      <td>202104</td>\n",
       "      <td>71203626</td>\n",
       "      <td>3104.200000</td>\n",
       "      <td>308.725723</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>1</th>\n",
       "      <td>202201</td>\n",
       "      <td>69084800</td>\n",
       "      <td>3246.194386</td>\n",
       "      <td>307.766528</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>2</th>\n",
       "      <td>202202</td>\n",
       "      <td>421960</td>\n",
       "      <td>3373.258353</td>\n",
       "      <td>270.051062</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>3</th>\n",
       "      <td>202203</td>\n",
       "      <td>73600</td>\n",
       "      <td>3373.258353</td>\n",
       "      <td>270.051062</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>4</th>\n",
       "      <td>202204</td>\n",
       "      <td>149680</td>\n",
       "      <td>3373.258353</td>\n",
       "      <td>270.051062</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>5</th>\n",
       "      <td>202301</td>\n",
       "      <td>1350520</td>\n",
       "      <td>3373.258353</td>\n",
       "      <td>267.539817</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>6</th>\n",
       "      <td>202302</td>\n",
       "      <td>422000</td>\n",
       "      <td>3373.258353</td>\n",
       "      <td>270.051062</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>7</th>\n",
       "      <td>202303</td>\n",
       "      <td>74200</td>\n",
       "      <td>3373.258353</td>\n",
       "      <td>270.051062</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>8</th>\n",
       "      <td>202304</td>\n",
       "      <td>149680</td>\n",
       "      <td>3373.258353</td>\n",
       "      <td>270.051062</td>\n",
       "    </tr>\n",
       "  </tbody>\n",
       "</table>\n",
       "</div>"
      ],
      "text/plain": [
       "   Periodo  num_casos   Total Paro  PIB Trimestral\n",
       "0   202104   71203626  3104.200000      308.725723\n",
       "1   202201   69084800  3246.194386      307.766528\n",
       "2   202202     421960  3373.258353      270.051062\n",
       "3   202203      73600  3373.258353      270.051062\n",
       "4   202204     149680  3373.258353      270.051062\n",
       "5   202301    1350520  3373.258353      267.539817\n",
       "6   202302     422000  3373.258353      270.051062\n",
       "7   202303      74200  3373.258353      270.051062\n",
       "8   202304     149680  3373.258353      270.051062"
      ]
     },
     "execution_count": 9,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "ModeloPIB = RandomForestRegressor(n_estimators = 1000, max_depth=9, random_state=0)\n",
    "ModeloPIB.fit(X,y.values.ravel())\n",
    "predictions = ModeloPIB.predict(X_predict)\n",
    "PIBpredic['PIB Trimestral'] = predictions\n",
    "PIBpredic"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 11,
   "metadata": {},
   "outputs": [],
   "source": [
    "PIBFinalFinal = pd.concat([dataPIBConCasosYParo,PIBpredic])\n",
    "PIBFinalFinal.to_csv('ModeloFinalPIBIntegrado.csv')"
   ]
  }
 ],
 "metadata": {
  "interpreter": {
   "hash": "fb698bdeab99cfbf9e31cab566d99a06e057f2dec3dcc8097ebabb329e860e16"
  },
  "kernelspec": {
   "display_name": "Python 3.8.8 ('base')",
   "language": "python",
   "name": "python3"
  },
  "language_info": {
   "codemirror_mode": {
    "name": "ipython",
    "version": 3
   },
   "file_extension": ".py",
   "mimetype": "text/x-python",
   "name": "python",
   "nbconvert_exporter": "python",
   "pygments_lexer": "ipython3",
   "version": "3.8.8"
  },
  "orig_nbformat": 4
 },
 "nbformat": 4,
 "nbformat_minor": 2
}