simulation.ipynb

{
 "cells": [
  {
   "cell_type": "code",
   "execution_count": 1,
   "id": "caf4295c-7afa-480d-a104-7f733d681129",
   "metadata": {},
   "outputs": [],
   "source": [
    "import os\n",
    "import numpy as np\n",
    "import matplotlib.pyplot as plt\n",
    "import pandas as pd\n",
    "import seaborn as sns\n",
    "from matplotlib import rc\n",
    "from numpy.linalg import inv \n",
    "from sklearn import linear_model "
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 1,
   "id": "59ada287",
   "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>0</th>\n",
       "      <th>1</th>\n",
       "      <th>2</th>\n",
       "      <th>3</th>\n",
       "      <th>4</th>\n",
       "      <th>5</th>\n",
       "      <th>6</th>\n",
       "      <th>7</th>\n",
       "      <th>8</th>\n",
       "      <th>9</th>\n",
       "      <th>10</th>\n",
       "      <th>11</th>\n",
       "      <th>12</th>\n",
       "      <th>13</th>\n",
       "      <th>14</th>\n",
       "      <th>15</th>\n",
       "      <th>16</th>\n",
       "      <th>17</th>\n",
       "      <th>18</th>\n",
       "      <th>19</th>\n",
       "      <th>20</th>\n",
       "      <th>21</th>\n",
       "      <th>22</th>\n",
       "      <th>23</th>\n",
       "      <th>24</th>\n",
       "      <th>25</th>\n",
       "      <th>26</th>\n",
       "      <th>27</th>\n",
       "      <th>28</th>\n",
       "      <th>29</th>\n",
       "      <th>30</th>\n",
       "      <th>31</th>\n",
       "      <th>32</th>\n",
       "      <th>33</th>\n",
       "      <th>34</th>\n",
       "      <th>35</th>\n",
       "      <th>36</th>\n",
       "      <th>37</th>\n",
       "      <th>38</th>\n",
       "      <th>39</th>\n",
       "      <th>40</th>\n",
       "      <th>41</th>\n",
       "      <th>42</th>\n",
       "      <th>43</th>\n",
       "      <th>44</th>\n",
       "      <th>45</th>\n",
       "      <th>46</th>\n",
       "      <th>47</th>\n",
       "      <th>48</th>\n",
       "      <th>49</th>\n",
       "      <th>...</th>\n",
       "      <th>620</th>\n",
       "      <th>621</th>\n",
       "      <th>622</th>\n",
       "      <th>623</th>\n",
       "      <th>624</th>\n",
       "      <th>625</th>\n",
       "      <th>626</th>\n",
       "      <th>627</th>\n",
       "      <th>628</th>\n",
       "      <th>629</th>\n",
       "      <th>630</th>\n",
       "      <th>631</th>\n",
       "      <th>632</th>\n",
       "      <th>633</th>\n",
       "      <th>634</th>\n",
       "      <th>635</th>\n",
       "      <th>636</th>\n",
       "      <th>637</th>\n",
       "      <th>638</th>\n",
       "      <th>639</th>\n",
       "      <th>640</th>\n",
       "      <th>641</th>\n",
       "      <th>642</th>\n",
       "      <th>643</th>\n",
       "      <th>644</th>\n",
       "      <th>645</th>\n",
       "      <th>646</th>\n",
       "      <th>647</th>\n",
       "      <th>648</th>\n",
       "      <th>649</th>\n",
       "      <th>650</th>\n",
       "      <th>651</th>\n",
       "      <th>652</th>\n",
       "      <th>653</th>\n",
       "      <th>654</th>\n",
       "      <th>655</th>\n",
       "      <th>656</th>\n",
       "      <th>657</th>\n",
       "      <th>658</th>\n",
       "      <th>659</th>\n",
       "      <th>660</th>\n",
       "      <th>661</th>\n",
       "      <th>662</th>\n",
       "      <th>663</th>\n",
       "      <th>664</th>\n",
       "      <th>665</th>\n",
       "      <th>666</th>\n",
       "      <th>667</th>\n",
       "      <th>668</th>\n",
       "      <th>669</th>\n",
       "    </tr>\n",
       "  </thead>\n",
       "  <tbody>\n",
       "    <tr>\n",
       "      <th>0</th>\n",
       "      <td>436363.0</td>\n",
       "      <td>437064.0</td>\n",
       "      <td>436469.0</td>\n",
       "      <td>437013.0</td>\n",
       "      <td>436983.0</td>\n",
       "      <td>436644.0</td>\n",
       "      <td>436768.0</td>\n",
       "      <td>436802.0</td>\n",
       "      <td>436286.0</td>\n",
       "      <td>437163.0</td>\n",
       "      <td>436833.0</td>\n",
       "      <td>436958.0</td>\n",
       "      <td>436212.0</td>\n",
       "      <td>436937.0</td>\n",
       "      <td>436565.0</td>\n",
       "      <td>436939.0</td>\n",
       "      <td>436773.0</td>\n",
       "      <td>435964.0</td>\n",
       "      <td>436779.0</td>\n",
       "      <td>436707.0</td>\n",
       "      <td>437358.0</td>\n",
       "      <td>436968.0</td>\n",
       "      <td>437294.0</td>\n",
       "      <td>437148.0</td>\n",
       "      <td>435829.0</td>\n",
       "      <td>436756.0</td>\n",
       "      <td>437266.0</td>\n",
       "      <td>436274.0</td>\n",
       "      <td>437747.0</td>\n",
       "      <td>436814.0</td>\n",
       "      <td>437155.0</td>\n",
       "      <td>435542.0</td>\n",
       "      <td>436515.0</td>\n",
       "      <td>436466.0</td>\n",
       "      <td>435281.0</td>\n",
       "      <td>436274.0</td>\n",
       "      <td>436366.0</td>\n",
       "      <td>421341.0</td>\n",
       "      <td>431323.0</td>\n",
       "      <td>433637.0</td>\n",
       "      <td>435091.0</td>\n",
       "      <td>433659.0</td>\n",
       "      <td>403742.0</td>\n",
       "      <td>415302.0</td>\n",
       "      <td>406736.0</td>\n",
       "      <td>413296.0</td>\n",
       "      <td>418489.0</td>\n",
       "      <td>350903.0</td>\n",
       "      <td>409586.0</td>\n",
       "      <td>416254.0</td>\n",
       "      <td>...</td>\n",
       "      <td>210561.0</td>\n",
       "      <td>209522.0</td>\n",
       "      <td>218191.0</td>\n",
       "      <td>214592.0</td>\n",
       "      <td>217988.0</td>\n",
       "      <td>216522.0</td>\n",
       "      <td>202751.0</td>\n",
       "      <td>200563.0</td>\n",
       "      <td>197799.0</td>\n",
       "      <td>190110.0</td>\n",
       "      <td>200160.0</td>\n",
       "      <td>129952.0</td>\n",
       "      <td>96157.0</td>\n",
       "      <td>131661.0</td>\n",
       "      <td>151361.0</td>\n",
       "      <td>166282.0</td>\n",
       "      <td>162639.0</td>\n",
       "      <td>189208.0</td>\n",
       "      <td>209711.0</td>\n",
       "      <td>211814.0</td>\n",
       "      <td>213893.0</td>\n",
       "      <td>211767.0</td>\n",
       "      <td>215793.0</td>\n",
       "      <td>219257.0</td>\n",
       "      <td>222014.0</td>\n",
       "      <td>222111.0</td>\n",
       "      <td>210061.0</td>\n",
       "      <td>213342.0</td>\n",
       "      <td>203625.0</td>\n",
       "      <td>223728.0</td>\n",
       "      <td>223085.0</td>\n",
       "      <td>222166.0</td>\n",
       "      <td>222981.0</td>\n",
       "      <td>222238.0</td>\n",
       "      <td>223187.0</td>\n",
       "      <td>221346.0</td>\n",
       "      <td>223653.0</td>\n",
       "      <td>222840.0</td>\n",
       "      <td>221726.0</td>\n",
       "      <td>223711.0</td>\n",
       "      <td>223564.0</td>\n",
       "      <td>223476.0</td>\n",
       "      <td>223653.0</td>\n",
       "      <td>223516.0</td>\n",
       "      <td>223699.0</td>\n",
       "      <td>223628.0</td>\n",
       "      <td>223997.0</td>\n",
       "      <td>223552.0</td>\n",
       "      <td>223703.0</td>\n",
       "      <td>223943.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>1</th>\n",
       "      <td>259.0</td>\n",
       "      <td>221.0</td>\n",
       "      <td>147.0</td>\n",
       "      <td>141.0</td>\n",
       "      <td>179.0</td>\n",
       "      <td>175.0</td>\n",
       "      <td>145.0</td>\n",
       "      <td>154.0</td>\n",
       "      <td>293.0</td>\n",
       "      <td>127.0</td>\n",
       "      <td>114.0</td>\n",
       "      <td>188.0</td>\n",
       "      <td>218.0</td>\n",
       "      <td>147.0</td>\n",
       "      <td>299.0</td>\n",
       "      <td>118.0</td>\n",
       "      <td>113.0</td>\n",
       "      <td>389.0</td>\n",
       "      <td>169.0</td>\n",
       "      <td>72.0</td>\n",
       "      <td>61.0</td>\n",
       "      <td>122.0</td>\n",
       "      <td>69.0</td>\n",
       "      <td>80.0</td>\n",
       "      <td>501.0</td>\n",
       "      <td>187.0</td>\n",
       "      <td>57.0</td>\n",
       "      <td>196.0</td>\n",
       "      <td>15.0</td>\n",
       "      <td>125.0</td>\n",
       "      <td>62.0</td>\n",
       "      <td>130.0</td>\n",
       "      <td>100.0</td>\n",
       "      <td>133.0</td>\n",
       "      <td>89.0</td>\n",
       "      <td>91.0</td>\n",
       "      <td>60.0</td>\n",
       "      <td>102.0</td>\n",
       "      <td>111.0</td>\n",
       "      <td>104.0</td>\n",
       "      <td>158.0</td>\n",
       "      <td>91.0</td>\n",
       "      <td>142.0</td>\n",
       "      <td>357.0</td>\n",
       "      <td>149.0</td>\n",
       "      <td>201.0</td>\n",
       "      <td>105.0</td>\n",
       "      <td>81.0</td>\n",
       "      <td>59.0</td>\n",
       "      <td>134.0</td>\n",
       "      <td>...</td>\n",
       "      <td>332.0</td>\n",
       "      <td>77.0</td>\n",
       "      <td>362.0</td>\n",
       "      <td>91.0</td>\n",
       "      <td>70.0</td>\n",
       "      <td>133.0</td>\n",
       "      <td>87.0</td>\n",
       "      <td>49.0</td>\n",
       "      <td>40.0</td>\n",
       "      <td>139.0</td>\n",
       "      <td>63.0</td>\n",
       "      <td>107.0</td>\n",
       "      <td>84.0</td>\n",
       "      <td>66.0</td>\n",
       "      <td>202.0</td>\n",
       "      <td>39.0</td>\n",
       "      <td>121.0</td>\n",
       "      <td>101.0</td>\n",
       "      <td>81.0</td>\n",
       "      <td>96.0</td>\n",
       "      <td>30.0</td>\n",
       "      <td>42.0</td>\n",
       "      <td>78.0</td>\n",
       "      <td>54.0</td>\n",
       "      <td>39.0</td>\n",
       "      <td>70.0</td>\n",
       "      <td>23.0</td>\n",
       "      <td>27.0</td>\n",
       "      <td>21.0</td>\n",
       "      <td>16.0</td>\n",
       "      <td>29.0</td>\n",
       "      <td>59.0</td>\n",
       "      <td>50.0</td>\n",
       "      <td>50.0</td>\n",
       "      <td>38.0</td>\n",
       "      <td>73.0</td>\n",
       "      <td>23.0</td>\n",
       "      <td>25.0</td>\n",
       "      <td>39.0</td>\n",
       "      <td>81.0</td>\n",
       "      <td>55.0</td>\n",
       "      <td>70.0</td>\n",
       "      <td>63.0</td>\n",
       "      <td>63.0</td>\n",
       "      <td>64.0</td>\n",
       "      <td>48.0</td>\n",
       "      <td>36.0</td>\n",
       "      <td>166.0</td>\n",
       "      <td>41.0</td>\n",
       "      <td>51.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>2</th>\n",
       "      <td>147.0</td>\n",
       "      <td>91.0</td>\n",
       "      <td>87.0</td>\n",
       "      <td>99.0</td>\n",
       "      <td>160.0</td>\n",
       "      <td>161.0</td>\n",
       "      <td>128.0</td>\n",
       "      <td>109.0</td>\n",
       "      <td>148.0</td>\n",
       "      <td>126.0</td>\n",
       "      <td>102.0</td>\n",
       "      <td>116.0</td>\n",
       "      <td>142.0</td>\n",
       "      <td>110.0</td>\n",
       "      <td>152.0</td>\n",
       "      <td>97.0</td>\n",
       "      <td>106.0</td>\n",
       "      <td>129.0</td>\n",
       "      <td>131.0</td>\n",
       "      <td>71.0</td>\n",
       "      <td>59.0</td>\n",
       "      <td>71.0</td>\n",
       "      <td>67.0</td>\n",
       "      <td>72.0</td>\n",
       "      <td>330.0</td>\n",
       "      <td>85.0</td>\n",
       "      <td>52.0</td>\n",
       "      <td>85.0</td>\n",
       "      <td>11.0</td>\n",
       "      <td>107.0</td>\n",
       "      <td>58.0</td>\n",
       "      <td>105.0</td>\n",
       "      <td>63.0</td>\n",
       "      <td>128.0</td>\n",
       "      <td>81.0</td>\n",
       "      <td>75.0</td>\n",
       "      <td>53.0</td>\n",
       "      <td>64.0</td>\n",
       "      <td>91.0</td>\n",
       "      <td>89.0</td>\n",
       "      <td>55.0</td>\n",
       "      <td>73.0</td>\n",
       "      <td>110.0</td>\n",
       "      <td>119.0</td>\n",
       "      <td>110.0</td>\n",
       "      <td>95.0</td>\n",
       "      <td>98.0</td>\n",
       "      <td>64.0</td>\n",
       "      <td>57.0</td>\n",
       "      <td>86.0</td>\n",
       "      <td>...</td>\n",
       "      <td>119.0</td>\n",
       "      <td>75.0</td>\n",
       "      <td>52.0</td>\n",
       "      <td>44.0</td>\n",
       "      <td>70.0</td>\n",
       "      <td>62.0</td>\n",
       "      <td>48.0</td>\n",
       "      <td>46.0</td>\n",
       "      <td>38.0</td>\n",
       "      <td>89.0</td>\n",
       "      <td>51.0</td>\n",
       "      <td>43.0</td>\n",
       "      <td>72.0</td>\n",
       "      <td>54.0</td>\n",
       "      <td>94.0</td>\n",
       "      <td>38.0</td>\n",
       "      <td>50.0</td>\n",
       "      <td>64.0</td>\n",
       "      <td>55.0</td>\n",
       "      <td>95.0</td>\n",
       "      <td>30.0</td>\n",
       "      <td>29.0</td>\n",
       "      <td>35.0</td>\n",
       "      <td>48.0</td>\n",
       "      <td>35.0</td>\n",
       "      <td>37.0</td>\n",
       "      <td>22.0</td>\n",
       "      <td>24.0</td>\n",
       "      <td>15.0</td>\n",
       "      <td>12.0</td>\n",
       "      <td>24.0</td>\n",
       "      <td>49.0</td>\n",
       "      <td>44.0</td>\n",
       "      <td>40.0</td>\n",
       "      <td>37.0</td>\n",
       "      <td>35.0</td>\n",
       "      <td>16.0</td>\n",
       "      <td>23.0</td>\n",
       "      <td>35.0</td>\n",
       "      <td>35.0</td>\n",
       "      <td>54.0</td>\n",
       "      <td>42.0</td>\n",
       "      <td>60.0</td>\n",
       "      <td>54.0</td>\n",
       "      <td>57.0</td>\n",
       "      <td>46.0</td>\n",
       "      <td>32.0</td>\n",
       "      <td>46.0</td>\n",
       "      <td>33.0</td>\n",
       "      <td>45.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>3</th>\n",
       "      <td>100.0</td>\n",
       "      <td>82.0</td>\n",
       "      <td>79.0</td>\n",
       "      <td>80.0</td>\n",
       "      <td>80.0</td>\n",
       "      <td>96.0</td>\n",
       "      <td>120.0</td>\n",
       "      <td>104.0</td>\n",
       "      <td>141.0</td>\n",
       "      <td>93.0</td>\n",
       "      <td>89.0</td>\n",
       "      <td>114.0</td>\n",
       "      <td>136.0</td>\n",
       "      <td>74.0</td>\n",
       "      <td>97.0</td>\n",
       "      <td>54.0</td>\n",
       "      <td>99.0</td>\n",
       "      <td>109.0</td>\n",
       "      <td>94.0</td>\n",
       "      <td>71.0</td>\n",
       "      <td>56.0</td>\n",
       "      <td>68.0</td>\n",
       "      <td>61.0</td>\n",
       "      <td>61.0</td>\n",
       "      <td>99.0</td>\n",
       "      <td>68.0</td>\n",
       "      <td>44.0</td>\n",
       "      <td>54.0</td>\n",
       "      <td>8.0</td>\n",
       "      <td>81.0</td>\n",
       "      <td>44.0</td>\n",
       "      <td>87.0</td>\n",
       "      <td>49.0</td>\n",
       "      <td>66.0</td>\n",
       "      <td>63.0</td>\n",
       "      <td>46.0</td>\n",
       "      <td>48.0</td>\n",
       "      <td>52.0</td>\n",
       "      <td>68.0</td>\n",
       "      <td>77.0</td>\n",
       "      <td>52.0</td>\n",
       "      <td>58.0</td>\n",
       "      <td>66.0</td>\n",
       "      <td>52.0</td>\n",
       "      <td>54.0</td>\n",
       "      <td>90.0</td>\n",
       "      <td>97.0</td>\n",
       "      <td>60.0</td>\n",
       "      <td>53.0</td>\n",
       "      <td>79.0</td>\n",
       "      <td>...</td>\n",
       "      <td>45.0</td>\n",
       "      <td>65.0</td>\n",
       "      <td>42.0</td>\n",
       "      <td>39.0</td>\n",
       "      <td>67.0</td>\n",
       "      <td>57.0</td>\n",
       "      <td>37.0</td>\n",
       "      <td>32.0</td>\n",
       "      <td>37.0</td>\n",
       "      <td>47.0</td>\n",
       "      <td>49.0</td>\n",
       "      <td>26.0</td>\n",
       "      <td>21.0</td>\n",
       "      <td>47.0</td>\n",
       "      <td>42.0</td>\n",
       "      <td>37.0</td>\n",
       "      <td>42.0</td>\n",
       "      <td>46.0</td>\n",
       "      <td>49.0</td>\n",
       "      <td>80.0</td>\n",
       "      <td>19.0</td>\n",
       "      <td>25.0</td>\n",
       "      <td>34.0</td>\n",
       "      <td>46.0</td>\n",
       "      <td>34.0</td>\n",
       "      <td>30.0</td>\n",
       "      <td>15.0</td>\n",
       "      <td>19.0</td>\n",
       "      <td>10.0</td>\n",
       "      <td>9.0</td>\n",
       "      <td>20.0</td>\n",
       "      <td>47.0</td>\n",
       "      <td>36.0</td>\n",
       "      <td>32.0</td>\n",
       "      <td>34.0</td>\n",
       "      <td>35.0</td>\n",
       "      <td>16.0</td>\n",
       "      <td>18.0</td>\n",
       "      <td>28.0</td>\n",
       "      <td>31.0</td>\n",
       "      <td>43.0</td>\n",
       "      <td>41.0</td>\n",
       "      <td>49.0</td>\n",
       "      <td>50.0</td>\n",
       "      <td>46.0</td>\n",
       "      <td>44.0</td>\n",
       "      <td>28.0</td>\n",
       "      <td>38.0</td>\n",
       "      <td>29.0</td>\n",
       "      <td>41.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>4</th>\n",
       "      <td>97.0</td>\n",
       "      <td>76.0</td>\n",
       "      <td>78.0</td>\n",
       "      <td>48.0</td>\n",
       "      <td>62.0</td>\n",
       "      <td>90.0</td>\n",
       "      <td>85.0</td>\n",
       "      <td>70.0</td>\n",
       "      <td>109.0</td>\n",
       "      <td>62.0</td>\n",
       "      <td>78.0</td>\n",
       "      <td>68.0</td>\n",
       "      <td>111.0</td>\n",
       "      <td>62.0</td>\n",
       "      <td>89.0</td>\n",
       "      <td>50.0</td>\n",
       "      <td>91.0</td>\n",
       "      <td>77.0</td>\n",
       "      <td>63.0</td>\n",
       "      <td>70.0</td>\n",
       "      <td>54.0</td>\n",
       "      <td>51.0</td>\n",
       "      <td>40.0</td>\n",
       "      <td>46.0</td>\n",
       "      <td>57.0</td>\n",
       "      <td>59.0</td>\n",
       "      <td>39.0</td>\n",
       "      <td>51.0</td>\n",
       "      <td>8.0</td>\n",
       "      <td>70.0</td>\n",
       "      <td>29.0</td>\n",
       "      <td>29.0</td>\n",
       "      <td>46.0</td>\n",
       "      <td>57.0</td>\n",
       "      <td>48.0</td>\n",
       "      <td>33.0</td>\n",
       "      <td>27.0</td>\n",
       "      <td>51.0</td>\n",
       "      <td>41.0</td>\n",
       "      <td>65.0</td>\n",
       "      <td>49.0</td>\n",
       "      <td>43.0</td>\n",
       "      <td>57.0</td>\n",
       "      <td>44.0</td>\n",
       "      <td>50.0</td>\n",
       "      <td>76.0</td>\n",
       "      <td>95.0</td>\n",
       "      <td>45.0</td>\n",
       "      <td>40.0</td>\n",
       "      <td>79.0</td>\n",
       "      <td>...</td>\n",
       "      <td>39.0</td>\n",
       "      <td>59.0</td>\n",
       "      <td>39.0</td>\n",
       "      <td>33.0</td>\n",
       "      <td>38.0</td>\n",
       "      <td>53.0</td>\n",
       "      <td>31.0</td>\n",
       "      <td>31.0</td>\n",
       "      <td>34.0</td>\n",
       "      <td>41.0</td>\n",
       "      <td>45.0</td>\n",
       "      <td>23.0</td>\n",
       "      <td>20.0</td>\n",
       "      <td>45.0</td>\n",
       "      <td>41.0</td>\n",
       "      <td>36.0</td>\n",
       "      <td>40.0</td>\n",
       "      <td>37.0</td>\n",
       "      <td>47.0</td>\n",
       "      <td>64.0</td>\n",
       "      <td>16.0</td>\n",
       "      <td>16.0</td>\n",
       "      <td>31.0</td>\n",
       "      <td>35.0</td>\n",
       "      <td>31.0</td>\n",
       "      <td>24.0</td>\n",
       "      <td>12.0</td>\n",
       "      <td>13.0</td>\n",
       "      <td>6.0</td>\n",
       "      <td>9.0</td>\n",
       "      <td>19.0</td>\n",
       "      <td>36.0</td>\n",
       "      <td>26.0</td>\n",
       "      <td>30.0</td>\n",
       "      <td>26.0</td>\n",
       "      <td>19.0</td>\n",
       "      <td>13.0</td>\n",
       "      <td>17.0</td>\n",
       "      <td>22.0</td>\n",
       "      <td>29.0</td>\n",
       "      <td>42.0</td>\n",
       "      <td>38.0</td>\n",
       "      <td>46.0</td>\n",
       "      <td>47.0</td>\n",
       "      <td>46.0</td>\n",
       "      <td>42.0</td>\n",
       "      <td>28.0</td>\n",
       "      <td>33.0</td>\n",
       "      <td>25.0</td>\n",
       "      <td>33.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>5</th>\n",
       "      <td>75.0</td>\n",
       "      <td>64.0</td>\n",
       "      <td>76.0</td>\n",
       "      <td>42.0</td>\n",
       "      <td>57.0</td>\n",
       "      <td>82.0</td>\n",
       "      <td>82.0</td>\n",
       "      <td>69.0</td>\n",
       "      <td>96.0</td>\n",
       "      <td>54.0</td>\n",
       "      <td>72.0</td>\n",
       "      <td>66.0</td>\n",
       "      <td>100.0</td>\n",
       "      <td>61.0</td>\n",
       "      <td>75.0</td>\n",
       "      <td>42.0</td>\n",
       "      <td>89.0</td>\n",
       "      <td>67.0</td>\n",
       "      <td>60.0</td>\n",
       "      <td>69.0</td>\n",
       "      <td>53.0</td>\n",
       "      <td>47.0</td>\n",
       "      <td>38.0</td>\n",
       "      <td>45.0</td>\n",
       "      <td>51.0</td>\n",
       "      <td>59.0</td>\n",
       "      <td>36.0</td>\n",
       "      <td>38.0</td>\n",
       "      <td>6.0</td>\n",
       "      <td>65.0</td>\n",
       "      <td>26.0</td>\n",
       "      <td>28.0</td>\n",
       "      <td>44.0</td>\n",
       "      <td>54.0</td>\n",
       "      <td>46.0</td>\n",
       "      <td>29.0</td>\n",
       "      <td>25.0</td>\n",
       "      <td>48.0</td>\n",
       "      <td>39.0</td>\n",
       "      <td>56.0</td>\n",
       "      <td>42.0</td>\n",
       "      <td>42.0</td>\n",
       "      <td>19.0</td>\n",
       "      <td>42.0</td>\n",
       "      <td>48.0</td>\n",
       "      <td>55.0</td>\n",
       "      <td>90.0</td>\n",
       "      <td>33.0</td>\n",
       "      <td>11.0</td>\n",
       "      <td>69.0</td>\n",
       "      <td>...</td>\n",
       "      <td>39.0</td>\n",
       "      <td>51.0</td>\n",
       "      <td>38.0</td>\n",
       "      <td>30.0</td>\n",
       "      <td>35.0</td>\n",
       "      <td>37.0</td>\n",
       "      <td>30.0</td>\n",
       "      <td>31.0</td>\n",
       "      <td>33.0</td>\n",
       "      <td>39.0</td>\n",
       "      <td>38.0</td>\n",
       "      <td>19.0</td>\n",
       "      <td>19.0</td>\n",
       "      <td>36.0</td>\n",
       "      <td>38.0</td>\n",
       "      <td>34.0</td>\n",
       "      <td>38.0</td>\n",
       "      <td>35.0</td>\n",
       "      <td>42.0</td>\n",
       "      <td>48.0</td>\n",
       "      <td>12.0</td>\n",
       "      <td>16.0</td>\n",
       "      <td>26.0</td>\n",
       "      <td>34.0</td>\n",
       "      <td>28.0</td>\n",
       "      <td>22.0</td>\n",
       "      <td>11.0</td>\n",
       "      <td>12.0</td>\n",
       "      <td>6.0</td>\n",
       "      <td>9.0</td>\n",
       "      <td>19.0</td>\n",
       "      <td>32.0</td>\n",
       "      <td>20.0</td>\n",
       "      <td>23.0</td>\n",
       "      <td>22.0</td>\n",
       "      <td>18.0</td>\n",
       "      <td>11.0</td>\n",
       "      <td>16.0</td>\n",
       "      <td>17.0</td>\n",
       "      <td>25.0</td>\n",
       "      <td>36.0</td>\n",
       "      <td>31.0</td>\n",
       "      <td>37.0</td>\n",
       "      <td>40.0</td>\n",
       "      <td>44.0</td>\n",
       "      <td>41.0</td>\n",
       "      <td>26.0</td>\n",
       "      <td>25.0</td>\n",
       "      <td>24.0</td>\n",
       "      <td>30.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>6</th>\n",
       "      <td>71.0</td>\n",
       "      <td>58.0</td>\n",
       "      <td>68.0</td>\n",
       "      <td>41.0</td>\n",
       "      <td>40.0</td>\n",
       "      <td>79.0</td>\n",
       "      <td>70.0</td>\n",
       "      <td>69.0</td>\n",
       "      <td>82.0</td>\n",
       "      <td>53.0</td>\n",
       "      <td>70.0</td>\n",
       "      <td>59.0</td>\n",
       "      <td>85.0</td>\n",
       "      <td>54.0</td>\n",
       "      <td>52.0</td>\n",
       "      <td>40.0</td>\n",
       "      <td>74.0</td>\n",
       "      <td>67.0</td>\n",
       "      <td>59.0</td>\n",
       "      <td>62.0</td>\n",
       "      <td>47.0</td>\n",
       "      <td>43.0</td>\n",
       "      <td>35.0</td>\n",
       "      <td>40.0</td>\n",
       "      <td>45.0</td>\n",
       "      <td>43.0</td>\n",
       "      <td>36.0</td>\n",
       "      <td>29.0</td>\n",
       "      <td>5.0</td>\n",
       "      <td>50.0</td>\n",
       "      <td>17.0</td>\n",
       "      <td>27.0</td>\n",
       "      <td>44.0</td>\n",
       "      <td>52.0</td>\n",
       "      <td>42.0</td>\n",
       "      <td>26.0</td>\n",
       "      <td>23.0</td>\n",
       "      <td>47.0</td>\n",
       "      <td>35.0</td>\n",
       "      <td>33.0</td>\n",
       "      <td>34.0</td>\n",
       "      <td>36.0</td>\n",
       "      <td>3.0</td>\n",
       "      <td>29.0</td>\n",
       "      <td>43.0</td>\n",
       "      <td>35.0</td>\n",
       "      <td>79.0</td>\n",
       "      <td>31.0</td>\n",
       "      <td>11.0</td>\n",
       "      <td>44.0</td>\n",
       "      <td>...</td>\n",
       "      <td>35.0</td>\n",
       "      <td>50.0</td>\n",
       "      <td>38.0</td>\n",
       "      <td>29.0</td>\n",
       "      <td>35.0</td>\n",
       "      <td>35.0</td>\n",
       "      <td>29.0</td>\n",
       "      <td>30.0</td>\n",
       "      <td>31.0</td>\n",
       "      <td>38.0</td>\n",
       "      <td>32.0</td>\n",
       "      <td>18.0</td>\n",
       "      <td>19.0</td>\n",
       "      <td>25.0</td>\n",
       "      <td>31.0</td>\n",
       "      <td>30.0</td>\n",
       "      <td>34.0</td>\n",
       "      <td>29.0</td>\n",
       "      <td>41.0</td>\n",
       "      <td>44.0</td>\n",
       "      <td>12.0</td>\n",
       "      <td>14.0</td>\n",
       "      <td>18.0</td>\n",
       "      <td>34.0</td>\n",
       "      <td>25.0</td>\n",
       "      <td>15.0</td>\n",
       "      <td>9.0</td>\n",
       "      <td>9.0</td>\n",
       "      <td>5.0</td>\n",
       "      <td>8.0</td>\n",
       "      <td>15.0</td>\n",
       "      <td>31.0</td>\n",
       "      <td>16.0</td>\n",
       "      <td>22.0</td>\n",
       "      <td>22.0</td>\n",
       "      <td>15.0</td>\n",
       "      <td>9.0</td>\n",
       "      <td>16.0</td>\n",
       "      <td>12.0</td>\n",
       "      <td>23.0</td>\n",
       "      <td>34.0</td>\n",
       "      <td>22.0</td>\n",
       "      <td>36.0</td>\n",
       "      <td>36.0</td>\n",
       "      <td>29.0</td>\n",
       "      <td>36.0</td>\n",
       "      <td>19.0</td>\n",
       "      <td>22.0</td>\n",
       "      <td>17.0</td>\n",
       "      <td>30.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>7</th>\n",
       "      <td>68.0</td>\n",
       "      <td>58.0</td>\n",
       "      <td>63.0</td>\n",
       "      <td>39.0</td>\n",
       "      <td>36.0</td>\n",
       "      <td>78.0</td>\n",
       "      <td>69.0</td>\n",
       "      <td>68.0</td>\n",
       "      <td>77.0</td>\n",
       "      <td>42.0</td>\n",
       "      <td>60.0</td>\n",
       "      <td>56.0</td>\n",
       "      <td>67.0</td>\n",
       "      <td>52.0</td>\n",
       "      <td>47.0</td>\n",
       "      <td>32.0</td>\n",
       "      <td>73.0</td>\n",
       "      <td>60.0</td>\n",
       "      <td>56.0</td>\n",
       "      <td>57.0</td>\n",
       "      <td>44.0</td>\n",
       "      <td>40.0</td>\n",
       "      <td>33.0</td>\n",
       "      <td>37.0</td>\n",
       "      <td>31.0</td>\n",
       "      <td>38.0</td>\n",
       "      <td>33.0</td>\n",
       "      <td>29.0</td>\n",
       "      <td>4.0</td>\n",
       "      <td>46.0</td>\n",
       "      <td>16.0</td>\n",
       "      <td>25.0</td>\n",
       "      <td>41.0</td>\n",
       "      <td>43.0</td>\n",
       "      <td>34.0</td>\n",
       "      <td>26.0</td>\n",
       "      <td>18.0</td>\n",
       "      <td>43.0</td>\n",
       "      <td>33.0</td>\n",
       "      <td>32.0</td>\n",
       "      <td>32.0</td>\n",
       "      <td>34.0</td>\n",
       "      <td>1.0</td>\n",
       "      <td>26.0</td>\n",
       "      <td>42.0</td>\n",
       "      <td>35.0</td>\n",
       "      <td>71.0</td>\n",
       "      <td>24.0</td>\n",
       "      <td>9.0</td>\n",
       "      <td>39.0</td>\n",
       "      <td>...</td>\n",
       "      <td>30.0</td>\n",
       "      <td>49.0</td>\n",
       "      <td>36.0</td>\n",
       "      <td>26.0</td>\n",
       "      <td>34.0</td>\n",
       "      <td>35.0</td>\n",
       "      <td>28.0</td>\n",
       "      <td>27.0</td>\n",
       "      <td>29.0</td>\n",
       "      <td>29.0</td>\n",
       "      <td>31.0</td>\n",
       "      <td>18.0</td>\n",
       "      <td>16.0</td>\n",
       "      <td>18.0</td>\n",
       "      <td>25.0</td>\n",
       "      <td>29.0</td>\n",
       "      <td>27.0</td>\n",
       "      <td>24.0</td>\n",
       "      <td>32.0</td>\n",
       "      <td>35.0</td>\n",
       "      <td>7.0</td>\n",
       "      <td>12.0</td>\n",
       "      <td>17.0</td>\n",
       "      <td>28.0</td>\n",
       "      <td>24.0</td>\n",
       "      <td>10.0</td>\n",
       "      <td>8.0</td>\n",
       "      <td>5.0</td>\n",
       "      <td>4.0</td>\n",
       "      <td>7.0</td>\n",
       "      <td>14.0</td>\n",
       "      <td>27.0</td>\n",
       "      <td>16.0</td>\n",
       "      <td>21.0</td>\n",
       "      <td>18.0</td>\n",
       "      <td>14.0</td>\n",
       "      <td>8.0</td>\n",
       "      <td>10.0</td>\n",
       "      <td>11.0</td>\n",
       "      <td>23.0</td>\n",
       "      <td>29.0</td>\n",
       "      <td>21.0</td>\n",
       "      <td>32.0</td>\n",
       "      <td>33.0</td>\n",
       "      <td>27.0</td>\n",
       "      <td>34.0</td>\n",
       "      <td>16.0</td>\n",
       "      <td>21.0</td>\n",
       "      <td>13.0</td>\n",
       "      <td>28.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>8</th>\n",
       "      <td>66.0</td>\n",
       "      <td>56.0</td>\n",
       "      <td>61.0</td>\n",
       "      <td>30.0</td>\n",
       "      <td>34.0</td>\n",
       "      <td>66.0</td>\n",
       "      <td>67.0</td>\n",
       "      <td>66.0</td>\n",
       "      <td>72.0</td>\n",
       "      <td>41.0</td>\n",
       "      <td>50.0</td>\n",
       "      <td>46.0</td>\n",
       "      <td>46.0</td>\n",
       "      <td>51.0</td>\n",
       "      <td>43.0</td>\n",
       "      <td>31.0</td>\n",
       "      <td>55.0</td>\n",
       "      <td>50.0</td>\n",
       "      <td>52.0</td>\n",
       "      <td>47.0</td>\n",
       "      <td>41.0</td>\n",
       "      <td>39.0</td>\n",
       "      <td>33.0</td>\n",
       "      <td>36.0</td>\n",
       "      <td>30.0</td>\n",
       "      <td>38.0</td>\n",
       "      <td>30.0</td>\n",
       "      <td>24.0</td>\n",
       "      <td>3.0</td>\n",
       "      <td>45.0</td>\n",
       "      <td>16.0</td>\n",
       "      <td>23.0</td>\n",
       "      <td>31.0</td>\n",
       "      <td>43.0</td>\n",
       "      <td>29.0</td>\n",
       "      <td>25.0</td>\n",
       "      <td>16.0</td>\n",
       "      <td>35.0</td>\n",
       "      <td>33.0</td>\n",
       "      <td>27.0</td>\n",
       "      <td>31.0</td>\n",
       "      <td>25.0</td>\n",
       "      <td>1.0</td>\n",
       "      <td>19.0</td>\n",
       "      <td>32.0</td>\n",
       "      <td>27.0</td>\n",
       "      <td>70.0</td>\n",
       "      <td>24.0</td>\n",
       "      <td>9.0</td>\n",
       "      <td>38.0</td>\n",
       "      <td>...</td>\n",
       "      <td>26.0</td>\n",
       "      <td>48.0</td>\n",
       "      <td>33.0</td>\n",
       "      <td>25.0</td>\n",
       "      <td>29.0</td>\n",
       "      <td>34.0</td>\n",
       "      <td>24.0</td>\n",
       "      <td>24.0</td>\n",
       "      <td>28.0</td>\n",
       "      <td>21.0</td>\n",
       "      <td>26.0</td>\n",
       "      <td>17.0</td>\n",
       "      <td>16.0</td>\n",
       "      <td>17.0</td>\n",
       "      <td>25.0</td>\n",
       "      <td>29.0</td>\n",
       "      <td>23.0</td>\n",
       "      <td>19.0</td>\n",
       "      <td>28.0</td>\n",
       "      <td>34.0</td>\n",
       "      <td>6.0</td>\n",
       "      <td>11.0</td>\n",
       "      <td>16.0</td>\n",
       "      <td>23.0</td>\n",
       "      <td>24.0</td>\n",
       "      <td>10.0</td>\n",
       "      <td>7.0</td>\n",
       "      <td>4.0</td>\n",
       "      <td>3.0</td>\n",
       "      <td>6.0</td>\n",
       "      <td>14.0</td>\n",
       "      <td>24.0</td>\n",
       "      <td>10.0</td>\n",
       "      <td>17.0</td>\n",
       "      <td>15.0</td>\n",
       "      <td>14.0</td>\n",
       "      <td>8.0</td>\n",
       "      <td>9.0</td>\n",
       "      <td>10.0</td>\n",
       "      <td>17.0</td>\n",
       "      <td>29.0</td>\n",
       "      <td>18.0</td>\n",
       "      <td>30.0</td>\n",
       "      <td>28.0</td>\n",
       "      <td>25.0</td>\n",
       "      <td>28.0</td>\n",
       "      <td>16.0</td>\n",
       "      <td>16.0</td>\n",
       "      <td>9.0</td>\n",
       "      <td>24.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>9</th>\n",
       "      <td>53.0</td>\n",
       "      <td>46.0</td>\n",
       "      <td>55.0</td>\n",
       "      <td>29.0</td>\n",
       "      <td>31.0</td>\n",
       "      <td>58.0</td>\n",
       "      <td>40.0</td>\n",
       "      <td>45.0</td>\n",
       "      <td>69.0</td>\n",
       "      <td>39.0</td>\n",
       "      <td>49.0</td>\n",
       "      <td>36.0</td>\n",
       "      <td>38.0</td>\n",
       "      <td>50.0</td>\n",
       "      <td>43.0</td>\n",
       "      <td>27.0</td>\n",
       "      <td>52.0</td>\n",
       "      <td>46.0</td>\n",
       "      <td>45.0</td>\n",
       "      <td>43.0</td>\n",
       "      <td>33.0</td>\n",
       "      <td>31.0</td>\n",
       "      <td>31.0</td>\n",
       "      <td>33.0</td>\n",
       "      <td>28.0</td>\n",
       "      <td>33.0</td>\n",
       "      <td>29.0</td>\n",
       "      <td>19.0</td>\n",
       "      <td>2.0</td>\n",
       "      <td>31.0</td>\n",
       "      <td>16.0</td>\n",
       "      <td>22.0</td>\n",
       "      <td>28.0</td>\n",
       "      <td>37.0</td>\n",
       "      <td>27.0</td>\n",
       "      <td>21.0</td>\n",
       "      <td>16.0</td>\n",
       "      <td>34.0</td>\n",
       "      <td>27.0</td>\n",
       "      <td>20.0</td>\n",
       "      <td>20.0</td>\n",
       "      <td>17.0</td>\n",
       "      <td>1.0</td>\n",
       "      <td>17.0</td>\n",
       "      <td>23.0</td>\n",
       "      <td>25.0</td>\n",
       "      <td>54.0</td>\n",
       "      <td>19.0</td>\n",
       "      <td>9.0</td>\n",
       "      <td>34.0</td>\n",
       "      <td>...</td>\n",
       "      <td>21.0</td>\n",
       "      <td>39.0</td>\n",
       "      <td>23.0</td>\n",
       "      <td>22.0</td>\n",
       "      <td>23.0</td>\n",
       "      <td>28.0</td>\n",
       "      <td>22.0</td>\n",
       "      <td>19.0</td>\n",
       "      <td>25.0</td>\n",
       "      <td>20.0</td>\n",
       "      <td>26.0</td>\n",
       "      <td>16.0</td>\n",
       "      <td>15.0</td>\n",
       "      <td>15.0</td>\n",
       "      <td>18.0</td>\n",
       "      <td>28.0</td>\n",
       "      <td>18.0</td>\n",
       "      <td>17.0</td>\n",
       "      <td>25.0</td>\n",
       "      <td>28.0</td>\n",
       "      <td>6.0</td>\n",
       "      <td>9.0</td>\n",
       "      <td>16.0</td>\n",
       "      <td>18.0</td>\n",
       "      <td>16.0</td>\n",
       "      <td>9.0</td>\n",
       "      <td>6.0</td>\n",
       "      <td>4.0</td>\n",
       "      <td>3.0</td>\n",
       "      <td>6.0</td>\n",
       "      <td>13.0</td>\n",
       "      <td>22.0</td>\n",
       "      <td>7.0</td>\n",
       "      <td>16.0</td>\n",
       "      <td>13.0</td>\n",
       "      <td>12.0</td>\n",
       "      <td>8.0</td>\n",
       "      <td>9.0</td>\n",
       "      <td>9.0</td>\n",
       "      <td>16.0</td>\n",
       "      <td>26.0</td>\n",
       "      <td>16.0</td>\n",
       "      <td>27.0</td>\n",
       "      <td>28.0</td>\n",
       "      <td>24.0</td>\n",
       "      <td>22.0</td>\n",
       "      <td>15.0</td>\n",
       "      <td>13.0</td>\n",
       "      <td>8.0</td>\n",
       "      <td>19.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>10</th>\n",
       "      <td>52.0</td>\n",
       "      <td>42.0</td>\n",
       "      <td>55.0</td>\n",
       "      <td>28.0</td>\n",
       "      <td>29.0</td>\n",
       "      <td>52.0</td>\n",
       "      <td>35.0</td>\n",
       "      <td>32.0</td>\n",
       "      <td>61.0</td>\n",
       "      <td>37.0</td>\n",
       "      <td>46.0</td>\n",
       "      <td>27.0</td>\n",
       "      <td>37.0</td>\n",
       "      <td>43.0</td>\n",
       "      <td>38.0</td>\n",
       "      <td>26.0</td>\n",
       "      <td>50.0</td>\n",
       "      <td>34.0</td>\n",
       "      <td>45.0</td>\n",
       "      <td>37.0</td>\n",
       "      <td>27.0</td>\n",
       "      <td>27.0</td>\n",
       "      <td>26.0</td>\n",
       "      <td>30.0</td>\n",
       "      <td>26.0</td>\n",
       "      <td>29.0</td>\n",
       "      <td>28.0</td>\n",
       "      <td>19.0</td>\n",
       "      <td>2.0</td>\n",
       "      <td>29.0</td>\n",
       "      <td>16.0</td>\n",
       "      <td>21.0</td>\n",
       "      <td>22.0</td>\n",
       "      <td>32.0</td>\n",
       "      <td>23.0</td>\n",
       "      <td>13.0</td>\n",
       "      <td>15.0</td>\n",
       "      <td>33.0</td>\n",
       "      <td>21.0</td>\n",
       "      <td>17.0</td>\n",
       "      <td>18.0</td>\n",
       "      <td>16.0</td>\n",
       "      <td>0.0</td>\n",
       "      <td>11.0</td>\n",
       "      <td>23.0</td>\n",
       "      <td>16.0</td>\n",
       "      <td>51.0</td>\n",
       "      <td>18.0</td>\n",
       "      <td>8.0</td>\n",
       "      <td>34.0</td>\n",
       "      <td>...</td>\n",
       "      <td>20.0</td>\n",
       "      <td>28.0</td>\n",
       "      <td>22.0</td>\n",
       "      <td>16.0</td>\n",
       "      <td>20.0</td>\n",
       "      <td>28.0</td>\n",
       "      <td>21.0</td>\n",
       "      <td>19.0</td>\n",
       "      <td>22.0</td>\n",
       "      <td>14.0</td>\n",
       "      <td>25.0</td>\n",
       "      <td>16.0</td>\n",
       "      <td>14.0</td>\n",
       "      <td>15.0</td>\n",
       "      <td>17.0</td>\n",
       "      <td>21.0</td>\n",
       "      <td>16.0</td>\n",
       "      <td>17.0</td>\n",
       "      <td>23.0</td>\n",
       "      <td>27.0</td>\n",
       "      <td>5.0</td>\n",
       "      <td>8.0</td>\n",
       "      <td>14.0</td>\n",
       "      <td>15.0</td>\n",
       "      <td>16.0</td>\n",
       "      <td>9.0</td>\n",
       "      <td>6.0</td>\n",
       "      <td>2.0</td>\n",
       "      <td>3.0</td>\n",
       "      <td>5.0</td>\n",
       "      <td>12.0</td>\n",
       "      <td>19.0</td>\n",
       "      <td>7.0</td>\n",
       "      <td>14.0</td>\n",
       "      <td>11.0</td>\n",
       "      <td>12.0</td>\n",
       "      <td>7.0</td>\n",
       "      <td>9.0</td>\n",
       "      <td>9.0</td>\n",
       "      <td>15.0</td>\n",
       "      <td>25.0</td>\n",
       "      <td>16.0</td>\n",
       "      <td>26.0</td>\n",
       "      <td>26.0</td>\n",
       "      <td>24.0</td>\n",
       "      <td>17.0</td>\n",
       "      <td>15.0</td>\n",
       "      <td>12.0</td>\n",
       "      <td>7.0</td>\n",
       "      <td>16.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>11</th>\n",
       "      <td>41.0</td>\n",
       "      <td>40.0</td>\n",
       "      <td>48.0</td>\n",
       "      <td>25.0</td>\n",
       "      <td>29.0</td>\n",
       "      <td>48.0</td>\n",
       "      <td>32.0</td>\n",
       "      <td>30.0</td>\n",
       "      <td>59.0</td>\n",
       "      <td>34.0</td>\n",
       "      <td>41.0</td>\n",
       "      <td>26.0</td>\n",
       "      <td>36.0</td>\n",
       "      <td>41.0</td>\n",
       "      <td>37.0</td>\n",
       "      <td>23.0</td>\n",
       "      <td>47.0</td>\n",
       "      <td>29.0</td>\n",
       "      <td>45.0</td>\n",
       "      <td>25.0</td>\n",
       "      <td>24.0</td>\n",
       "      <td>27.0</td>\n",
       "      <td>25.0</td>\n",
       "      <td>25.0</td>\n",
       "      <td>22.0</td>\n",
       "      <td>28.0</td>\n",
       "      <td>25.0</td>\n",
       "      <td>15.0</td>\n",
       "      <td>2.0</td>\n",
       "      <td>26.0</td>\n",
       "      <td>15.0</td>\n",
       "      <td>17.0</td>\n",
       "      <td>18.0</td>\n",
       "      <td>31.0</td>\n",
       "      <td>22.0</td>\n",
       "      <td>13.0</td>\n",
       "      <td>14.0</td>\n",
       "      <td>27.0</td>\n",
       "      <td>16.0</td>\n",
       "      <td>16.0</td>\n",
       "      <td>17.0</td>\n",
       "      <td>14.0</td>\n",
       "      <td>0.0</td>\n",
       "      <td>11.0</td>\n",
       "      <td>20.0</td>\n",
       "      <td>14.0</td>\n",
       "      <td>36.0</td>\n",
       "      <td>15.0</td>\n",
       "      <td>6.0</td>\n",
       "      <td>32.0</td>\n",
       "      <td>...</td>\n",
       "      <td>19.0</td>\n",
       "      <td>22.0</td>\n",
       "      <td>20.0</td>\n",
       "      <td>16.0</td>\n",
       "      <td>18.0</td>\n",
       "      <td>25.0</td>\n",
       "      <td>16.0</td>\n",
       "      <td>17.0</td>\n",
       "      <td>21.0</td>\n",
       "      <td>13.0</td>\n",
       "      <td>24.0</td>\n",
       "      <td>13.0</td>\n",
       "      <td>14.0</td>\n",
       "      <td>12.0</td>\n",
       "      <td>17.0</td>\n",
       "      <td>21.0</td>\n",
       "      <td>14.0</td>\n",
       "      <td>14.0</td>\n",
       "      <td>18.0</td>\n",
       "      <td>27.0</td>\n",
       "      <td>5.0</td>\n",
       "      <td>8.0</td>\n",
       "      <td>14.0</td>\n",
       "      <td>13.0</td>\n",
       "      <td>16.0</td>\n",
       "      <td>7.0</td>\n",
       "      <td>6.0</td>\n",
       "      <td>2.0</td>\n",
       "      <td>3.0</td>\n",
       "      <td>5.0</td>\n",
       "      <td>12.0</td>\n",
       "      <td>18.0</td>\n",
       "      <td>7.0</td>\n",
       "      <td>12.0</td>\n",
       "      <td>9.0</td>\n",
       "      <td>11.0</td>\n",
       "      <td>6.0</td>\n",
       "      <td>9.0</td>\n",
       "      <td>6.0</td>\n",
       "      <td>11.0</td>\n",
       "      <td>23.0</td>\n",
       "      <td>16.0</td>\n",
       "      <td>23.0</td>\n",
       "      <td>25.0</td>\n",
       "      <td>24.0</td>\n",
       "      <td>17.0</td>\n",
       "      <td>15.0</td>\n",
       "      <td>12.0</td>\n",
       "      <td>7.0</td>\n",
       "      <td>15.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>12</th>\n",
       "      <td>38.0</td>\n",
       "      <td>38.0</td>\n",
       "      <td>45.0</td>\n",
       "      <td>22.0</td>\n",
       "      <td>23.0</td>\n",
       "      <td>45.0</td>\n",
       "      <td>30.0</td>\n",
       "      <td>30.0</td>\n",
       "      <td>54.0</td>\n",
       "      <td>33.0</td>\n",
       "      <td>31.0</td>\n",
       "      <td>24.0</td>\n",
       "      <td>36.0</td>\n",
       "      <td>39.0</td>\n",
       "      <td>34.0</td>\n",
       "      <td>23.0</td>\n",
       "      <td>39.0</td>\n",
       "      <td>27.0</td>\n",
       "      <td>44.0</td>\n",
       "      <td>25.0</td>\n",
       "      <td>23.0</td>\n",
       "      <td>26.0</td>\n",
       "      <td>24.0</td>\n",
       "      <td>24.0</td>\n",
       "      <td>19.0</td>\n",
       "      <td>24.0</td>\n",
       "      <td>22.0</td>\n",
       "      <td>14.0</td>\n",
       "      <td>1.0</td>\n",
       "      <td>20.0</td>\n",
       "      <td>15.0</td>\n",
       "      <td>16.0</td>\n",
       "      <td>14.0</td>\n",
       "      <td>28.0</td>\n",
       "      <td>21.0</td>\n",
       "      <td>12.0</td>\n",
       "      <td>14.0</td>\n",
       "      <td>27.0</td>\n",
       "      <td>16.0</td>\n",
       "      <td>16.0</td>\n",
       "      <td>14.0</td>\n",
       "      <td>13.0</td>\n",
       "      <td>0.0</td>\n",
       "      <td>7.0</td>\n",
       "      <td>20.0</td>\n",
       "      <td>13.0</td>\n",
       "      <td>35.0</td>\n",
       "      <td>13.0</td>\n",
       "      <td>4.0</td>\n",
       "      <td>29.0</td>\n",
       "      <td>...</td>\n",
       "      <td>18.0</td>\n",
       "      <td>19.0</td>\n",
       "      <td>19.0</td>\n",
       "      <td>16.0</td>\n",
       "      <td>14.0</td>\n",
       "      <td>23.0</td>\n",
       "      <td>13.0</td>\n",
       "      <td>11.0</td>\n",
       "      <td>19.0</td>\n",
       "      <td>11.0</td>\n",
       "      <td>23.0</td>\n",
       "      <td>12.0</td>\n",
       "      <td>11.0</td>\n",
       "      <td>11.0</td>\n",
       "      <td>13.0</td>\n",
       "      <td>16.0</td>\n",
       "      <td>13.0</td>\n",
       "      <td>14.0</td>\n",
       "      <td>17.0</td>\n",
       "      <td>24.0</td>\n",
       "      <td>5.0</td>\n",
       "      <td>8.0</td>\n",
       "      <td>9.0</td>\n",
       "      <td>13.0</td>\n",
       "      <td>14.0</td>\n",
       "      <td>7.0</td>\n",
       "      <td>6.0</td>\n",
       "      <td>2.0</td>\n",
       "      <td>3.0</td>\n",
       "      <td>4.0</td>\n",
       "      <td>9.0</td>\n",
       "      <td>17.0</td>\n",
       "      <td>7.0</td>\n",
       "      <td>12.0</td>\n",
       "      <td>8.0</td>\n",
       "      <td>11.0</td>\n",
       "      <td>5.0</td>\n",
       "      <td>8.0</td>\n",
       "      <td>6.0</td>\n",
       "      <td>11.0</td>\n",
       "      <td>21.0</td>\n",
       "      <td>15.0</td>\n",
       "      <td>22.0</td>\n",
       "      <td>23.0</td>\n",
       "      <td>20.0</td>\n",
       "      <td>17.0</td>\n",
       "      <td>13.0</td>\n",
       "      <td>11.0</td>\n",
       "      <td>7.0</td>\n",
       "      <td>14.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>13</th>\n",
       "      <td>35.0</td>\n",
       "      <td>33.0</td>\n",
       "      <td>43.0</td>\n",
       "      <td>22.0</td>\n",
       "      <td>23.0</td>\n",
       "      <td>40.0</td>\n",
       "      <td>29.0</td>\n",
       "      <td>29.0</td>\n",
       "      <td>43.0</td>\n",
       "      <td>31.0</td>\n",
       "      <td>28.0</td>\n",
       "      <td>20.0</td>\n",
       "      <td>35.0</td>\n",
       "      <td>33.0</td>\n",
       "      <td>31.0</td>\n",
       "      <td>23.0</td>\n",
       "      <td>38.0</td>\n",
       "      <td>25.0</td>\n",
       "      <td>41.0</td>\n",
       "      <td>25.0</td>\n",
       "      <td>22.0</td>\n",
       "      <td>24.0</td>\n",
       "      <td>23.0</td>\n",
       "      <td>23.0</td>\n",
       "      <td>19.0</td>\n",
       "      <td>21.0</td>\n",
       "      <td>20.0</td>\n",
       "      <td>14.0</td>\n",
       "      <td>1.0</td>\n",
       "      <td>16.0</td>\n",
       "      <td>13.0</td>\n",
       "      <td>16.0</td>\n",
       "      <td>14.0</td>\n",
       "      <td>28.0</td>\n",
       "      <td>19.0</td>\n",
       "      <td>9.0</td>\n",
       "      <td>13.0</td>\n",
       "      <td>26.0</td>\n",
       "      <td>14.0</td>\n",
       "      <td>14.0</td>\n",
       "      <td>13.0</td>\n",
       "      <td>12.0</td>\n",
       "      <td>0.0</td>\n",
       "      <td>6.0</td>\n",
       "      <td>18.0</td>\n",
       "      <td>13.0</td>\n",
       "      <td>29.0</td>\n",
       "      <td>11.0</td>\n",
       "      <td>4.0</td>\n",
       "      <td>27.0</td>\n",
       "      <td>...</td>\n",
       "      <td>15.0</td>\n",
       "      <td>19.0</td>\n",
       "      <td>18.0</td>\n",
       "      <td>14.0</td>\n",
       "      <td>14.0</td>\n",
       "      <td>23.0</td>\n",
       "      <td>12.0</td>\n",
       "      <td>11.0</td>\n",
       "      <td>18.0</td>\n",
       "      <td>10.0</td>\n",
       "      <td>20.0</td>\n",
       "      <td>10.0</td>\n",
       "      <td>11.0</td>\n",
       "      <td>9.0</td>\n",
       "      <td>13.0</td>\n",
       "      <td>13.0</td>\n",
       "      <td>13.0</td>\n",
       "      <td>12.0</td>\n",
       "      <td>17.0</td>\n",
       "      <td>21.0</td>\n",
       "      <td>4.0</td>\n",
       "      <td>8.0</td>\n",
       "      <td>8.0</td>\n",
       "      <td>10.0</td>\n",
       "      <td>14.0</td>\n",
       "      <td>6.0</td>\n",
       "      <td>5.0</td>\n",
       "      <td>1.0</td>\n",
       "      <td>2.0</td>\n",
       "      <td>4.0</td>\n",
       "      <td>8.0</td>\n",
       "      <td>15.0</td>\n",
       "      <td>6.0</td>\n",
       "      <td>9.0</td>\n",
       "      <td>7.0</td>\n",
       "      <td>10.0</td>\n",
       "      <td>5.0</td>\n",
       "      <td>8.0</td>\n",
       "      <td>6.0</td>\n",
       "      <td>10.0</td>\n",
       "      <td>18.0</td>\n",
       "      <td>14.0</td>\n",
       "      <td>21.0</td>\n",
       "      <td>22.0</td>\n",
       "      <td>18.0</td>\n",
       "      <td>17.0</td>\n",
       "      <td>13.0</td>\n",
       "      <td>10.0</td>\n",
       "      <td>6.0</td>\n",
       "      <td>12.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>14</th>\n",
       "      <td>35.0</td>\n",
       "      <td>33.0</td>\n",
       "      <td>37.0</td>\n",
       "      <td>21.0</td>\n",
       "      <td>23.0</td>\n",
       "      <td>36.0</td>\n",
       "      <td>28.0</td>\n",
       "      <td>27.0</td>\n",
       "      <td>43.0</td>\n",
       "      <td>27.0</td>\n",
       "      <td>24.0</td>\n",
       "      <td>19.0</td>\n",
       "      <td>34.0</td>\n",
       "      <td>28.0</td>\n",
       "      <td>28.0</td>\n",
       "      <td>18.0</td>\n",
       "      <td>25.0</td>\n",
       "      <td>24.0</td>\n",
       "      <td>36.0</td>\n",
       "      <td>23.0</td>\n",
       "      <td>20.0</td>\n",
       "      <td>24.0</td>\n",
       "      <td>22.0</td>\n",
       "      <td>21.0</td>\n",
       "      <td>18.0</td>\n",
       "      <td>20.0</td>\n",
       "      <td>18.0</td>\n",
       "      <td>12.0</td>\n",
       "      <td>1.0</td>\n",
       "      <td>15.0</td>\n",
       "      <td>12.0</td>\n",
       "      <td>12.0</td>\n",
       "      <td>14.0</td>\n",
       "      <td>24.0</td>\n",
       "      <td>18.0</td>\n",
       "      <td>9.0</td>\n",
       "      <td>11.0</td>\n",
       "      <td>21.0</td>\n",
       "      <td>14.0</td>\n",
       "      <td>11.0</td>\n",
       "      <td>11.0</td>\n",
       "      <td>8.0</td>\n",
       "      <td>0.0</td>\n",
       "      <td>5.0</td>\n",
       "      <td>11.0</td>\n",
       "      <td>11.0</td>\n",
       "      <td>28.0</td>\n",
       "      <td>9.0</td>\n",
       "      <td>4.0</td>\n",
       "      <td>21.0</td>\n",
       "      <td>...</td>\n",
       "      <td>15.0</td>\n",
       "      <td>19.0</td>\n",
       "      <td>17.0</td>\n",
       "      <td>14.0</td>\n",
       "      <td>13.0</td>\n",
       "      <td>23.0</td>\n",
       "      <td>12.0</td>\n",
       "      <td>10.0</td>\n",
       "      <td>15.0</td>\n",
       "      <td>9.0</td>\n",
       "      <td>17.0</td>\n",
       "      <td>10.0</td>\n",
       "      <td>10.0</td>\n",
       "      <td>9.0</td>\n",
       "      <td>11.0</td>\n",
       "      <td>13.0</td>\n",
       "      <td>13.0</td>\n",
       "      <td>12.0</td>\n",
       "      <td>16.0</td>\n",
       "      <td>18.0</td>\n",
       "      <td>3.0</td>\n",
       "      <td>7.0</td>\n",
       "      <td>8.0</td>\n",
       "      <td>7.0</td>\n",
       "      <td>14.0</td>\n",
       "      <td>5.0</td>\n",
       "      <td>3.0</td>\n",
       "      <td>1.0</td>\n",
       "      <td>2.0</td>\n",
       "      <td>4.0</td>\n",
       "      <td>7.0</td>\n",
       "      <td>13.0</td>\n",
       "      <td>4.0</td>\n",
       "      <td>9.0</td>\n",
       "      <td>7.0</td>\n",
       "      <td>9.0</td>\n",
       "      <td>3.0</td>\n",
       "      <td>7.0</td>\n",
       "      <td>5.0</td>\n",
       "      <td>8.0</td>\n",
       "      <td>18.0</td>\n",
       "      <td>13.0</td>\n",
       "      <td>19.0</td>\n",
       "      <td>21.0</td>\n",
       "      <td>18.0</td>\n",
       "      <td>17.0</td>\n",
       "      <td>10.0</td>\n",
       "      <td>9.0</td>\n",
       "      <td>6.0</td>\n",
       "      <td>11.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>15</th>\n",
       "      <td>33.0</td>\n",
       "      <td>30.0</td>\n",
       "      <td>31.0</td>\n",
       "      <td>21.0</td>\n",
       "      <td>22.0</td>\n",
       "      <td>34.0</td>\n",
       "      <td>24.0</td>\n",
       "      <td>26.0</td>\n",
       "      <td>38.0</td>\n",
       "      <td>25.0</td>\n",
       "      <td>23.0</td>\n",
       "      <td>19.0</td>\n",
       "      <td>31.0</td>\n",
       "      <td>27.0</td>\n",
       "      <td>27.0</td>\n",
       "      <td>13.0</td>\n",
       "      <td>25.0</td>\n",
       "      <td>20.0</td>\n",
       "      <td>29.0</td>\n",
       "      <td>18.0</td>\n",
       "      <td>18.0</td>\n",
       "      <td>24.0</td>\n",
       "      <td>20.0</td>\n",
       "      <td>21.0</td>\n",
       "      <td>18.0</td>\n",
       "      <td>15.0</td>\n",
       "      <td>18.0</td>\n",
       "      <td>10.0</td>\n",
       "      <td>1.0</td>\n",
       "      <td>15.0</td>\n",
       "      <td>11.0</td>\n",
       "      <td>11.0</td>\n",
       "      <td>14.0</td>\n",
       "      <td>22.0</td>\n",
       "      <td>15.0</td>\n",
       "      <td>7.0</td>\n",
       "      <td>10.0</td>\n",
       "      <td>19.0</td>\n",
       "      <td>13.0</td>\n",
       "      <td>10.0</td>\n",
       "      <td>11.0</td>\n",
       "      <td>8.0</td>\n",
       "      <td>0.0</td>\n",
       "      <td>5.0</td>\n",
       "      <td>9.0</td>\n",
       "      <td>10.0</td>\n",
       "      <td>23.0</td>\n",
       "      <td>9.0</td>\n",
       "      <td>3.0</td>\n",
       "      <td>20.0</td>\n",
       "      <td>...</td>\n",
       "      <td>13.0</td>\n",
       "      <td>18.0</td>\n",
       "      <td>12.0</td>\n",
       "      <td>13.0</td>\n",
       "      <td>12.0</td>\n",
       "      <td>19.0</td>\n",
       "      <td>12.0</td>\n",
       "      <td>9.0</td>\n",
       "      <td>14.0</td>\n",
       "      <td>9.0</td>\n",
       "      <td>16.0</td>\n",
       "      <td>9.0</td>\n",
       "      <td>6.0</td>\n",
       "      <td>9.0</td>\n",
       "      <td>11.0</td>\n",
       "      <td>11.0</td>\n",
       "      <td>12.0</td>\n",
       "      <td>10.0</td>\n",
       "      <td>13.0</td>\n",
       "      <td>14.0</td>\n",
       "      <td>3.0</td>\n",
       "      <td>6.0</td>\n",
       "      <td>7.0</td>\n",
       "      <td>6.0</td>\n",
       "      <td>10.0</td>\n",
       "      <td>3.0</td>\n",
       "      <td>3.0</td>\n",
       "      <td>1.0</td>\n",
       "      <td>2.0</td>\n",
       "      <td>3.0</td>\n",
       "      <td>7.0</td>\n",
       "      <td>11.0</td>\n",
       "      <td>2.0</td>\n",
       "      <td>8.0</td>\n",
       "      <td>6.0</td>\n",
       "      <td>7.0</td>\n",
       "      <td>2.0</td>\n",
       "      <td>6.0</td>\n",
       "      <td>4.0</td>\n",
       "      <td>8.0</td>\n",
       "      <td>16.0</td>\n",
       "      <td>13.0</td>\n",
       "      <td>19.0</td>\n",
       "      <td>19.0</td>\n",
       "      <td>17.0</td>\n",
       "      <td>16.0</td>\n",
       "      <td>10.0</td>\n",
       "      <td>8.0</td>\n",
       "      <td>6.0</td>\n",
       "      <td>9.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>16</th>\n",
       "      <td>30.0</td>\n",
       "      <td>30.0</td>\n",
       "      <td>31.0</td>\n",
       "      <td>17.0</td>\n",
       "      <td>22.0</td>\n",
       "      <td>26.0</td>\n",
       "      <td>21.0</td>\n",
       "      <td>24.0</td>\n",
       "      <td>37.0</td>\n",
       "      <td>24.0</td>\n",
       "      <td>23.0</td>\n",
       "      <td>13.0</td>\n",
       "      <td>25.0</td>\n",
       "      <td>22.0</td>\n",
       "      <td>27.0</td>\n",
       "      <td>10.0</td>\n",
       "      <td>24.0</td>\n",
       "      <td>19.0</td>\n",
       "      <td>27.0</td>\n",
       "      <td>17.0</td>\n",
       "      <td>18.0</td>\n",
       "      <td>23.0</td>\n",
       "      <td>20.0</td>\n",
       "      <td>18.0</td>\n",
       "      <td>11.0</td>\n",
       "      <td>15.0</td>\n",
       "      <td>12.0</td>\n",
       "      <td>10.0</td>\n",
       "      <td>0.0</td>\n",
       "      <td>15.0</td>\n",
       "      <td>10.0</td>\n",
       "      <td>11.0</td>\n",
       "      <td>12.0</td>\n",
       "      <td>18.0</td>\n",
       "      <td>14.0</td>\n",
       "      <td>7.0</td>\n",
       "      <td>10.0</td>\n",
       "      <td>18.0</td>\n",
       "      <td>12.0</td>\n",
       "      <td>9.0</td>\n",
       "      <td>9.0</td>\n",
       "      <td>8.0</td>\n",
       "      <td>0.0</td>\n",
       "      <td>5.0</td>\n",
       "      <td>8.0</td>\n",
       "      <td>8.0</td>\n",
       "      <td>23.0</td>\n",
       "      <td>5.0</td>\n",
       "      <td>3.0</td>\n",
       "      <td>17.0</td>\n",
       "      <td>...</td>\n",
       "      <td>12.0</td>\n",
       "      <td>18.0</td>\n",
       "      <td>12.0</td>\n",
       "      <td>13.0</td>\n",
       "      <td>12.0</td>\n",
       "      <td>17.0</td>\n",
       "      <td>7.0</td>\n",
       "      <td>9.0</td>\n",
       "      <td>14.0</td>\n",
       "      <td>8.0</td>\n",
       "      <td>14.0</td>\n",
       "      <td>9.0</td>\n",
       "      <td>6.0</td>\n",
       "      <td>8.0</td>\n",
       "      <td>10.0</td>\n",
       "      <td>9.0</td>\n",
       "      <td>11.0</td>\n",
       "      <td>10.0</td>\n",
       "      <td>12.0</td>\n",
       "      <td>13.0</td>\n",
       "      <td>3.0</td>\n",
       "      <td>5.0</td>\n",
       "      <td>7.0</td>\n",
       "      <td>6.0</td>\n",
       "      <td>9.0</td>\n",
       "      <td>3.0</td>\n",
       "      <td>3.0</td>\n",
       "      <td>1.0</td>\n",
       "      <td>2.0</td>\n",
       "      <td>3.0</td>\n",
       "      <td>6.0</td>\n",
       "      <td>9.0</td>\n",
       "      <td>2.0</td>\n",
       "      <td>7.0</td>\n",
       "      <td>6.0</td>\n",
       "      <td>6.0</td>\n",
       "      <td>2.0</td>\n",
       "      <td>6.0</td>\n",
       "      <td>3.0</td>\n",
       "      <td>7.0</td>\n",
       "      <td>16.0</td>\n",
       "      <td>12.0</td>\n",
       "      <td>17.0</td>\n",
       "      <td>18.0</td>\n",
       "      <td>16.0</td>\n",
       "      <td>16.0</td>\n",
       "      <td>10.0</td>\n",
       "      <td>7.0</td>\n",
       "      <td>4.0</td>\n",
       "      <td>9.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>17</th>\n",
       "      <td>24.0</td>\n",
       "      <td>28.0</td>\n",
       "      <td>28.0</td>\n",
       "      <td>17.0</td>\n",
       "      <td>21.0</td>\n",
       "      <td>21.0</td>\n",
       "      <td>20.0</td>\n",
       "      <td>20.0</td>\n",
       "      <td>32.0</td>\n",
       "      <td>16.0</td>\n",
       "      <td>20.0</td>\n",
       "      <td>11.0</td>\n",
       "      <td>19.0</td>\n",
       "      <td>22.0</td>\n",
       "      <td>24.0</td>\n",
       "      <td>10.0</td>\n",
       "      <td>18.0</td>\n",
       "      <td>17.0</td>\n",
       "      <td>18.0</td>\n",
       "      <td>15.0</td>\n",
       "      <td>13.0</td>\n",
       "      <td>21.0</td>\n",
       "      <td>18.0</td>\n",
       "      <td>15.0</td>\n",
       "      <td>10.0</td>\n",
       "      <td>12.0</td>\n",
       "      <td>11.0</td>\n",
       "      <td>7.0</td>\n",
       "      <td>0.0</td>\n",
       "      <td>13.0</td>\n",
       "      <td>10.0</td>\n",
       "      <td>8.0</td>\n",
       "      <td>11.0</td>\n",
       "      <td>14.0</td>\n",
       "      <td>11.0</td>\n",
       "      <td>6.0</td>\n",
       "      <td>8.0</td>\n",
       "      <td>17.0</td>\n",
       "      <td>11.0</td>\n",
       "      <td>9.0</td>\n",
       "      <td>7.0</td>\n",
       "      <td>6.0</td>\n",
       "      <td>0.0</td>\n",
       "      <td>4.0</td>\n",
       "      <td>7.0</td>\n",
       "      <td>7.0</td>\n",
       "      <td>17.0</td>\n",
       "      <td>4.0</td>\n",
       "      <td>1.0</td>\n",
       "      <td>17.0</td>\n",
       "      <td>...</td>\n",
       "      <td>10.0</td>\n",
       "      <td>16.0</td>\n",
       "      <td>11.0</td>\n",
       "      <td>13.0</td>\n",
       "      <td>10.0</td>\n",
       "      <td>10.0</td>\n",
       "      <td>6.0</td>\n",
       "      <td>5.0</td>\n",
       "      <td>13.0</td>\n",
       "      <td>8.0</td>\n",
       "      <td>12.0</td>\n",
       "      <td>8.0</td>\n",
       "      <td>5.0</td>\n",
       "      <td>7.0</td>\n",
       "      <td>10.0</td>\n",
       "      <td>6.0</td>\n",
       "      <td>11.0</td>\n",
       "      <td>9.0</td>\n",
       "      <td>11.0</td>\n",
       "      <td>12.0</td>\n",
       "      <td>2.0</td>\n",
       "      <td>3.0</td>\n",
       "      <td>5.0</td>\n",
       "      <td>4.0</td>\n",
       "      <td>5.0</td>\n",
       "      <td>3.0</td>\n",
       "      <td>3.0</td>\n",
       "      <td>1.0</td>\n",
       "      <td>2.0</td>\n",
       "      <td>3.0</td>\n",
       "      <td>3.0</td>\n",
       "      <td>9.0</td>\n",
       "      <td>2.0</td>\n",
       "      <td>6.0</td>\n",
       "      <td>5.0</td>\n",
       "      <td>6.0</td>\n",
       "      <td>2.0</td>\n",
       "      <td>5.0</td>\n",
       "      <td>3.0</td>\n",
       "      <td>5.0</td>\n",
       "      <td>15.0</td>\n",
       "      <td>10.0</td>\n",
       "      <td>16.0</td>\n",
       "      <td>15.0</td>\n",
       "      <td>15.0</td>\n",
       "      <td>15.0</td>\n",
       "      <td>9.0</td>\n",
       "      <td>6.0</td>\n",
       "      <td>4.0</td>\n",
       "      <td>8.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>18</th>\n",
       "      <td>14.0</td>\n",
       "      <td>17.0</td>\n",
       "      <td>27.0</td>\n",
       "      <td>16.0</td>\n",
       "      <td>18.0</td>\n",
       "      <td>14.0</td>\n",
       "      <td>16.0</td>\n",
       "      <td>18.0</td>\n",
       "      <td>26.0</td>\n",
       "      <td>12.0</td>\n",
       "      <td>18.0</td>\n",
       "      <td>10.0</td>\n",
       "      <td>18.0</td>\n",
       "      <td>20.0</td>\n",
       "      <td>22.0</td>\n",
       "      <td>6.0</td>\n",
       "      <td>9.0</td>\n",
       "      <td>15.0</td>\n",
       "      <td>17.0</td>\n",
       "      <td>14.0</td>\n",
       "      <td>12.0</td>\n",
       "      <td>20.0</td>\n",
       "      <td>11.0</td>\n",
       "      <td>11.0</td>\n",
       "      <td>9.0</td>\n",
       "      <td>12.0</td>\n",
       "      <td>11.0</td>\n",
       "      <td>4.0</td>\n",
       "      <td>0.0</td>\n",
       "      <td>13.0</td>\n",
       "      <td>8.0</td>\n",
       "      <td>6.0</td>\n",
       "      <td>8.0</td>\n",
       "      <td>8.0</td>\n",
       "      <td>10.0</td>\n",
       "      <td>6.0</td>\n",
       "      <td>6.0</td>\n",
       "      <td>16.0</td>\n",
       "      <td>11.0</td>\n",
       "      <td>8.0</td>\n",
       "      <td>3.0</td>\n",
       "      <td>5.0</td>\n",
       "      <td>0.0</td>\n",
       "      <td>4.0</td>\n",
       "      <td>6.0</td>\n",
       "      <td>6.0</td>\n",
       "      <td>17.0</td>\n",
       "      <td>3.0</td>\n",
       "      <td>1.0</td>\n",
       "      <td>12.0</td>\n",
       "      <td>...</td>\n",
       "      <td>10.0</td>\n",
       "      <td>15.0</td>\n",
       "      <td>10.0</td>\n",
       "      <td>12.0</td>\n",
       "      <td>8.0</td>\n",
       "      <td>8.0</td>\n",
       "      <td>6.0</td>\n",
       "      <td>3.0</td>\n",
       "      <td>13.0</td>\n",
       "      <td>5.0</td>\n",
       "      <td>12.0</td>\n",
       "      <td>7.0</td>\n",
       "      <td>4.0</td>\n",
       "      <td>6.0</td>\n",
       "      <td>8.0</td>\n",
       "      <td>5.0</td>\n",
       "      <td>10.0</td>\n",
       "      <td>9.0</td>\n",
       "      <td>10.0</td>\n",
       "      <td>11.0</td>\n",
       "      <td>2.0</td>\n",
       "      <td>2.0</td>\n",
       "      <td>3.0</td>\n",
       "      <td>3.0</td>\n",
       "      <td>5.0</td>\n",
       "      <td>2.0</td>\n",
       "      <td>2.0</td>\n",
       "      <td>0.0</td>\n",
       "      <td>1.0</td>\n",
       "      <td>2.0</td>\n",
       "      <td>2.0</td>\n",
       "      <td>8.0</td>\n",
       "      <td>1.0</td>\n",
       "      <td>5.0</td>\n",
       "      <td>5.0</td>\n",
       "      <td>6.0</td>\n",
       "      <td>1.0</td>\n",
       "      <td>5.0</td>\n",
       "      <td>3.0</td>\n",
       "      <td>5.0</td>\n",
       "      <td>12.0</td>\n",
       "      <td>7.0</td>\n",
       "      <td>15.0</td>\n",
       "      <td>14.0</td>\n",
       "      <td>14.0</td>\n",
       "      <td>14.0</td>\n",
       "      <td>9.0</td>\n",
       "      <td>5.0</td>\n",
       "      <td>4.0</td>\n",
       "      <td>6.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>19</th>\n",
       "      <td>14.0</td>\n",
       "      <td>9.0</td>\n",
       "      <td>12.0</td>\n",
       "      <td>8.0</td>\n",
       "      <td>14.0</td>\n",
       "      <td>14.0</td>\n",
       "      <td>11.0</td>\n",
       "      <td>11.0</td>\n",
       "      <td>18.0</td>\n",
       "      <td>9.0</td>\n",
       "      <td>17.0</td>\n",
       "      <td>8.0</td>\n",
       "      <td>12.0</td>\n",
       "      <td>17.0</td>\n",
       "      <td>22.0</td>\n",
       "      <td>6.0</td>\n",
       "      <td>9.0</td>\n",
       "      <td>13.0</td>\n",
       "      <td>14.0</td>\n",
       "      <td>10.0</td>\n",
       "      <td>8.0</td>\n",
       "      <td>11.0</td>\n",
       "      <td>11.0</td>\n",
       "      <td>11.0</td>\n",
       "      <td>4.0</td>\n",
       "      <td>11.0</td>\n",
       "      <td>5.0</td>\n",
       "      <td>3.0</td>\n",
       "      <td>0.0</td>\n",
       "      <td>5.0</td>\n",
       "      <td>8.0</td>\n",
       "      <td>6.0</td>\n",
       "      <td>8.0</td>\n",
       "      <td>8.0</td>\n",
       "      <td>10.0</td>\n",
       "      <td>5.0</td>\n",
       "      <td>4.0</td>\n",
       "      <td>16.0</td>\n",
       "      <td>9.0</td>\n",
       "      <td>4.0</td>\n",
       "      <td>3.0</td>\n",
       "      <td>5.0</td>\n",
       "      <td>0.0</td>\n",
       "      <td>4.0</td>\n",
       "      <td>4.0</td>\n",
       "      <td>5.0</td>\n",
       "      <td>9.0</td>\n",
       "      <td>3.0</td>\n",
       "      <td>1.0</td>\n",
       "      <td>11.0</td>\n",
       "      <td>...</td>\n",
       "      <td>5.0</td>\n",
       "      <td>13.0</td>\n",
       "      <td>10.0</td>\n",
       "      <td>12.0</td>\n",
       "      <td>1.0</td>\n",
       "      <td>8.0</td>\n",
       "      <td>1.0</td>\n",
       "      <td>3.0</td>\n",
       "      <td>13.0</td>\n",
       "      <td>4.0</td>\n",
       "      <td>11.0</td>\n",
       "      <td>5.0</td>\n",
       "      <td>4.0</td>\n",
       "      <td>6.0</td>\n",
       "      <td>8.0</td>\n",
       "      <td>5.0</td>\n",
       "      <td>6.0</td>\n",
       "      <td>6.0</td>\n",
       "      <td>10.0</td>\n",
       "      <td>7.0</td>\n",
       "      <td>2.0</td>\n",
       "      <td>0.0</td>\n",
       "      <td>2.0</td>\n",
       "      <td>2.0</td>\n",
       "      <td>4.0</td>\n",
       "      <td>1.0</td>\n",
       "      <td>2.0</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0.0</td>\n",
       "      <td>1.0</td>\n",
       "      <td>2.0</td>\n",
       "      <td>7.0</td>\n",
       "      <td>0.0</td>\n",
       "      <td>2.0</td>\n",
       "      <td>5.0</td>\n",
       "      <td>5.0</td>\n",
       "      <td>1.0</td>\n",
       "      <td>4.0</td>\n",
       "      <td>2.0</td>\n",
       "      <td>5.0</td>\n",
       "      <td>12.0</td>\n",
       "      <td>4.0</td>\n",
       "      <td>10.0</td>\n",
       "      <td>12.0</td>\n",
       "      <td>12.0</td>\n",
       "      <td>14.0</td>\n",
       "      <td>4.0</td>\n",
       "      <td>3.0</td>\n",
       "      <td>4.0</td>\n",
       "      <td>5.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>20</th>\n",
       "      <td>6.0</td>\n",
       "      <td>8.0</td>\n",
       "      <td>9.0</td>\n",
       "      <td>6.0</td>\n",
       "      <td>12.0</td>\n",
       "      <td>8.0</td>\n",
       "      <td>7.0</td>\n",
       "      <td>5.0</td>\n",
       "      <td>13.0</td>\n",
       "      <td>5.0</td>\n",
       "      <td>15.0</td>\n",
       "      <td>7.0</td>\n",
       "      <td>9.0</td>\n",
       "      <td>12.0</td>\n",
       "      <td>16.0</td>\n",
       "      <td>4.0</td>\n",
       "      <td>3.0</td>\n",
       "      <td>12.0</td>\n",
       "      <td>3.0</td>\n",
       "      <td>7.0</td>\n",
       "      <td>8.0</td>\n",
       "      <td>9.0</td>\n",
       "      <td>7.0</td>\n",
       "      <td>8.0</td>\n",
       "      <td>3.0</td>\n",
       "      <td>8.0</td>\n",
       "      <td>3.0</td>\n",
       "      <td>2.0</td>\n",
       "      <td>0.0</td>\n",
       "      <td>3.0</td>\n",
       "      <td>4.0</td>\n",
       "      <td>4.0</td>\n",
       "      <td>5.0</td>\n",
       "      <td>0.0</td>\n",
       "      <td>8.0</td>\n",
       "      <td>3.0</td>\n",
       "      <td>4.0</td>\n",
       "      <td>3.0</td>\n",
       "      <td>5.0</td>\n",
       "      <td>3.0</td>\n",
       "      <td>2.0</td>\n",
       "      <td>1.0</td>\n",
       "      <td>0.0</td>\n",
       "      <td>2.0</td>\n",
       "      <td>0.0</td>\n",
       "      <td>3.0</td>\n",
       "      <td>4.0</td>\n",
       "      <td>2.0</td>\n",
       "      <td>0.0</td>\n",
       "      <td>3.0</td>\n",
       "      <td>...</td>\n",
       "      <td>4.0</td>\n",
       "      <td>7.0</td>\n",
       "      <td>6.0</td>\n",
       "      <td>8.0</td>\n",
       "      <td>0.0</td>\n",
       "      <td>4.0</td>\n",
       "      <td>1.0</td>\n",
       "      <td>3.0</td>\n",
       "      <td>9.0</td>\n",
       "      <td>2.0</td>\n",
       "      <td>8.0</td>\n",
       "      <td>3.0</td>\n",
       "      <td>2.0</td>\n",
       "      <td>1.0</td>\n",
       "      <td>3.0</td>\n",
       "      <td>3.0</td>\n",
       "      <td>6.0</td>\n",
       "      <td>1.0</td>\n",
       "      <td>9.0</td>\n",
       "      <td>7.0</td>\n",
       "      <td>1.0</td>\n",
       "      <td>0.0</td>\n",
       "      <td>1.0</td>\n",
       "      <td>0.0</td>\n",
       "      <td>4.0</td>\n",
       "      <td>0.0</td>\n",
       "      <td>1.0</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0.0</td>\n",
       "      <td>1.0</td>\n",
       "      <td>5.0</td>\n",
       "      <td>0.0</td>\n",
       "      <td>2.0</td>\n",
       "      <td>4.0</td>\n",
       "      <td>4.0</td>\n",
       "      <td>1.0</td>\n",
       "      <td>3.0</td>\n",
       "      <td>1.0</td>\n",
       "      <td>3.0</td>\n",
       "      <td>5.0</td>\n",
       "      <td>0.0</td>\n",
       "      <td>2.0</td>\n",
       "      <td>1.0</td>\n",
       "      <td>9.0</td>\n",
       "      <td>9.0</td>\n",
       "      <td>3.0</td>\n",
       "      <td>3.0</td>\n",
       "      <td>1.0</td>\n",
       "      <td>4.0</td>\n",
       "    </tr>\n",
       "  </tbody>\n",
       "</table>\n",
       "<p>21 rows × 670 columns</p>\n",
       "</div>"
      ],
      "text/plain": [
       "         0         1         2         3         4         5         6    \\\n",
       "0   436363.0  437064.0  436469.0  437013.0  436983.0  436644.0  436768.0   \n",
       "1      259.0     221.0     147.0     141.0     179.0     175.0     145.0   \n",
       "2      147.0      91.0      87.0      99.0     160.0     161.0     128.0   \n",
       "3      100.0      82.0      79.0      80.0      80.0      96.0     120.0   \n",
       "4       97.0      76.0      78.0      48.0      62.0      90.0      85.0   \n",
       "5       75.0      64.0      76.0      42.0      57.0      82.0      82.0   \n",
       "6       71.0      58.0      68.0      41.0      40.0      79.0      70.0   \n",
       "7       68.0      58.0      63.0      39.0      36.0      78.0      69.0   \n",
       "8       66.0      56.0      61.0      30.0      34.0      66.0      67.0   \n",
       "9       53.0      46.0      55.0      29.0      31.0      58.0      40.0   \n",
       "10      52.0      42.0      55.0      28.0      29.0      52.0      35.0   \n",
       "11      41.0      40.0      48.0      25.0      29.0      48.0      32.0   \n",
       "12      38.0      38.0      45.0      22.0      23.0      45.0      30.0   \n",
       "13      35.0      33.0      43.0      22.0      23.0      40.0      29.0   \n",
       "14      35.0      33.0      37.0      21.0      23.0      36.0      28.0   \n",
       "15      33.0      30.0      31.0      21.0      22.0      34.0      24.0   \n",
       "16      30.0      30.0      31.0      17.0      22.0      26.0      21.0   \n",
       "17      24.0      28.0      28.0      17.0      21.0      21.0      20.0   \n",
       "18      14.0      17.0      27.0      16.0      18.0      14.0      16.0   \n",
       "19      14.0       9.0      12.0       8.0      14.0      14.0      11.0   \n",
       "20       6.0       8.0       9.0       6.0      12.0       8.0       7.0   \n",
       "\n",
       "         7         8         9         10        11        12        13   \\\n",
       "0   436802.0  436286.0  437163.0  436833.0  436958.0  436212.0  436937.0   \n",
       "1      154.0     293.0     127.0     114.0     188.0     218.0     147.0   \n",
       "2      109.0     148.0     126.0     102.0     116.0     142.0     110.0   \n",
       "3      104.0     141.0      93.0      89.0     114.0     136.0      74.0   \n",
       "4       70.0     109.0      62.0      78.0      68.0     111.0      62.0   \n",
       "5       69.0      96.0      54.0      72.0      66.0     100.0      61.0   \n",
       "6       69.0      82.0      53.0      70.0      59.0      85.0      54.0   \n",
       "7       68.0      77.0      42.0      60.0      56.0      67.0      52.0   \n",
       "8       66.0      72.0      41.0      50.0      46.0      46.0      51.0   \n",
       "9       45.0      69.0      39.0      49.0      36.0      38.0      50.0   \n",
       "10      32.0      61.0      37.0      46.0      27.0      37.0      43.0   \n",
       "11      30.0      59.0      34.0      41.0      26.0      36.0      41.0   \n",
       "12      30.0      54.0      33.0      31.0      24.0      36.0      39.0   \n",
       "13      29.0      43.0      31.0      28.0      20.0      35.0      33.0   \n",
       "14      27.0      43.0      27.0      24.0      19.0      34.0      28.0   \n",
       "15      26.0      38.0      25.0      23.0      19.0      31.0      27.0   \n",
       "16      24.0      37.0      24.0      23.0      13.0      25.0      22.0   \n",
       "17      20.0      32.0      16.0      20.0      11.0      19.0      22.0   \n",
       "18      18.0      26.0      12.0      18.0      10.0      18.0      20.0   \n",
       "19      11.0      18.0       9.0      17.0       8.0      12.0      17.0   \n",
       "20       5.0      13.0       5.0      15.0       7.0       9.0      12.0   \n",
       "\n",
       "         14        15        16        17        18        19        20   \\\n",
       "0   436565.0  436939.0  436773.0  435964.0  436779.0  436707.0  437358.0   \n",
       "1      299.0     118.0     113.0     389.0     169.0      72.0      61.0   \n",
       "2      152.0      97.0     106.0     129.0     131.0      71.0      59.0   \n",
       "3       97.0      54.0      99.0     109.0      94.0      71.0      56.0   \n",
       "4       89.0      50.0      91.0      77.0      63.0      70.0      54.0   \n",
       "5       75.0      42.0      89.0      67.0      60.0      69.0      53.0   \n",
       "6       52.0      40.0      74.0      67.0      59.0      62.0      47.0   \n",
       "7       47.0      32.0      73.0      60.0      56.0      57.0      44.0   \n",
       "8       43.0      31.0      55.0      50.0      52.0      47.0      41.0   \n",
       "9       43.0      27.0      52.0      46.0      45.0      43.0      33.0   \n",
       "10      38.0      26.0      50.0      34.0      45.0      37.0      27.0   \n",
       "11      37.0      23.0      47.0      29.0      45.0      25.0      24.0   \n",
       "12      34.0      23.0      39.0      27.0      44.0      25.0      23.0   \n",
       "13      31.0      23.0      38.0      25.0      41.0      25.0      22.0   \n",
       "14      28.0      18.0      25.0      24.0      36.0      23.0      20.0   \n",
       "15      27.0      13.0      25.0      20.0      29.0      18.0      18.0   \n",
       "16      27.0      10.0      24.0      19.0      27.0      17.0      18.0   \n",
       "17      24.0      10.0      18.0      17.0      18.0      15.0      13.0   \n",
       "18      22.0       6.0       9.0      15.0      17.0      14.0      12.0   \n",
       "19      22.0       6.0       9.0      13.0      14.0      10.0       8.0   \n",
       "20      16.0       4.0       3.0      12.0       3.0       7.0       8.0   \n",
       "\n",
       "         21        22        23        24        25        26        27   \\\n",
       "0   436968.0  437294.0  437148.0  435829.0  436756.0  437266.0  436274.0   \n",
       "1      122.0      69.0      80.0     501.0     187.0      57.0     196.0   \n",
       "2       71.0      67.0      72.0     330.0      85.0      52.0      85.0   \n",
       "3       68.0      61.0      61.0      99.0      68.0      44.0      54.0   \n",
       "4       51.0      40.0      46.0      57.0      59.0      39.0      51.0   \n",
       "5       47.0      38.0      45.0      51.0      59.0      36.0      38.0   \n",
       "6       43.0      35.0      40.0      45.0      43.0      36.0      29.0   \n",
       "7       40.0      33.0      37.0      31.0      38.0      33.0      29.0   \n",
       "8       39.0      33.0      36.0      30.0      38.0      30.0      24.0   \n",
       "9       31.0      31.0      33.0      28.0      33.0      29.0      19.0   \n",
       "10      27.0      26.0      30.0      26.0      29.0      28.0      19.0   \n",
       "11      27.0      25.0      25.0      22.0      28.0      25.0      15.0   \n",
       "12      26.0      24.0      24.0      19.0      24.0      22.0      14.0   \n",
       "13      24.0      23.0      23.0      19.0      21.0      20.0      14.0   \n",
       "14      24.0      22.0      21.0      18.0      20.0      18.0      12.0   \n",
       "15      24.0      20.0      21.0      18.0      15.0      18.0      10.0   \n",
       "16      23.0      20.0      18.0      11.0      15.0      12.0      10.0   \n",
       "17      21.0      18.0      15.0      10.0      12.0      11.0       7.0   \n",
       "18      20.0      11.0      11.0       9.0      12.0      11.0       4.0   \n",
       "19      11.0      11.0      11.0       4.0      11.0       5.0       3.0   \n",
       "20       9.0       7.0       8.0       3.0       8.0       3.0       2.0   \n",
       "\n",
       "         28        29        30        31        32        33        34   \\\n",
       "0   437747.0  436814.0  437155.0  435542.0  436515.0  436466.0  435281.0   \n",
       "1       15.0     125.0      62.0     130.0     100.0     133.0      89.0   \n",
       "2       11.0     107.0      58.0     105.0      63.0     128.0      81.0   \n",
       "3        8.0      81.0      44.0      87.0      49.0      66.0      63.0   \n",
       "4        8.0      70.0      29.0      29.0      46.0      57.0      48.0   \n",
       "5        6.0      65.0      26.0      28.0      44.0      54.0      46.0   \n",
       "6        5.0      50.0      17.0      27.0      44.0      52.0      42.0   \n",
       "7        4.0      46.0      16.0      25.0      41.0      43.0      34.0   \n",
       "8        3.0      45.0      16.0      23.0      31.0      43.0      29.0   \n",
       "9        2.0      31.0      16.0      22.0      28.0      37.0      27.0   \n",
       "10       2.0      29.0      16.0      21.0      22.0      32.0      23.0   \n",
       "11       2.0      26.0      15.0      17.0      18.0      31.0      22.0   \n",
       "12       1.0      20.0      15.0      16.0      14.0      28.0      21.0   \n",
       "13       1.0      16.0      13.0      16.0      14.0      28.0      19.0   \n",
       "14       1.0      15.0      12.0      12.0      14.0      24.0      18.0   \n",
       "15       1.0      15.0      11.0      11.0      14.0      22.0      15.0   \n",
       "16       0.0      15.0      10.0      11.0      12.0      18.0      14.0   \n",
       "17       0.0      13.0      10.0       8.0      11.0      14.0      11.0   \n",
       "18       0.0      13.0       8.0       6.0       8.0       8.0      10.0   \n",
       "19       0.0       5.0       8.0       6.0       8.0       8.0      10.0   \n",
       "20       0.0       3.0       4.0       4.0       5.0       0.0       8.0   \n",
       "\n",
       "         35        36        37        38        39        40        41   \\\n",
       "0   436274.0  436366.0  421341.0  431323.0  433637.0  435091.0  433659.0   \n",
       "1       91.0      60.0     102.0     111.0     104.0     158.0      91.0   \n",
       "2       75.0      53.0      64.0      91.0      89.0      55.0      73.0   \n",
       "3       46.0      48.0      52.0      68.0      77.0      52.0      58.0   \n",
       "4       33.0      27.0      51.0      41.0      65.0      49.0      43.0   \n",
       "5       29.0      25.0      48.0      39.0      56.0      42.0      42.0   \n",
       "6       26.0      23.0      47.0      35.0      33.0      34.0      36.0   \n",
       "7       26.0      18.0      43.0      33.0      32.0      32.0      34.0   \n",
       "8       25.0      16.0      35.0      33.0      27.0      31.0      25.0   \n",
       "9       21.0      16.0      34.0      27.0      20.0      20.0      17.0   \n",
       "10      13.0      15.0      33.0      21.0      17.0      18.0      16.0   \n",
       "11      13.0      14.0      27.0      16.0      16.0      17.0      14.0   \n",
       "12      12.0      14.0      27.0      16.0      16.0      14.0      13.0   \n",
       "13       9.0      13.0      26.0      14.0      14.0      13.0      12.0   \n",
       "14       9.0      11.0      21.0      14.0      11.0      11.0       8.0   \n",
       "15       7.0      10.0      19.0      13.0      10.0      11.0       8.0   \n",
       "16       7.0      10.0      18.0      12.0       9.0       9.0       8.0   \n",
       "17       6.0       8.0      17.0      11.0       9.0       7.0       6.0   \n",
       "18       6.0       6.0      16.0      11.0       8.0       3.0       5.0   \n",
       "19       5.0       4.0      16.0       9.0       4.0       3.0       5.0   \n",
       "20       3.0       4.0       3.0       5.0       3.0       2.0       1.0   \n",
       "\n",
       "         42        43        44        45        46        47        48   \\\n",
       "0   403742.0  415302.0  406736.0  413296.0  418489.0  350903.0  409586.0   \n",
       "1      142.0     357.0     149.0     201.0     105.0      81.0      59.0   \n",
       "2      110.0     119.0     110.0      95.0      98.0      64.0      57.0   \n",
       "3       66.0      52.0      54.0      90.0      97.0      60.0      53.0   \n",
       "4       57.0      44.0      50.0      76.0      95.0      45.0      40.0   \n",
       "5       19.0      42.0      48.0      55.0      90.0      33.0      11.0   \n",
       "6        3.0      29.0      43.0      35.0      79.0      31.0      11.0   \n",
       "7        1.0      26.0      42.0      35.0      71.0      24.0       9.0   \n",
       "8        1.0      19.0      32.0      27.0      70.0      24.0       9.0   \n",
       "9        1.0      17.0      23.0      25.0      54.0      19.0       9.0   \n",
       "10       0.0      11.0      23.0      16.0      51.0      18.0       8.0   \n",
       "11       0.0      11.0      20.0      14.0      36.0      15.0       6.0   \n",
       "12       0.0       7.0      20.0      13.0      35.0      13.0       4.0   \n",
       "13       0.0       6.0      18.0      13.0      29.0      11.0       4.0   \n",
       "14       0.0       5.0      11.0      11.0      28.0       9.0       4.0   \n",
       "15       0.0       5.0       9.0      10.0      23.0       9.0       3.0   \n",
       "16       0.0       5.0       8.0       8.0      23.0       5.0       3.0   \n",
       "17       0.0       4.0       7.0       7.0      17.0       4.0       1.0   \n",
       "18       0.0       4.0       6.0       6.0      17.0       3.0       1.0   \n",
       "19       0.0       4.0       4.0       5.0       9.0       3.0       1.0   \n",
       "20       0.0       2.0       0.0       3.0       4.0       2.0       0.0   \n",
       "\n",
       "         49   ...       620       621       622       623       624       625  \\\n",
       "0   416254.0  ...  210561.0  209522.0  218191.0  214592.0  217988.0  216522.0   \n",
       "1      134.0  ...     332.0      77.0     362.0      91.0      70.0     133.0   \n",
       "2       86.0  ...     119.0      75.0      52.0      44.0      70.0      62.0   \n",
       "3       79.0  ...      45.0      65.0      42.0      39.0      67.0      57.0   \n",
       "4       79.0  ...      39.0      59.0      39.0      33.0      38.0      53.0   \n",
       "5       69.0  ...      39.0      51.0      38.0      30.0      35.0      37.0   \n",
       "6       44.0  ...      35.0      50.0      38.0      29.0      35.0      35.0   \n",
       "7       39.0  ...      30.0      49.0      36.0      26.0      34.0      35.0   \n",
       "8       38.0  ...      26.0      48.0      33.0      25.0      29.0      34.0   \n",
       "9       34.0  ...      21.0      39.0      23.0      22.0      23.0      28.0   \n",
       "10      34.0  ...      20.0      28.0      22.0      16.0      20.0      28.0   \n",
       "11      32.0  ...      19.0      22.0      20.0      16.0      18.0      25.0   \n",
       "12      29.0  ...      18.0      19.0      19.0      16.0      14.0      23.0   \n",
       "13      27.0  ...      15.0      19.0      18.0      14.0      14.0      23.0   \n",
       "14      21.0  ...      15.0      19.0      17.0      14.0      13.0      23.0   \n",
       "15      20.0  ...      13.0      18.0      12.0      13.0      12.0      19.0   \n",
       "16      17.0  ...      12.0      18.0      12.0      13.0      12.0      17.0   \n",
       "17      17.0  ...      10.0      16.0      11.0      13.0      10.0      10.0   \n",
       "18      12.0  ...      10.0      15.0      10.0      12.0       8.0       8.0   \n",
       "19      11.0  ...       5.0      13.0      10.0      12.0       1.0       8.0   \n",
       "20       3.0  ...       4.0       7.0       6.0       8.0       0.0       4.0   \n",
       "\n",
       "         626       627       628       629       630       631      632  \\\n",
       "0   202751.0  200563.0  197799.0  190110.0  200160.0  129952.0  96157.0   \n",
       "1       87.0      49.0      40.0     139.0      63.0     107.0     84.0   \n",
       "2       48.0      46.0      38.0      89.0      51.0      43.0     72.0   \n",
       "3       37.0      32.0      37.0      47.0      49.0      26.0     21.0   \n",
       "4       31.0      31.0      34.0      41.0      45.0      23.0     20.0   \n",
       "5       30.0      31.0      33.0      39.0      38.0      19.0     19.0   \n",
       "6       29.0      30.0      31.0      38.0      32.0      18.0     19.0   \n",
       "7       28.0      27.0      29.0      29.0      31.0      18.0     16.0   \n",
       "8       24.0      24.0      28.0      21.0      26.0      17.0     16.0   \n",
       "9       22.0      19.0      25.0      20.0      26.0      16.0     15.0   \n",
       "10      21.0      19.0      22.0      14.0      25.0      16.0     14.0   \n",
       "11      16.0      17.0      21.0      13.0      24.0      13.0     14.0   \n",
       "12      13.0      11.0      19.0      11.0      23.0      12.0     11.0   \n",
       "13      12.0      11.0      18.0      10.0      20.0      10.0     11.0   \n",
       "14      12.0      10.0      15.0       9.0      17.0      10.0     10.0   \n",
       "15      12.0       9.0      14.0       9.0      16.0       9.0      6.0   \n",
       "16       7.0       9.0      14.0       8.0      14.0       9.0      6.0   \n",
       "17       6.0       5.0      13.0       8.0      12.0       8.0      5.0   \n",
       "18       6.0       3.0      13.0       5.0      12.0       7.0      4.0   \n",
       "19       1.0       3.0      13.0       4.0      11.0       5.0      4.0   \n",
       "20       1.0       3.0       9.0       2.0       8.0       3.0      2.0   \n",
       "\n",
       "         633       634       635       636       637       638       639  \\\n",
       "0   131661.0  151361.0  166282.0  162639.0  189208.0  209711.0  211814.0   \n",
       "1       66.0     202.0      39.0     121.0     101.0      81.0      96.0   \n",
       "2       54.0      94.0      38.0      50.0      64.0      55.0      95.0   \n",
       "3       47.0      42.0      37.0      42.0      46.0      49.0      80.0   \n",
       "4       45.0      41.0      36.0      40.0      37.0      47.0      64.0   \n",
       "5       36.0      38.0      34.0      38.0      35.0      42.0      48.0   \n",
       "6       25.0      31.0      30.0      34.0      29.0      41.0      44.0   \n",
       "7       18.0      25.0      29.0      27.0      24.0      32.0      35.0   \n",
       "8       17.0      25.0      29.0      23.0      19.0      28.0      34.0   \n",
       "9       15.0      18.0      28.0      18.0      17.0      25.0      28.0   \n",
       "10      15.0      17.0      21.0      16.0      17.0      23.0      27.0   \n",
       "11      12.0      17.0      21.0      14.0      14.0      18.0      27.0   \n",
       "12      11.0      13.0      16.0      13.0      14.0      17.0      24.0   \n",
       "13       9.0      13.0      13.0      13.0      12.0      17.0      21.0   \n",
       "14       9.0      11.0      13.0      13.0      12.0      16.0      18.0   \n",
       "15       9.0      11.0      11.0      12.0      10.0      13.0      14.0   \n",
       "16       8.0      10.0       9.0      11.0      10.0      12.0      13.0   \n",
       "17       7.0      10.0       6.0      11.0       9.0      11.0      12.0   \n",
       "18       6.0       8.0       5.0      10.0       9.0      10.0      11.0   \n",
       "19       6.0       8.0       5.0       6.0       6.0      10.0       7.0   \n",
       "20       1.0       3.0       3.0       6.0       1.0       9.0       7.0   \n",
       "\n",
       "         640       641       642       643       644       645       646  \\\n",
       "0   213893.0  211767.0  215793.0  219257.0  222014.0  222111.0  210061.0   \n",
       "1       30.0      42.0      78.0      54.0      39.0      70.0      23.0   \n",
       "2       30.0      29.0      35.0      48.0      35.0      37.0      22.0   \n",
       "3       19.0      25.0      34.0      46.0      34.0      30.0      15.0   \n",
       "4       16.0      16.0      31.0      35.0      31.0      24.0      12.0   \n",
       "5       12.0      16.0      26.0      34.0      28.0      22.0      11.0   \n",
       "6       12.0      14.0      18.0      34.0      25.0      15.0       9.0   \n",
       "7        7.0      12.0      17.0      28.0      24.0      10.0       8.0   \n",
       "8        6.0      11.0      16.0      23.0      24.0      10.0       7.0   \n",
       "9        6.0       9.0      16.0      18.0      16.0       9.0       6.0   \n",
       "10       5.0       8.0      14.0      15.0      16.0       9.0       6.0   \n",
       "11       5.0       8.0      14.0      13.0      16.0       7.0       6.0   \n",
       "12       5.0       8.0       9.0      13.0      14.0       7.0       6.0   \n",
       "13       4.0       8.0       8.0      10.0      14.0       6.0       5.0   \n",
       "14       3.0       7.0       8.0       7.0      14.0       5.0       3.0   \n",
       "15       3.0       6.0       7.0       6.0      10.0       3.0       3.0   \n",
       "16       3.0       5.0       7.0       6.0       9.0       3.0       3.0   \n",
       "17       2.0       3.0       5.0       4.0       5.0       3.0       3.0   \n",
       "18       2.0       2.0       3.0       3.0       5.0       2.0       2.0   \n",
       "19       2.0       0.0       2.0       2.0       4.0       1.0       2.0   \n",
       "20       1.0       0.0       1.0       0.0       4.0       0.0       1.0   \n",
       "\n",
       "         647       648       649       650       651       652       653  \\\n",
       "0   213342.0  203625.0  223728.0  223085.0  222166.0  222981.0  222238.0   \n",
       "1       27.0      21.0      16.0      29.0      59.0      50.0      50.0   \n",
       "2       24.0      15.0      12.0      24.0      49.0      44.0      40.0   \n",
       "3       19.0      10.0       9.0      20.0      47.0      36.0      32.0   \n",
       "4       13.0       6.0       9.0      19.0      36.0      26.0      30.0   \n",
       "5       12.0       6.0       9.0      19.0      32.0      20.0      23.0   \n",
       "6        9.0       5.0       8.0      15.0      31.0      16.0      22.0   \n",
       "7        5.0       4.0       7.0      14.0      27.0      16.0      21.0   \n",
       "8        4.0       3.0       6.0      14.0      24.0      10.0      17.0   \n",
       "9        4.0       3.0       6.0      13.0      22.0       7.0      16.0   \n",
       "10       2.0       3.0       5.0      12.0      19.0       7.0      14.0   \n",
       "11       2.0       3.0       5.0      12.0      18.0       7.0      12.0   \n",
       "12       2.0       3.0       4.0       9.0      17.0       7.0      12.0   \n",
       "13       1.0       2.0       4.0       8.0      15.0       6.0       9.0   \n",
       "14       1.0       2.0       4.0       7.0      13.0       4.0       9.0   \n",
       "15       1.0       2.0       3.0       7.0      11.0       2.0       8.0   \n",
       "16       1.0       2.0       3.0       6.0       9.0       2.0       7.0   \n",
       "17       1.0       2.0       3.0       3.0       9.0       2.0       6.0   \n",
       "18       0.0       1.0       2.0       2.0       8.0       1.0       5.0   \n",
       "19       0.0       0.0       1.0       2.0       7.0       0.0       2.0   \n",
       "20       0.0       0.0       0.0       1.0       5.0       0.0       2.0   \n",
       "\n",
       "         654       655       656       657       658       659       660  \\\n",
       "0   223187.0  221346.0  223653.0  222840.0  221726.0  223711.0  223564.0   \n",
       "1       38.0      73.0      23.0      25.0      39.0      81.0      55.0   \n",
       "2       37.0      35.0      16.0      23.0      35.0      35.0      54.0   \n",
       "3       34.0      35.0      16.0      18.0      28.0      31.0      43.0   \n",
       "4       26.0      19.0      13.0      17.0      22.0      29.0      42.0   \n",
       "5       22.0      18.0      11.0      16.0      17.0      25.0      36.0   \n",
       "6       22.0      15.0       9.0      16.0      12.0      23.0      34.0   \n",
       "7       18.0      14.0       8.0      10.0      11.0      23.0      29.0   \n",
       "8       15.0      14.0       8.0       9.0      10.0      17.0      29.0   \n",
       "9       13.0      12.0       8.0       9.0       9.0      16.0      26.0   \n",
       "10      11.0      12.0       7.0       9.0       9.0      15.0      25.0   \n",
       "11       9.0      11.0       6.0       9.0       6.0      11.0      23.0   \n",
       "12       8.0      11.0       5.0       8.0       6.0      11.0      21.0   \n",
       "13       7.0      10.0       5.0       8.0       6.0      10.0      18.0   \n",
       "14       7.0       9.0       3.0       7.0       5.0       8.0      18.0   \n",
       "15       6.0       7.0       2.0       6.0       4.0       8.0      16.0   \n",
       "16       6.0       6.0       2.0       6.0       3.0       7.0      16.0   \n",
       "17       5.0       6.0       2.0       5.0       3.0       5.0      15.0   \n",
       "18       5.0       6.0       1.0       5.0       3.0       5.0      12.0   \n",
       "19       5.0       5.0       1.0       4.0       2.0       5.0      12.0   \n",
       "20       4.0       4.0       1.0       3.0       1.0       3.0       5.0   \n",
       "\n",
       "         661       662       663       664       665       666       667  \\\n",
       "0   223476.0  223653.0  223516.0  223699.0  223628.0  223997.0  223552.0   \n",
       "1       70.0      63.0      63.0      64.0      48.0      36.0     166.0   \n",
       "2       42.0      60.0      54.0      57.0      46.0      32.0      46.0   \n",
       "3       41.0      49.0      50.0      46.0      44.0      28.0      38.0   \n",
       "4       38.0      46.0      47.0      46.0      42.0      28.0      33.0   \n",
       "5       31.0      37.0      40.0      44.0      41.0      26.0      25.0   \n",
       "6       22.0      36.0      36.0      29.0      36.0      19.0      22.0   \n",
       "7       21.0      32.0      33.0      27.0      34.0      16.0      21.0   \n",
       "8       18.0      30.0      28.0      25.0      28.0      16.0      16.0   \n",
       "9       16.0      27.0      28.0      24.0      22.0      15.0      13.0   \n",
       "10      16.0      26.0      26.0      24.0      17.0      15.0      12.0   \n",
       "11      16.0      23.0      25.0      24.0      17.0      15.0      12.0   \n",
       "12      15.0      22.0      23.0      20.0      17.0      13.0      11.0   \n",
       "13      14.0      21.0      22.0      18.0      17.0      13.0      10.0   \n",
       "14      13.0      19.0      21.0      18.0      17.0      10.0       9.0   \n",
       "15      13.0      19.0      19.0      17.0      16.0      10.0       8.0   \n",
       "16      12.0      17.0      18.0      16.0      16.0      10.0       7.0   \n",
       "17      10.0      16.0      15.0      15.0      15.0       9.0       6.0   \n",
       "18       7.0      15.0      14.0      14.0      14.0       9.0       5.0   \n",
       "19       4.0      10.0      12.0      12.0      14.0       4.0       3.0   \n",
       "20       0.0       2.0       1.0       9.0       9.0       3.0       3.0   \n",
       "\n",
       "         668       669  \n",
       "0   223703.0  223943.0  \n",
       "1       41.0      51.0  \n",
       "2       33.0      45.0  \n",
       "3       29.0      41.0  \n",
       "4       25.0      33.0  \n",
       "5       24.0      30.0  \n",
       "6       17.0      30.0  \n",
       "7       13.0      28.0  \n",
       "8        9.0      24.0  \n",
       "9        8.0      19.0  \n",
       "10       7.0      16.0  \n",
       "11       7.0      15.0  \n",
       "12       7.0      14.0  \n",
       "13       6.0      12.0  \n",
       "14       6.0      11.0  \n",
       "15       6.0       9.0  \n",
       "16       4.0       9.0  \n",
       "17       4.0       8.0  \n",
       "18       4.0       6.0  \n",
       "19       4.0       5.0  \n",
       "20       1.0       4.0  \n",
       "\n",
       "[21 rows x 670 columns]"
      ]
     },
     "execution_count": 1,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "import os\n",
    "import numpy as np\n",
    "import matplotlib.pyplot as plt\n",
    "import pandas as pd\n",
    "import seaborn as sns\n",
    "from matplotlib import rc\n",
    "from numpy.linalg import inv \n",
    "from sklearn import linear_model\n",
    "\n",
    "import pystan\n",
    "\n",
    "rc('text', usetex=True)\n",
    "\n",
    "pd.set_option('display.max_columns', 100)\n",
    "\n",
    "aa2codon = {                                                         # DNA codon table\n",
    "    'A' : ['GCT', 'GCC', 'GCA', 'GCG'],\n",
    "    'R' : ['CGT', 'CGC', 'CGA', 'CGG', 'AGA', 'AGG'],\n",
    "    'N' : ['AAT', 'AAC'],\n",
    "    'D' : ['GAT', 'GAC'],\n",
    "    'C' : ['TGT', 'TGC'],\n",
    "    'E' : ['GAA', 'GAG'],\n",
    "    'Q' : ['CAA', 'CAG'],\n",
    "    'G' : ['GGT', 'GGC', 'GGA', 'GGG'],\n",
    "    'H' : ['CAT', 'CAC'],\n",
    "    'I' : ['ATT', 'ATC', 'ATA'],\n",
    "    'L' : ['TTA', 'TTG', 'CTT', 'CTC', 'CTA', 'CTG'],\n",
    "    'K' : ['AAA', 'AAG'],\n",
    "    'M' : ['ATG'],\n",
    "    'F' : ['TTT', 'TTC'],\n",
    "    'P' : ['CCT', 'CCC', 'CCA', 'CCG'],\n",
    "    'S' : ['TCT', 'TCC', 'TCA', 'TCG', 'AGT', 'AGC'],\n",
    "    'T' : ['ACT', 'ACC', 'ACA', 'ACG'],\n",
    "    'W' : ['TGG'],\n",
    "    'Y' : ['TAT', 'TAC'],\n",
    "    'V' : ['GTT', 'GTC', 'GTA', 'GTG'],\n",
    "    '*' : ['TAA', 'TGA', 'TAG'],\n",
    "    '-' : ['---'],\n",
    "    }\n",
    "\n",
    "#BG505 mutDNA1\n",
    "init_pop = pd.read_csv(\"./output/simulation/init_sampling.csv\")\n",
    "codon_list = init_pop.columns[2:]\n",
    "key_list = list(aa2codon.keys())\n",
    "val_list = list(aa2codon.values())\n",
    "WT_seq = []\n",
    "for i in range(init_pop.shape[0]):\n",
    "    list_of_keys = [key for key, list_of_values in aa2codon.items() if init_pop['wildtype'].tolist()[i] in list_of_values][0]\n",
    "    WT_seq.append(list_of_keys)\n",
    "\n",
    "df_col = []\n",
    "for i in range(len(codon_list)):\n",
    "    list_of_keys = [key for key, list_of_values in aa2codon.items() if codon_list[i] in list_of_values][0]\n",
    "    df_col.append(list_of_keys)\n",
    "    \n",
    "df_seperate = []\n",
    "for i in range(len(codon_list)):\n",
    "    df_seperate.append(init_pop[['site', 'wildtype', codon_list[i]]])\n",
    "\n",
    "for i in range(len(codon_list)):\n",
    "    df_seperate[i] = df_seperate[i].rename(columns = {df_seperate[i].columns[2]: [key for key, list_of_values in aa2codon.items() if df_seperate[i].columns[2] in list_of_values][0]})\n",
    "    temp_list = []\n",
    "    for j in range(init_pop.shape[0]):\n",
    "        temp_list.append(df_seperate[i].columns[2])\n",
    "    df_seperate[i]['AA'] = temp_list\n",
    "    \n",
    "df_AA = df_seperate[0].copy()\n",
    "for i in range(len(codon_list)-1):\n",
    "    df_AA = pd.concat([df_AA, df_seperate[i+1]]).groupby(['site', 'wildtype', 'AA']).sum().reset_index()\n",
    "df_AA = df_AA.groupby(['site', 'wildtype']).sum().reset_index()\n",
    "\n",
    "total_count = df_AA[df_AA.columns[2:]].T.sum().tolist()\n",
    "df_AA['total_counts'] = total_count\n",
    "\n",
    "df_temp = df_AA.copy()\n",
    "df_temp = df_temp.drop('total_counts', axis=1)\n",
    "AA_list = df_temp.columns[2:]\n",
    "df_distribution = df_temp[AA_list]\n",
    "df_distribution = df_distribution.T\n",
    "arr = df_distribution.values\n",
    "arr.sort(axis=0)\n",
    "arr = arr[::-1]\n",
    "\n",
    "df_distribution = pd.DataFrame(arr, columns=df_distribution.columns)\n",
    "df_distribution"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 2,
   "id": "130a2128",
   "metadata": {},
   "outputs": [
    {
     "name": "stderr",
     "output_type": "stream",
     "text": [
      "INFO:pystan:COMPILING THE C++ CODE FOR MODEL anon_model_78a200dd4b762d3294f98602e987e4aa NOW.\n"
     ]
    }
   ],
   "source": [
    "model = \"\"\"\n",
    "data {\n",
    "    int N;\n",
    "    int K;\n",
    "    int y[N,K];\n",
    "}\n",
    "\n",
    "parameters {\n",
    "    simplex[K] theta;\n",
    "}\n",
    "\n",
    "model {\n",
    "    for (i in 1:N) {\n",
    "        y[i,] ~ multinomial(theta);\n",
    "    } \n",
    "}\n",
    "\"\"\"\n",
    "multinomial = pystan.StanModel(model_code=model)\n",
    "\n",
    "# true frequency distribution"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 3,
   "id": "a50c96df",
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "Inference for Stan model: anon_model_78a200dd4b762d3294f98602e987e4aa.\n",
      "4 chains, each with iter=1000; warmup=500; thin=1; \n",
      "post-warmup draws per chain=500, total post-warmup draws=2000.\n",
      "\n",
      "            mean se_mean     sd   2.5%    25%    50%    75%  97.5%  n_eff   Rhat\n",
      "theta[1]     1.0  6.8e-8 3.2e-6    1.0    1.0    1.0    1.0    1.0   2185    1.0\n",
      "theta[2]  3.6e-4  2.6e-8 1.4e-6 3.6e-4 3.6e-4 3.6e-4 3.7e-4 3.7e-4   2806    1.0\n",
      "theta[3]  2.2e-4  2.0e-8 1.1e-6 2.2e-4 2.2e-4 2.2e-4 2.3e-4 2.3e-4   2979    1.0\n",
      "theta[4]  1.8e-4  1.7e-8 9.6e-7 1.8e-4 1.8e-4 1.8e-4 1.8e-4 1.8e-4   3063    1.0\n",
      "theta[5]  1.5e-4  1.5e-8 8.8e-7 1.5e-4 1.5e-4 1.5e-4 1.5e-4 1.5e-4   3554    1.0\n",
      "theta[6]  1.4e-4  1.4e-8 8.4e-7 1.3e-4 1.4e-4 1.4e-4 1.4e-4 1.4e-4   3489    1.0\n",
      "theta[7]  1.2e-4  1.5e-8 7.8e-7 1.2e-4 1.2e-4 1.2e-4 1.2e-4 1.2e-4   2628    1.0\n",
      "theta[8]  1.1e-4  1.3e-8 7.6e-7 1.1e-4 1.1e-4 1.1e-4 1.1e-4 1.1e-4   3200    1.0\n",
      "theta[9]  9.8e-5  1.3e-8 7.1e-7 9.7e-5 9.8e-5 9.8e-5 9.9e-510.0e-5   2974    1.0\n",
      "theta[10] 8.7e-5  1.3e-8 6.9e-7 8.5e-5 8.6e-5 8.7e-5 8.7e-5 8.8e-5   2753    1.0\n",
      "theta[11] 7.7e-5  1.1e-8 6.3e-7 7.6e-5 7.7e-5 7.7e-5 7.8e-5 7.8e-5   3372    1.0\n",
      "theta[12] 6.9e-5  1.0e-8 6.0e-7 6.8e-5 6.9e-5 6.9e-5 7.0e-5 7.0e-5   3439    1.0\n",
      "theta[13] 6.3e-5  1.1e-8 5.8e-7 6.2e-5 6.3e-5 6.3e-5 6.3e-5 6.4e-5   2748    1.0\n",
      "theta[14] 5.7e-5  9.7e-9 5.6e-7 5.6e-5 5.7e-5 5.7e-5 5.8e-5 5.9e-5   3309    1.0\n",
      "theta[15] 5.2e-5  9.5e-9 5.3e-7 5.1e-5 5.2e-5 5.2e-5 5.3e-5 5.3e-5   3138    1.0\n",
      "theta[16] 4.7e-5  9.1e-9 4.9e-7 4.6e-5 4.7e-5 4.7e-5 4.8e-5 4.8e-5   2945    1.0\n",
      "theta[17] 4.2e-5  9.6e-9 4.6e-7 4.1e-5 4.2e-5 4.2e-5 4.3e-5 4.3e-5   2292    1.0\n",
      "theta[18] 3.8e-5  8.0e-9 4.5e-7 3.7e-5 3.7e-5 3.8e-5 3.8e-5 3.8e-5   3123    1.0\n",
      "theta[19] 3.2e-5  8.0e-9 4.2e-7 3.1e-5 3.2e-5 3.2e-5 3.3e-5 3.3e-5   2785    1.0\n",
      "theta[20] 2.5e-5  7.7e-9 3.6e-7 2.5e-5 2.5e-5 2.5e-5 2.5e-5 2.6e-5   2202    1.0\n",
      "theta[21] 1.7e-5  5.7e-9 3.0e-7 1.7e-5 1.7e-5 1.7e-5 1.7e-5 1.8e-5   2836    1.0\n",
      "lp__      -3.8e6    0.11   3.21 -3.8e6 -3.8e6 -3.8e6 -3.8e6 -3.8e6    924    1.0\n",
      "\n",
      "Samples were drawn using NUTS at Thu Sep 21 12:17:24 2023.\n",
      "For each parameter, n_eff is a crude measure of effective sample size,\n",
      "and Rhat is the potential scale reduction factor on split chains (at \n",
      "convergence, Rhat=1).\n"
     ]
    },
    {
     "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>0</th>\n",
       "      <th>1</th>\n",
       "      <th>2</th>\n",
       "      <th>3</th>\n",
       "      <th>4</th>\n",
       "      <th>5</th>\n",
       "      <th>6</th>\n",
       "      <th>7</th>\n",
       "      <th>8</th>\n",
       "      <th>9</th>\n",
       "      <th>10</th>\n",
       "      <th>11</th>\n",
       "      <th>12</th>\n",
       "      <th>13</th>\n",
       "      <th>14</th>\n",
       "      <th>15</th>\n",
       "      <th>16</th>\n",
       "      <th>17</th>\n",
       "      <th>18</th>\n",
       "      <th>19</th>\n",
       "      <th>20</th>\n",
       "    </tr>\n",
       "  </thead>\n",
       "  <tbody>\n",
       "    <tr>\n",
       "      <th>0</th>\n",
       "      <td>0.998004</td>\n",
       "      <td>0.000364</td>\n",
       "      <td>0.000226</td>\n",
       "      <td>0.000179</td>\n",
       "      <td>0.000154</td>\n",
       "      <td>0.000136</td>\n",
       "      <td>0.000121</td>\n",
       "      <td>0.000110</td>\n",
       "      <td>0.000099</td>\n",
       "      <td>0.000087</td>\n",
       "      <td>0.000077</td>\n",
       "      <td>0.000070</td>\n",
       "      <td>0.000063</td>\n",
       "      <td>0.000058</td>\n",
       "      <td>0.000052</td>\n",
       "      <td>0.000047</td>\n",
       "      <td>0.000043</td>\n",
       "      <td>0.000038</td>\n",
       "      <td>0.000032</td>\n",
       "      <td>0.000026</td>\n",
       "      <td>0.000017</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>1</th>\n",
       "      <td>0.998005</td>\n",
       "      <td>0.000365</td>\n",
       "      <td>0.000224</td>\n",
       "      <td>0.000179</td>\n",
       "      <td>0.000153</td>\n",
       "      <td>0.000135</td>\n",
       "      <td>0.000122</td>\n",
       "      <td>0.000111</td>\n",
       "      <td>0.000098</td>\n",
       "      <td>0.000088</td>\n",
       "      <td>0.000077</td>\n",
       "      <td>0.000069</td>\n",
       "      <td>0.000062</td>\n",
       "      <td>0.000058</td>\n",
       "      <td>0.000053</td>\n",
       "      <td>0.000047</td>\n",
       "      <td>0.000042</td>\n",
       "      <td>0.000038</td>\n",
       "      <td>0.000032</td>\n",
       "      <td>0.000025</td>\n",
       "      <td>0.000017</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>2</th>\n",
       "      <td>0.998006</td>\n",
       "      <td>0.000368</td>\n",
       "      <td>0.000223</td>\n",
       "      <td>0.000177</td>\n",
       "      <td>0.000154</td>\n",
       "      <td>0.000136</td>\n",
       "      <td>0.000121</td>\n",
       "      <td>0.000111</td>\n",
       "      <td>0.000099</td>\n",
       "      <td>0.000087</td>\n",
       "      <td>0.000077</td>\n",
       "      <td>0.000069</td>\n",
       "      <td>0.000063</td>\n",
       "      <td>0.000057</td>\n",
       "      <td>0.000052</td>\n",
       "      <td>0.000047</td>\n",
       "      <td>0.000042</td>\n",
       "      <td>0.000038</td>\n",
       "      <td>0.000032</td>\n",
       "      <td>0.000026</td>\n",
       "      <td>0.000017</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>3</th>\n",
       "      <td>0.998003</td>\n",
       "      <td>0.000363</td>\n",
       "      <td>0.000225</td>\n",
       "      <td>0.000177</td>\n",
       "      <td>0.000155</td>\n",
       "      <td>0.000136</td>\n",
       "      <td>0.000122</td>\n",
       "      <td>0.000111</td>\n",
       "      <td>0.000098</td>\n",
       "      <td>0.000087</td>\n",
       "      <td>0.000077</td>\n",
       "      <td>0.000070</td>\n",
       "      <td>0.000063</td>\n",
       "      <td>0.000057</td>\n",
       "      <td>0.000053</td>\n",
       "      <td>0.000047</td>\n",
       "      <td>0.000042</td>\n",
       "      <td>0.000038</td>\n",
       "      <td>0.000032</td>\n",
       "      <td>0.000025</td>\n",
       "      <td>0.000017</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>4</th>\n",
       "      <td>0.998011</td>\n",
       "      <td>0.000363</td>\n",
       "      <td>0.000223</td>\n",
       "      <td>0.000179</td>\n",
       "      <td>0.000152</td>\n",
       "      <td>0.000137</td>\n",
       "      <td>0.000122</td>\n",
       "      <td>0.000109</td>\n",
       "      <td>0.000097</td>\n",
       "      <td>0.000086</td>\n",
       "      <td>0.000077</td>\n",
       "      <td>0.000069</td>\n",
       "      <td>0.000063</td>\n",
       "      <td>0.000058</td>\n",
       "      <td>0.000053</td>\n",
       "      <td>0.000046</td>\n",
       "      <td>0.000042</td>\n",
       "      <td>0.000037</td>\n",
       "      <td>0.000032</td>\n",
       "      <td>0.000025</td>\n",
       "      <td>0.000017</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>5</th>\n",
       "      <td>0.998003</td>\n",
       "      <td>0.000364</td>\n",
       "      <td>0.000224</td>\n",
       "      <td>0.000178</td>\n",
       "      <td>0.000154</td>\n",
       "      <td>0.000136</td>\n",
       "      <td>0.000123</td>\n",
       "      <td>0.000111</td>\n",
       "      <td>0.000098</td>\n",
       "      <td>0.000087</td>\n",
       "      <td>0.000077</td>\n",
       "      <td>0.000070</td>\n",
       "      <td>0.000063</td>\n",
       "      <td>0.000058</td>\n",
       "      <td>0.000052</td>\n",
       "      <td>0.000047</td>\n",
       "      <td>0.000043</td>\n",
       "      <td>0.000037</td>\n",
       "      <td>0.000032</td>\n",
       "      <td>0.000025</td>\n",
       "      <td>0.000018</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>6</th>\n",
       "      <td>0.998008</td>\n",
       "      <td>0.000364</td>\n",
       "      <td>0.000223</td>\n",
       "      <td>0.000176</td>\n",
       "      <td>0.000153</td>\n",
       "      <td>0.000135</td>\n",
       "      <td>0.000122</td>\n",
       "      <td>0.000111</td>\n",
       "      <td>0.000099</td>\n",
       "      <td>0.000086</td>\n",
       "      <td>0.000079</td>\n",
       "      <td>0.000070</td>\n",
       "      <td>0.000062</td>\n",
       "      <td>0.000057</td>\n",
       "      <td>0.000052</td>\n",
       "      <td>0.000048</td>\n",
       "      <td>0.000041</td>\n",
       "      <td>0.000037</td>\n",
       "      <td>0.000032</td>\n",
       "      <td>0.000025</td>\n",
       "      <td>0.000017</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>7</th>\n",
       "      <td>0.998005</td>\n",
       "      <td>0.000363</td>\n",
       "      <td>0.000223</td>\n",
       "      <td>0.000178</td>\n",
       "      <td>0.000154</td>\n",
       "      <td>0.000136</td>\n",
       "      <td>0.000123</td>\n",
       "      <td>0.000111</td>\n",
       "      <td>0.000098</td>\n",
       "      <td>0.000086</td>\n",
       "      <td>0.000077</td>\n",
       "      <td>0.000071</td>\n",
       "      <td>0.000063</td>\n",
       "      <td>0.000057</td>\n",
       "      <td>0.000052</td>\n",
       "      <td>0.000047</td>\n",
       "      <td>0.000042</td>\n",
       "      <td>0.000038</td>\n",
       "      <td>0.000032</td>\n",
       "      <td>0.000025</td>\n",
       "      <td>0.000017</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>8</th>\n",
       "      <td>0.998007</td>\n",
       "      <td>0.000365</td>\n",
       "      <td>0.000224</td>\n",
       "      <td>0.000177</td>\n",
       "      <td>0.000151</td>\n",
       "      <td>0.000135</td>\n",
       "      <td>0.000123</td>\n",
       "      <td>0.000111</td>\n",
       "      <td>0.000099</td>\n",
       "      <td>0.000086</td>\n",
       "      <td>0.000077</td>\n",
       "      <td>0.000070</td>\n",
       "      <td>0.000063</td>\n",
       "      <td>0.000058</td>\n",
       "      <td>0.000052</td>\n",
       "      <td>0.000047</td>\n",
       "      <td>0.000042</td>\n",
       "      <td>0.000037</td>\n",
       "      <td>0.000032</td>\n",
       "      <td>0.000025</td>\n",
       "      <td>0.000017</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>9</th>\n",
       "      <td>0.998005</td>\n",
       "      <td>0.000365</td>\n",
       "      <td>0.000225</td>\n",
       "      <td>0.000179</td>\n",
       "      <td>0.000151</td>\n",
       "      <td>0.000135</td>\n",
       "      <td>0.000124</td>\n",
       "      <td>0.000109</td>\n",
       "      <td>0.000097</td>\n",
       "      <td>0.000088</td>\n",
       "      <td>0.000078</td>\n",
       "      <td>0.000069</td>\n",
       "      <td>0.000062</td>\n",
       "      <td>0.000056</td>\n",
       "      <td>0.000053</td>\n",
       "      <td>0.000048</td>\n",
       "      <td>0.000042</td>\n",
       "      <td>0.000037</td>\n",
       "      <td>0.000032</td>\n",
       "      <td>0.000026</td>\n",
       "      <td>0.000018</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>10</th>\n",
       "      <td>0.998009</td>\n",
       "      <td>0.000363</td>\n",
       "      <td>0.000223</td>\n",
       "      <td>0.000176</td>\n",
       "      <td>0.000153</td>\n",
       "      <td>0.000136</td>\n",
       "      <td>0.000122</td>\n",
       "      <td>0.000110</td>\n",
       "      <td>0.000097</td>\n",
       "      <td>0.000087</td>\n",
       "      <td>0.000077</td>\n",
       "      <td>0.000069</td>\n",
       "      <td>0.000063</td>\n",
       "      <td>0.000057</td>\n",
       "      <td>0.000053</td>\n",
       "      <td>0.000048</td>\n",
       "      <td>0.000043</td>\n",
       "      <td>0.000038</td>\n",
       "      <td>0.000032</td>\n",
       "      <td>0.000026</td>\n",
       "      <td>0.000017</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>11</th>\n",
       "      <td>0.998003</td>\n",
       "      <td>0.000367</td>\n",
       "      <td>0.000226</td>\n",
       "      <td>0.000177</td>\n",
       "      <td>0.000152</td>\n",
       "      <td>0.000136</td>\n",
       "      <td>0.000123</td>\n",
       "      <td>0.000109</td>\n",
       "      <td>0.000099</td>\n",
       "      <td>0.000086</td>\n",
       "      <td>0.000077</td>\n",
       "      <td>0.000068</td>\n",
       "      <td>0.000063</td>\n",
       "      <td>0.000058</td>\n",
       "      <td>0.000053</td>\n",
       "      <td>0.000048</td>\n",
       "      <td>0.000043</td>\n",
       "      <td>0.000038</td>\n",
       "      <td>0.000032</td>\n",
       "      <td>0.000025</td>\n",
       "      <td>0.000018</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>12</th>\n",
       "      <td>0.998004</td>\n",
       "      <td>0.000363</td>\n",
       "      <td>0.000224</td>\n",
       "      <td>0.000181</td>\n",
       "      <td>0.000153</td>\n",
       "      <td>0.000136</td>\n",
       "      <td>0.000123</td>\n",
       "      <td>0.000112</td>\n",
       "      <td>0.000098</td>\n",
       "      <td>0.000086</td>\n",
       "      <td>0.000076</td>\n",
       "      <td>0.000069</td>\n",
       "      <td>0.000063</td>\n",
       "      <td>0.000057</td>\n",
       "      <td>0.000053</td>\n",
       "      <td>0.000048</td>\n",
       "      <td>0.000042</td>\n",
       "      <td>0.000038</td>\n",
       "      <td>0.000031</td>\n",
       "      <td>0.000025</td>\n",
       "      <td>0.000017</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>13</th>\n",
       "      <td>0.998009</td>\n",
       "      <td>0.000364</td>\n",
       "      <td>0.000223</td>\n",
       "      <td>0.000178</td>\n",
       "      <td>0.000153</td>\n",
       "      <td>0.000136</td>\n",
       "      <td>0.000122</td>\n",
       "      <td>0.000110</td>\n",
       "      <td>0.000098</td>\n",
       "      <td>0.000086</td>\n",
       "      <td>0.000078</td>\n",
       "      <td>0.000068</td>\n",
       "      <td>0.000063</td>\n",
       "      <td>0.000057</td>\n",
       "      <td>0.000052</td>\n",
       "      <td>0.000047</td>\n",
       "      <td>0.000042</td>\n",
       "      <td>0.000038</td>\n",
       "      <td>0.000033</td>\n",
       "      <td>0.000025</td>\n",
       "      <td>0.000017</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>14</th>\n",
       "      <td>0.998004</td>\n",
       "      <td>0.000365</td>\n",
       "      <td>0.000225</td>\n",
       "      <td>0.000178</td>\n",
       "      <td>0.000152</td>\n",
       "      <td>0.000136</td>\n",
       "      <td>0.000123</td>\n",
       "      <td>0.000110</td>\n",
       "      <td>0.000099</td>\n",
       "      <td>0.000086</td>\n",
       "      <td>0.000076</td>\n",
       "      <td>0.000069</td>\n",
       "      <td>0.000063</td>\n",
       "      <td>0.000057</td>\n",
       "      <td>0.000052</td>\n",
       "      <td>0.000048</td>\n",
       "      <td>0.000043</td>\n",
       "      <td>0.000038</td>\n",
       "      <td>0.000033</td>\n",
       "      <td>0.000026</td>\n",
       "      <td>0.000017</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>15</th>\n",
       "      <td>0.998005</td>\n",
       "      <td>0.000364</td>\n",
       "      <td>0.000225</td>\n",
       "      <td>0.000178</td>\n",
       "      <td>0.000153</td>\n",
       "      <td>0.000136</td>\n",
       "      <td>0.000122</td>\n",
       "      <td>0.000111</td>\n",
       "      <td>0.000098</td>\n",
       "      <td>0.000086</td>\n",
       "      <td>0.000078</td>\n",
       "      <td>0.000070</td>\n",
       "      <td>0.000063</td>\n",
       "      <td>0.000057</td>\n",
       "      <td>0.000052</td>\n",
       "      <td>0.000048</td>\n",
       "      <td>0.000043</td>\n",
       "      <td>0.000038</td>\n",
       "      <td>0.000033</td>\n",
       "      <td>0.000025</td>\n",
       "      <td>0.000017</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>16</th>\n",
       "      <td>0.998007</td>\n",
       "      <td>0.000363</td>\n",
       "      <td>0.000223</td>\n",
       "      <td>0.000178</td>\n",
       "      <td>0.000152</td>\n",
       "      <td>0.000135</td>\n",
       "      <td>0.000122</td>\n",
       "      <td>0.000112</td>\n",
       "      <td>0.000099</td>\n",
       "      <td>0.000087</td>\n",
       "      <td>0.000077</td>\n",
       "      <td>0.000069</td>\n",
       "      <td>0.000062</td>\n",
       "      <td>0.000058</td>\n",
       "      <td>0.000052</td>\n",
       "      <td>0.000047</td>\n",
       "      <td>0.000043</td>\n",
       "      <td>0.000037</td>\n",
       "      <td>0.000032</td>\n",
       "      <td>0.000026</td>\n",
       "      <td>0.000017</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>17</th>\n",
       "      <td>0.998004</td>\n",
       "      <td>0.000365</td>\n",
       "      <td>0.000225</td>\n",
       "      <td>0.000178</td>\n",
       "      <td>0.000153</td>\n",
       "      <td>0.000135</td>\n",
       "      <td>0.000122</td>\n",
       "      <td>0.000110</td>\n",
       "      <td>0.000099</td>\n",
       "      <td>0.000087</td>\n",
       "      <td>0.000078</td>\n",
       "      <td>0.000069</td>\n",
       "      <td>0.000063</td>\n",
       "      <td>0.000057</td>\n",
       "      <td>0.000053</td>\n",
       "      <td>0.000047</td>\n",
       "      <td>0.000042</td>\n",
       "      <td>0.000037</td>\n",
       "      <td>0.000032</td>\n",
       "      <td>0.000025</td>\n",
       "      <td>0.000017</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>18</th>\n",
       "      <td>0.998004</td>\n",
       "      <td>0.000363</td>\n",
       "      <td>0.000224</td>\n",
       "      <td>0.000180</td>\n",
       "      <td>0.000152</td>\n",
       "      <td>0.000136</td>\n",
       "      <td>0.000122</td>\n",
       "      <td>0.000110</td>\n",
       "      <td>0.000098</td>\n",
       "      <td>0.000087</td>\n",
       "      <td>0.000078</td>\n",
       "      <td>0.000069</td>\n",
       "      <td>0.000062</td>\n",
       "      <td>0.000057</td>\n",
       "      <td>0.000053</td>\n",
       "      <td>0.000048</td>\n",
       "      <td>0.000042</td>\n",
       "      <td>0.000038</td>\n",
       "      <td>0.000031</td>\n",
       "      <td>0.000025</td>\n",
       "      <td>0.000018</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>19</th>\n",
       "      <td>0.998008</td>\n",
       "      <td>0.000363</td>\n",
       "      <td>0.000225</td>\n",
       "      <td>0.000177</td>\n",
       "      <td>0.000153</td>\n",
       "      <td>0.000136</td>\n",
       "      <td>0.000122</td>\n",
       "      <td>0.000110</td>\n",
       "      <td>0.000097</td>\n",
       "      <td>0.000086</td>\n",
       "      <td>0.000077</td>\n",
       "      <td>0.000070</td>\n",
       "      <td>0.000062</td>\n",
       "      <td>0.000057</td>\n",
       "      <td>0.000052</td>\n",
       "      <td>0.000048</td>\n",
       "      <td>0.000042</td>\n",
       "      <td>0.000038</td>\n",
       "      <td>0.000032</td>\n",
       "      <td>0.000025</td>\n",
       "      <td>0.000017</td>\n",
       "    </tr>\n",
       "  </tbody>\n",
       "</table>\n",
       "</div>"
      ],
      "text/plain": [
       "          0         1         2         3         4         5         6   \\\n",
       "0   0.998004  0.000364  0.000226  0.000179  0.000154  0.000136  0.000121   \n",
       "1   0.998005  0.000365  0.000224  0.000179  0.000153  0.000135  0.000122   \n",
       "2   0.998006  0.000368  0.000223  0.000177  0.000154  0.000136  0.000121   \n",
       "3   0.998003  0.000363  0.000225  0.000177  0.000155  0.000136  0.000122   \n",
       "4   0.998011  0.000363  0.000223  0.000179  0.000152  0.000137  0.000122   \n",
       "5   0.998003  0.000364  0.000224  0.000178  0.000154  0.000136  0.000123   \n",
       "6   0.998008  0.000364  0.000223  0.000176  0.000153  0.000135  0.000122   \n",
       "7   0.998005  0.000363  0.000223  0.000178  0.000154  0.000136  0.000123   \n",
       "8   0.998007  0.000365  0.000224  0.000177  0.000151  0.000135  0.000123   \n",
       "9   0.998005  0.000365  0.000225  0.000179  0.000151  0.000135  0.000124   \n",
       "10  0.998009  0.000363  0.000223  0.000176  0.000153  0.000136  0.000122   \n",
       "11  0.998003  0.000367  0.000226  0.000177  0.000152  0.000136  0.000123   \n",
       "12  0.998004  0.000363  0.000224  0.000181  0.000153  0.000136  0.000123   \n",
       "13  0.998009  0.000364  0.000223  0.000178  0.000153  0.000136  0.000122   \n",
       "14  0.998004  0.000365  0.000225  0.000178  0.000152  0.000136  0.000123   \n",
       "15  0.998005  0.000364  0.000225  0.000178  0.000153  0.000136  0.000122   \n",
       "16  0.998007  0.000363  0.000223  0.000178  0.000152  0.000135  0.000122   \n",
       "17  0.998004  0.000365  0.000225  0.000178  0.000153  0.000135  0.000122   \n",
       "18  0.998004  0.000363  0.000224  0.000180  0.000152  0.000136  0.000122   \n",
       "19  0.998008  0.000363  0.000225  0.000177  0.000153  0.000136  0.000122   \n",
       "\n",
       "          7         8         9         10        11        12        13  \\\n",
       "0   0.000110  0.000099  0.000087  0.000077  0.000070  0.000063  0.000058   \n",
       "1   0.000111  0.000098  0.000088  0.000077  0.000069  0.000062  0.000058   \n",
       "2   0.000111  0.000099  0.000087  0.000077  0.000069  0.000063  0.000057   \n",
       "3   0.000111  0.000098  0.000087  0.000077  0.000070  0.000063  0.000057   \n",
       "4   0.000109  0.000097  0.000086  0.000077  0.000069  0.000063  0.000058   \n",
       "5   0.000111  0.000098  0.000087  0.000077  0.000070  0.000063  0.000058   \n",
       "6   0.000111  0.000099  0.000086  0.000079  0.000070  0.000062  0.000057   \n",
       "7   0.000111  0.000098  0.000086  0.000077  0.000071  0.000063  0.000057   \n",
       "8   0.000111  0.000099  0.000086  0.000077  0.000070  0.000063  0.000058   \n",
       "9   0.000109  0.000097  0.000088  0.000078  0.000069  0.000062  0.000056   \n",
       "10  0.000110  0.000097  0.000087  0.000077  0.000069  0.000063  0.000057   \n",
       "11  0.000109  0.000099  0.000086  0.000077  0.000068  0.000063  0.000058   \n",
       "12  0.000112  0.000098  0.000086  0.000076  0.000069  0.000063  0.000057   \n",
       "13  0.000110  0.000098  0.000086  0.000078  0.000068  0.000063  0.000057   \n",
       "14  0.000110  0.000099  0.000086  0.000076  0.000069  0.000063  0.000057   \n",
       "15  0.000111  0.000098  0.000086  0.000078  0.000070  0.000063  0.000057   \n",
       "16  0.000112  0.000099  0.000087  0.000077  0.000069  0.000062  0.000058   \n",
       "17  0.000110  0.000099  0.000087  0.000078  0.000069  0.000063  0.000057   \n",
       "18  0.000110  0.000098  0.000087  0.000078  0.000069  0.000062  0.000057   \n",
       "19  0.000110  0.000097  0.000086  0.000077  0.000070  0.000062  0.000057   \n",
       "\n",
       "          14        15        16        17        18        19        20  \n",
       "0   0.000052  0.000047  0.000043  0.000038  0.000032  0.000026  0.000017  \n",
       "1   0.000053  0.000047  0.000042  0.000038  0.000032  0.000025  0.000017  \n",
       "2   0.000052  0.000047  0.000042  0.000038  0.000032  0.000026  0.000017  \n",
       "3   0.000053  0.000047  0.000042  0.000038  0.000032  0.000025  0.000017  \n",
       "4   0.000053  0.000046  0.000042  0.000037  0.000032  0.000025  0.000017  \n",
       "5   0.000052  0.000047  0.000043  0.000037  0.000032  0.000025  0.000018  \n",
       "6   0.000052  0.000048  0.000041  0.000037  0.000032  0.000025  0.000017  \n",
       "7   0.000052  0.000047  0.000042  0.000038  0.000032  0.000025  0.000017  \n",
       "8   0.000052  0.000047  0.000042  0.000037  0.000032  0.000025  0.000017  \n",
       "9   0.000053  0.000048  0.000042  0.000037  0.000032  0.000026  0.000018  \n",
       "10  0.000053  0.000048  0.000043  0.000038  0.000032  0.000026  0.000017  \n",
       "11  0.000053  0.000048  0.000043  0.000038  0.000032  0.000025  0.000018  \n",
       "12  0.000053  0.000048  0.000042  0.000038  0.000031  0.000025  0.000017  \n",
       "13  0.000052  0.000047  0.000042  0.000038  0.000033  0.000025  0.000017  \n",
       "14  0.000052  0.000048  0.000043  0.000038  0.000033  0.000026  0.000017  \n",
       "15  0.000052  0.000048  0.000043  0.000038  0.000033  0.000025  0.000017  \n",
       "16  0.000052  0.000047  0.000043  0.000037  0.000032  0.000026  0.000017  \n",
       "17  0.000053  0.000047  0.000042  0.000037  0.000032  0.000025  0.000017  \n",
       "18  0.000053  0.000048  0.000042  0.000038  0.000031  0.000025  0.000018  \n",
       "19  0.000052  0.000048  0.000042  0.000038  0.000032  0.000025  0.000017  "
      ]
     },
     "execution_count": 3,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "np.random.seed(0)\n",
    "df_distribution_ = df_distribution.astype('int32')\n",
    "df_multinomial = df_distribution_.T\n",
    "data = df_multinomial.to_numpy()\n",
    "input_data = {'N': len(data), 'K': len(data[0]), 'y': data}\n",
    "fit = multinomial.sampling(data=input_data, iter=1000, chains=4, warmup=500, thin=1, seed=101)\n",
    "print(fit)\n",
    "\n",
    "np.random.seed(0)\n",
    "random_sampling = np.random.randint(2000, size=20)\n",
    "frequency_list = []\n",
    "for i in range(len(random_sampling)):\n",
    "    frequency_list.append([])\n",
    "    for j in range(21):\n",
    "        frequency_list[i].append(fit['theta['+str(j+1)+']'][i])\n",
    "\n",
    "df_sampling = pd.DataFrame.from_records(frequency_list)\n",
    "df_sampling"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 4,
   "id": "80fbe8a6",
   "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>K</th>\n",
       "      <th>N</th>\n",
       "      <th>T</th>\n",
       "      <th>R</th>\n",
       "      <th>S</th>\n",
       "      <th>I</th>\n",
       "      <th>M</th>\n",
       "      <th>Q</th>\n",
       "      <th>H</th>\n",
       "      <th>P</th>\n",
       "      <th>L</th>\n",
       "      <th>E</th>\n",
       "      <th>D</th>\n",
       "      <th>A</th>\n",
       "      <th>G</th>\n",
       "      <th>V</th>\n",
       "      <th>*</th>\n",
       "      <th>Y</th>\n",
       "      <th>C</th>\n",
       "      <th>W</th>\n",
       "      <th>F</th>\n",
       "    </tr>\n",
       "  </thead>\n",
       "  <tbody>\n",
       "    <tr>\n",
       "      <th>0</th>\n",
       "      <td>0.176405</td>\n",
       "      <td>0.040016</td>\n",
       "      <td>0.097874</td>\n",
       "      <td>0.224089</td>\n",
       "      <td>0.186756</td>\n",
       "      <td>-0.097728</td>\n",
       "      <td>0.095009</td>\n",
       "      <td>-0.015136</td>\n",
       "      <td>-0.010322</td>\n",
       "      <td>0.041060</td>\n",
       "      <td>0.014404</td>\n",
       "      <td>0.145427</td>\n",
       "      <td>0.076104</td>\n",
       "      <td>0.012168</td>\n",
       "      <td>0.044386</td>\n",
       "      <td>0.033367</td>\n",
       "      <td>0.149408</td>\n",
       "      <td>-0.020516</td>\n",
       "      <td>0.031307</td>\n",
       "      <td>-0.085410</td>\n",
       "      <td>-0.255299</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>1</th>\n",
       "      <td>0.065362</td>\n",
       "      <td>0.086444</td>\n",
       "      <td>-0.074217</td>\n",
       "      <td>0.226975</td>\n",
       "      <td>-0.145437</td>\n",
       "      <td>0.004576</td>\n",
       "      <td>-0.018718</td>\n",
       "      <td>0.153278</td>\n",
       "      <td>0.146936</td>\n",
       "      <td>0.015495</td>\n",
       "      <td>0.037816</td>\n",
       "      <td>-0.088779</td>\n",
       "      <td>-0.198080</td>\n",
       "      <td>-0.034791</td>\n",
       "      <td>0.015635</td>\n",
       "      <td>0.123029</td>\n",
       "      <td>0.120238</td>\n",
       "      <td>-0.038733</td>\n",
       "      <td>-0.030230</td>\n",
       "      <td>-0.104855</td>\n",
       "      <td>-0.142002</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>2</th>\n",
       "      <td>-0.170627</td>\n",
       "      <td>0.195078</td>\n",
       "      <td>-0.050965</td>\n",
       "      <td>-0.043807</td>\n",
       "      <td>-0.125280</td>\n",
       "      <td>0.077749</td>\n",
       "      <td>-0.161390</td>\n",
       "      <td>-0.021274</td>\n",
       "      <td>-0.089547</td>\n",
       "      <td>0.038690</td>\n",
       "      <td>-0.051081</td>\n",
       "      <td>-0.118063</td>\n",
       "      <td>-0.002818</td>\n",
       "      <td>0.042833</td>\n",
       "      <td>0.006652</td>\n",
       "      <td>0.030247</td>\n",
       "      <td>-0.063432</td>\n",
       "      <td>-0.036274</td>\n",
       "      <td>-0.067246</td>\n",
       "      <td>-0.035955</td>\n",
       "      <td>-0.081315</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>3</th>\n",
       "      <td>-0.172628</td>\n",
       "      <td>0.017743</td>\n",
       "      <td>-0.040178</td>\n",
       "      <td>-0.163020</td>\n",
       "      <td>0.046278</td>\n",
       "      <td>-0.090730</td>\n",
       "      <td>0.005195</td>\n",
       "      <td>0.072909</td>\n",
       "      <td>0.012898</td>\n",
       "      <td>0.113940</td>\n",
       "      <td>-0.123483</td>\n",
       "      <td>0.040234</td>\n",
       "      <td>-0.068481</td>\n",
       "      <td>-0.087080</td>\n",
       "      <td>-0.057885</td>\n",
       "      <td>-0.031155</td>\n",
       "      <td>0.005617</td>\n",
       "      <td>-0.116515</td>\n",
       "      <td>0.090083</td>\n",
       "      <td>0.046566</td>\n",
       "      <td>-0.153624</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>4</th>\n",
       "      <td>0.148825</td>\n",
       "      <td>0.189589</td>\n",
       "      <td>0.117878</td>\n",
       "      <td>-0.017992</td>\n",
       "      <td>-0.107075</td>\n",
       "      <td>0.105445</td>\n",
       "      <td>-0.040318</td>\n",
       "      <td>0.122245</td>\n",
       "      <td>0.020827</td>\n",
       "      <td>0.097664</td>\n",
       "      <td>0.035637</td>\n",
       "      <td>0.070657</td>\n",
       "      <td>0.001050</td>\n",
       "      <td>0.178587</td>\n",
       "      <td>0.012691</td>\n",
       "      <td>0.040199</td>\n",
       "      <td>0.188315</td>\n",
       "      <td>-0.134776</td>\n",
       "      <td>-0.127048</td>\n",
       "      <td>0.096940</td>\n",
       "      <td>-0.117312</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>...</th>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>665</th>\n",
       "      <td>-0.058381</td>\n",
       "      <td>-0.003250</td>\n",
       "      <td>0.020394</td>\n",
       "      <td>0.033251</td>\n",
       "      <td>0.132713</td>\n",
       "      <td>0.142915</td>\n",
       "      <td>0.161423</td>\n",
       "      <td>-0.025664</td>\n",
       "      <td>0.008917</td>\n",
       "      <td>0.107464</td>\n",
       "      <td>0.006117</td>\n",
       "      <td>0.138730</td>\n",
       "      <td>-0.231020</td>\n",
       "      <td>0.159364</td>\n",
       "      <td>-0.069079</td>\n",
       "      <td>-0.006949</td>\n",
       "      <td>-0.029728</td>\n",
       "      <td>0.083112</td>\n",
       "      <td>-0.017773</td>\n",
       "      <td>0.201848</td>\n",
       "      <td>0.056561</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>666</th>\n",
       "      <td>-0.020514</td>\n",
       "      <td>0.093294</td>\n",
       "      <td>0.062806</td>\n",
       "      <td>-0.017377</td>\n",
       "      <td>0.124928</td>\n",
       "      <td>0.084395</td>\n",
       "      <td>0.120722</td>\n",
       "      <td>-0.102591</td>\n",
       "      <td>-0.029740</td>\n",
       "      <td>-0.040436</td>\n",
       "      <td>0.084855</td>\n",
       "      <td>0.101054</td>\n",
       "      <td>0.101676</td>\n",
       "      <td>-0.034397</td>\n",
       "      <td>-0.070641</td>\n",
       "      <td>-0.030292</td>\n",
       "      <td>0.146353</td>\n",
       "      <td>0.180025</td>\n",
       "      <td>0.046512</td>\n",
       "      <td>-0.037158</td>\n",
       "      <td>-0.029711</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>667</th>\n",
       "      <td>0.035719</td>\n",
       "      <td>-0.267841</td>\n",
       "      <td>0.036494</td>\n",
       "      <td>-0.037777</td>\n",
       "      <td>-0.075906</td>\n",
       "      <td>-0.001517</td>\n",
       "      <td>-0.088212</td>\n",
       "      <td>0.059519</td>\n",
       "      <td>-0.008527</td>\n",
       "      <td>0.097502</td>\n",
       "      <td>-0.202646</td>\n",
       "      <td>0.032400</td>\n",
       "      <td>-0.154683</td>\n",
       "      <td>-0.060360</td>\n",
       "      <td>0.127331</td>\n",
       "      <td>-0.051708</td>\n",
       "      <td>0.138863</td>\n",
       "      <td>0.201289</td>\n",
       "      <td>0.068413</td>\n",
       "      <td>0.056427</td>\n",
       "      <td>-0.170783</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>668</th>\n",
       "      <td>-0.012207</td>\n",
       "      <td>0.080225</td>\n",
       "      <td>0.294022</td>\n",
       "      <td>-0.054400</td>\n",
       "      <td>0.054404</td>\n",
       "      <td>-0.030283</td>\n",
       "      <td>-0.012365</td>\n",
       "      <td>0.107111</td>\n",
       "      <td>0.017116</td>\n",
       "      <td>-0.116770</td>\n",
       "      <td>-0.159486</td>\n",
       "      <td>0.114986</td>\n",
       "      <td>0.170744</td>\n",
       "      <td>-0.179575</td>\n",
       "      <td>-0.031497</td>\n",
       "      <td>0.015683</td>\n",
       "      <td>-0.031149</td>\n",
       "      <td>0.002117</td>\n",
       "      <td>0.018902</td>\n",
       "      <td>0.095553</td>\n",
       "      <td>-0.027450</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>669</th>\n",
       "      <td>0.276624</td>\n",
       "      <td>0.041472</td>\n",
       "      <td>-0.040345</td>\n",
       "      <td>0.037136</td>\n",
       "      <td>0.036917</td>\n",
       "      <td>-0.081644</td>\n",
       "      <td>-0.004200</td>\n",
       "      <td>0.085010</td>\n",
       "      <td>0.041747</td>\n",
       "      <td>-0.046616</td>\n",
       "      <td>0.071059</td>\n",
       "      <td>-0.012790</td>\n",
       "      <td>0.035234</td>\n",
       "      <td>-0.020739</td>\n",
       "      <td>-0.119886</td>\n",
       "      <td>-0.003518</td>\n",
       "      <td>0.016737</td>\n",
       "      <td>0.216290</td>\n",
       "      <td>0.237209</td>\n",
       "      <td>0.057075</td>\n",
       "      <td>-0.128590</td>\n",
       "    </tr>\n",
       "  </tbody>\n",
       "</table>\n",
       "<p>670 rows × 21 columns</p>\n",
       "</div>"
      ],
      "text/plain": [
       "            K         N         T         R         S         I         M  \\\n",
       "0    0.176405  0.040016  0.097874  0.224089  0.186756 -0.097728  0.095009   \n",
       "1    0.065362  0.086444 -0.074217  0.226975 -0.145437  0.004576 -0.018718   \n",
       "2   -0.170627  0.195078 -0.050965 -0.043807 -0.125280  0.077749 -0.161390   \n",
       "3   -0.172628  0.017743 -0.040178 -0.163020  0.046278 -0.090730  0.005195   \n",
       "4    0.148825  0.189589  0.117878 -0.017992 -0.107075  0.105445 -0.040318   \n",
       "..        ...       ...       ...       ...       ...       ...       ...   \n",
       "665 -0.058381 -0.003250  0.020394  0.033251  0.132713  0.142915  0.161423   \n",
       "666 -0.020514  0.093294  0.062806 -0.017377  0.124928  0.084395  0.120722   \n",
       "667  0.035719 -0.267841  0.036494 -0.037777 -0.075906 -0.001517 -0.088212   \n",
       "668 -0.012207  0.080225  0.294022 -0.054400  0.054404 -0.030283 -0.012365   \n",
       "669  0.276624  0.041472 -0.040345  0.037136  0.036917 -0.081644 -0.004200   \n",
       "\n",
       "            Q         H         P         L         E         D         A  \\\n",
       "0   -0.015136 -0.010322  0.041060  0.014404  0.145427  0.076104  0.012168   \n",
       "1    0.153278  0.146936  0.015495  0.037816 -0.088779 -0.198080 -0.034791   \n",
       "2   -0.021274 -0.089547  0.038690 -0.051081 -0.118063 -0.002818  0.042833   \n",
       "3    0.072909  0.012898  0.113940 -0.123483  0.040234 -0.068481 -0.087080   \n",
       "4    0.122245  0.020827  0.097664  0.035637  0.070657  0.001050  0.178587   \n",
       "..        ...       ...       ...       ...       ...       ...       ...   \n",
       "665 -0.025664  0.008917  0.107464  0.006117  0.138730 -0.231020  0.159364   \n",
       "666 -0.102591 -0.029740 -0.040436  0.084855  0.101054  0.101676 -0.034397   \n",
       "667  0.059519 -0.008527  0.097502 -0.202646  0.032400 -0.154683 -0.060360   \n",
       "668  0.107111  0.017116 -0.116770 -0.159486  0.114986  0.170744 -0.179575   \n",
       "669  0.085010  0.041747 -0.046616  0.071059 -0.012790  0.035234 -0.020739   \n",
       "\n",
       "            G         V         *         Y         C         W         F  \n",
       "0    0.044386  0.033367  0.149408 -0.020516  0.031307 -0.085410 -0.255299  \n",
       "1    0.015635  0.123029  0.120238 -0.038733 -0.030230 -0.104855 -0.142002  \n",
       "2    0.006652  0.030247 -0.063432 -0.036274 -0.067246 -0.035955 -0.081315  \n",
       "3   -0.057885 -0.031155  0.005617 -0.116515  0.090083  0.046566 -0.153624  \n",
       "4    0.012691  0.040199  0.188315 -0.134776 -0.127048  0.096940 -0.117312  \n",
       "..        ...       ...       ...       ...       ...       ...       ...  \n",
       "665 -0.069079 -0.006949 -0.029728  0.083112 -0.017773  0.201848  0.056561  \n",
       "666 -0.070641 -0.030292  0.146353  0.180025  0.046512 -0.037158 -0.029711  \n",
       "667  0.127331 -0.051708  0.138863  0.201289  0.068413  0.056427 -0.170783  \n",
       "668 -0.031497  0.015683 -0.031149  0.002117  0.018902  0.095553 -0.027450  \n",
       "669 -0.119886 -0.003518  0.016737  0.216290  0.237209  0.057075 -0.128590  \n",
       "\n",
       "[670 rows x 21 columns]"
      ]
     },
     "execution_count": 4,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "# each entry of dataframe is one genotype/allele\n",
    "np.random.seed(0)\n",
    "df_selection = df_AA.copy().drop(['site','wildtype', 'total_counts'], axis=1)\n",
    "columns = df_selection.columns\n",
    "for index, row in df_selection.iterrows():\n",
    "   # print(row)\n",
    "    for column in columns:\n",
    "        df_selection.loc[index, column] = np.random.normal(0.0, 0.1, 1)[0]\n",
    "df_selection\n",
    "\n",
    "#fix seed"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "80419e02",
   "metadata": {},
   "outputs": [],
   "source": [
    "def true_trajectory(population_size, finite_sampling, generation, sample_index):\n",
    "    np.random.seed(10)\n",
    "    \n",
    "    trajectory = pd.DataFrame(columns = df_selection.columns)\n",
    "    trajectory = trajectory.T\n",
    "    select_list = df_selection.iloc[6]\n",
    "    trajectory['selection_coefficients'] = select_list+1\n",
    "    trajectory[0] = np.random.multinomial(population_size, df_sampling.iloc[sample_index].tolist())\n",
    "    for i in range(generation):\n",
    "        trajectory[i+1] = np.random.multinomial(population_size, \n",
    "                                                (((trajectory['selection_coefficients'])*trajectory[i])/((trajectory['selection_coefficients'])*trajectory[i]).sum()).values.tolist())\n",
    "\n",
    "    trajectory_save = trajectory[trajectory.columns[1:]]/population_size\n",
    "    # trajectory_save.to_csv('./output/simulation/WF_simulation.csv')\n",
    "    \n",
    "    return trajectory\n",
    "\n",
    "\n",
    "def other_methods(replicates, finite_sampling, generation_list, regularization=1):\n",
    "    inferred_log = {}\n",
    "    inferred_ratio = {}\n",
    "    inferred_regression = {}\n",
    "    inferred_selection = {}\n",
    "    for generation in generation_list:\n",
    "        inferred_log[generation] = []\n",
    "        inferred_ratio[generation] = []\n",
    "        inferred_regression[generation] = []\n",
    "        inferred_selection[generation] = []\n",
    "\n",
    "    for rep_num in range(replicates):        \n",
    "        plot_list_all=pd.read_csv('./output/simulation/WF_finite_sampling/'+'rep-%s.csv.zip' %rep_num, index_col=0)\n",
    "        plot_list_all = plot_list_all.T\n",
    "        regression_list = plot_list_all.copy().T\n",
    "        regression_list = np.log(regression_list)\n",
    "        regression_list = regression_list.T\n",
    "        regression_list['generation'] = [i for i in range(regression_list.shape[0])]\n",
    "        plot_list_all['generation'] = [i for i in range(regression_list.shape[0])]\n",
    "        # frequency change        \n",
    "        for generation in generation_list:\n",
    "            plot_list = plot_list_all.iloc[:generation]\n",
    "            # frequency change\n",
    "            freq_table = plot_list.drop('generation', axis=1).copy()\n",
    "\n",
    "            freq_diff = freq_table.iloc[-1]-freq_table.iloc[0]\n",
    "            freq_diff = np.array(freq_diff.tolist())\n",
    "\n",
    "            # covariance/ vector\n",
    "            cov_matx = np.zeros((21, 21))\n",
    "            for gen in range(freq_table.shape[0]-1):\n",
    "                variant_freq = freq_table.iloc[gen].tolist()\n",
    "                for i in range(21):\n",
    "                    cov_matx[i,i] += variant_freq[i]*(1-variant_freq[i])\n",
    "                    for j in range(i+1,21):\n",
    "                        cov_matx[i,j] -= variant_freq[i]*variant_freq[j]\n",
    "                        cov_matx[j,i] -= variant_freq[i]*variant_freq[j]\n",
    "                        \n",
    "            selection = inv(cov_matx + np.identity(21)*regularization).dot(freq_diff)\n",
    "            inferred_selection[generation].append(selection)\n",
    "\n",
    "            # ratio\n",
    "            raw = freq_table.iloc[-1]/freq_table.iloc[0]\n",
    "            inferred_ratio[generation].append([i/sum(raw) for i in raw])\n",
    "\n",
    "            # log enrich\n",
    "            inferred_log[generation].append([np.log(i/sum(raw)) for i in raw])\n",
    "\n",
    "            # log regression\n",
    "            df_regression = regression_list.iloc[:generation].copy()\n",
    "            temp_enrich = []\n",
    "            for i in df_regression.columns[:-1]:\n",
    "                Y = df_regression[i].values.reshape(-1, 1)\n",
    "                X = df_regression['generation'].values.reshape(-1, 1)\n",
    "                temp_enrich.append(linear_model.LinearRegression().fit(X,Y).coef_[0][0])\n",
    "            inferred_regression[generation].append(temp_enrich)\n",
    "        \n",
    "    print(\"Saving popDMS inference with regularization = %.1f\"%regularization)                   \n",
    "    for generation in generation_list:\n",
    "        df_enrichment_regress = pd.DataFrame(inferred_regression[generation])\n",
    "        df_enrichment_ratio = pd.DataFrame(inferred_ratio[generation])\n",
    "        df_enrichment_log = pd.DataFrame(inferred_log[generation])\n",
    "        df_selection = pd.DataFrame(inferred_selection[generation])\n",
    "        # df_enrichment_regress.to_csv('./output/simulation/WF_mutational_effects/'+'log_regression_gen-%s_'%(generation)+'sampling-%s'%finite_sampling+'.csv.zip', sep = ',', index = False, compression = 'zip')\n",
    "        # df_enrichment_ratio.to_csv('./output/simulation/WF_mutational_effects/'+'enrichment_ratio_gen-%s_'%(generation)+'sampling-%s'%finite_sampling+'.csv.zip', sep = ',', index = False, compression = 'zip')\n",
    "        # df_enrichment_log.to_csv('./output/simulation/WF_mutational_effects/'+'enrichment_ratio_log_gen-%s_'%(generation)+'sampling-%s'%finite_sampling+'.csv.zip', sep = ',', index = False, compression = 'zip')\n",
    "        # df_selection.to_csv('./output/simulation/WF_mutational_effects/'+'selection_coefficients_gen-%s_'%(generation)+'sampling-%s'%finite_sampling+'.csv.zip', sep = ',', index = False, compression = 'zip')\n",
    "\n",
    "population_size = 100000000\n",
    "finite_sampling = 50000\n",
    "generations = 30\n",
    "replicates = 100\n",
    "sample_index = 10\n",
    "generation_list = [2, 5, 10]\n",
    "opt_gen = 5\n",
    "Drop_percent = 20\n",
    "\n",
    "df_trajectory = true_trajectory(population_size, finite_sampling, generations, sample_index)\n",
    "other_methods(replicates, finite_sampling, generation_list)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "f436c001",
   "metadata": {},
   "outputs": [],
   "source": [
    "from matplotlib.lines import Line2D\n",
    "\n",
    "replicates = 100\n",
    "marker = ['o', 'v', '*']\n",
    "finite_list=[50000]\n",
    "plt.figure(figsize=[8,5])\n",
    "for finite_sampling in finite_list:\n",
    "    temp = [[],[],[],[],[]]\n",
    "    for generation in generation_list:\n",
    "\n",
    "        df_select = pd.read_csv('./output/simulation/WF_mutational_effects/selection_coefficients_gen-%s_'%generation+'sampling-%s'%finite_sampling+'.csv.zip', index_col=0)\n",
    "        df_enrichment_regress = pd.read_csv('./output/simulation/WF_mutational_effects/log_regression_gen-%s_'%generation+'sampling-%s'%finite_sampling+'.csv.zip', index_col=0)\n",
    "        df_enrichment_ratio = pd.read_csv('./output/simulation/WF_mutational_effects/enrichment_ratio_gen-%s_'%generation+'sampling-%s'%finite_sampling+'.csv.zip', index_col=0)\n",
    "        df_enrichment_ratio_log = pd.read_csv('./output/simulation/WF_mutational_effects/enrichment_ratio_log_gen-%s_'%generation+'sampling-%s'%finite_sampling+'.csv.zip', index_col=0)\n",
    "\n",
    "        enrichment_ratio_corr = df_enrichment_ratio.T.corr(method='pearson')\n",
    "        log_regression_corr = df_enrichment_regress.T.corr(method='pearson')\n",
    "        selection_coefficients_corr = df_select.T.corr(method='pearson')\n",
    "        enrichment_ratio_log_corr = df_enrichment_ratio_log.T.corr(method='pearson')\n",
    "        \n",
    "        enrichment_ratio_corr = (enrichment_ratio_corr.sum().sum() - replicates)/(replicates*replicates-replicates)\n",
    "        log_regression_corr = (log_regression_corr.sum().sum() - replicates)/(replicates*replicates-replicates)\n",
    "        selection_coefficients_corr = (selection_coefficients_corr.sum().sum() - replicates)/(replicates*replicates-replicates)  \n",
    "        enrichment_ratio_log_corr = (enrichment_ratio_log_corr.sum().sum() - replicates)/(replicates*replicates-replicates)\n",
    "        \n",
    "        temp[0].append(enrichment_ratio_corr)\n",
    "        temp[1].append(log_regression_corr)\n",
    "        temp[2].append(selection_coefficients_corr)\n",
    "        temp[3].append(enrichment_ratio_log_corr)\n",
    "\n",
    "    plt.plot(generation_list, temp[0], c='red', marker=marker[finite_list.index(finite_sampling)], markersize=10, alpha=0.6)\n",
    "    plt.plot(generation_list, temp[1], c='green', marker=marker[finite_list.index(finite_sampling)], markersize=10, alpha=0.6)\n",
    "    plt.plot(generation_list, temp[2], c='blue', marker=marker[finite_list.index(finite_sampling)], markersize=10, alpha=0.6)      \n",
    "    plt.plot(generation_list, temp[3], c='orange', marker=marker[finite_list.index(finite_sampling)], markersize=10, alpha=0.6)      \n",
    "    plt.xlabel('Generation length of simulation')\n",
    "    plt.ylabel('Pearson correlation coefficients')\n",
    "    plt.xticks([2,5,10])\n",
    "    \n",
    "legend_elements = [Line2D([0], [0], color='white', lw=0.1, label=r'$\\textbf{Methods}$'),\n",
    "                   Line2D([0], [0], color='blue', lw=4, label='selection coefficients, Lambda=1', alpha=0.6),\n",
    "                   Line2D([0], [0], color='red', lw=4, label='enrichment ratio'),\n",
    "                   Line2D([0], [0], color='orange', lw=4, label='enrichment ratio log scaled', alpha=0.6),\n",
    "                   Line2D([0], [0], color='green', lw=4, label='log regression'),\n",
    "#                    Line2D([0], [0], color='white', lw=0.1, label=r'$\\textbf{Finite sampling size}$'),\n",
    "#                    Line2D([0], [0], marker='*', color='grey', label='100000', markersize=10),\n",
    "#                    Line2D([0], [0], marker='v', color='grey', label='50000', markersize=10),\n",
    "#                    Line2D([0], [0], marker='o', color='grey', label='10000', markersize=10)\n",
    "                  ]\n",
    "plt.legend(handles=legend_elements)\n",
    "plt.title('Consistency of the methods across 100 replicates')\n"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "62972ce3",
   "metadata": {},
   "outputs": [],
   "source": []
  }
 ],
 "metadata": {
  "kernelspec": {
   "display_name": "Python 3 (ipykernel)",
   "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.10.13"
  }
 },
 "nbformat": 4,
 "nbformat_minor": 5
}