{
 "cells": [
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "# Livestock Module Example"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "### Imports \n",
    "\n",
    "We will start by importing ```AnimalData``` and the ```Exports``` classes, along with a few others."
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 1,
   "metadata": {},
   "outputs": [],
   "source": [
    "from livestock_generation.livestock import AnimalData\n",
    "from livestock_generation.livestock_exports import Exports\n",
    "import pandas as pd\n",
    "import os"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "Now, we set the path to our data and we will import our scenario dataframe"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 2,
   "metadata": {},
   "outputs": [],
   "source": [
    "path = \"../tests/data/\"\n",
    "\n",
    "scenario_dataframe = pd.read_csv(os.path.join(path, \"scenario_input_dataframe.csv\"))"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "Our Scenario Data frame looks like this ... "
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 3,
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "   index  Scenarios  Dairy pop  Beef pop  Upland sheep pop  Lowland sheep pop  \\\n",
      "0      0          0    1060000         0                 0                  0   \n",
      "1      1          0          0      1000                 0                  0   \n",
      "2      2          0          0         0              3000                  0   \n",
      "3      3          0          0         0                 0              50000   \n",
      "4      4          1    1060000         0                 0                  0   \n",
      "5      5          1          0      1000                 0                  0   \n",
      "6      6          1          0         0                 0                  0   \n",
      "7      7          1          0         0                 0                  0   \n",
      "\n",
      "   Dairy prod  Beef prod  Upland sheep prod  Lowland sheep prod  ...  \\\n",
      "0           0          0                  0                   0  ...   \n",
      "1           0          0                  0                   0  ...   \n",
      "2           0          0                  0                   0  ...   \n",
      "3           0          0                  0                   0  ...   \n",
      "4           0          0                  0                   0  ...   \n",
      "5           0          0                  0                   0  ...   \n",
      "6           0          0                  0                   0  ...   \n",
      "7           0          0                  0                   0  ...   \n",
      "\n",
      "  Crop area  Wetland area  Forest area  Conifer proportion  \\\n",
      "0         0             0            1                 0.7   \n",
      "1         0             0            1                 0.7   \n",
      "2         0             0            1                 0.7   \n",
      "3         0             0            1                 0.7   \n",
      "4         0             0            1                 0.7   \n",
      "5         0             0            1                 0.7   \n",
      "6         0             0            1                 0.7   \n",
      "7         0             0            1                 0.7   \n",
      "\n",
      "   Broadleaf proportion  Conifer harvest  Broadleaf harvest  Conifer thinned  \\\n",
      "0                   0.3             0.05                  0              0.1   \n",
      "1                   0.3             0.05                  0              0.1   \n",
      "2                   0.3             0.05                  0              0.1   \n",
      "3                   0.3             0.05                  0              0.1   \n",
      "4                   0.3             0.05                  0              0.8   \n",
      "5                   0.3             0.05                  0              0.8   \n",
      "6                   0.3             0.05                  0              0.8   \n",
      "7                   0.3             0.05                  0              0.8   \n",
      "\n",
      "   Afforest year  Cattle systems  \n",
      "0           2080           Dairy  \n",
      "1           2080            Beef  \n",
      "2           2080    Upland sheep  \n",
      "3           2080   Lowland sheep  \n",
      "4           2080           Dairy  \n",
      "5           2080            Beef  \n",
      "6           2080    Upland sheep  \n",
      "7           2080   Lowland sheep  \n",
      "\n",
      "[8 rows x 29 columns]\n"
     ]
    }
   ],
   "source": [
    "print(scenario_dataframe)"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "## Define additional variables\n",
    "\n",
    "We also need to define the ```calibration_year```, which is the year reference year, the ```target_year```, which is the year the scenarios run to, and the ```ef_country```, which is the emissions factor country. "
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": []
  },
  {
   "cell_type": "code",
   "execution_count": 4,
   "metadata": {},
   "outputs": [],
   "source": [
    "baseline_year = 2020\n",
    "target_year = 2050\n",
    "ef_country = \"ireland\""
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "## Class instances \n",
    "\n",
    "Now we set our classes, and use the methods to generate our data. "
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 5,
   "metadata": {},
   "outputs": [],
   "source": [
    "animal_class = AnimalData(ef_country, baseline_year, target_year, scenario_dataframe)\n",
    "export_class = Exports(ef_country, baseline_year, target_year, scenario_dataframe)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 6,
   "metadata": {},
   "outputs": [],
   "source": [
    "# create dataframe for baseline year animals\n",
    "baseline_data = animal_class.create_baseline_animal_dataframe()"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 7,
   "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>ef_country</th>\n",
       "      <th>farm_id</th>\n",
       "      <th>Scenarios</th>\n",
       "      <th>year</th>\n",
       "      <th>cohort</th>\n",
       "      <th>pop</th>\n",
       "      <th>daily_milk</th>\n",
       "      <th>weight</th>\n",
       "      <th>forage</th>\n",
       "      <th>grazing</th>\n",
       "      <th>con_type</th>\n",
       "      <th>con_amount</th>\n",
       "      <th>wool</th>\n",
       "      <th>t_outdoors</th>\n",
       "      <th>t_indoors</th>\n",
       "      <th>t_stabled</th>\n",
       "      <th>mm_storage</th>\n",
       "      <th>daily_spreading</th>\n",
       "      <th>n_sold</th>\n",
       "      <th>n_bought</th>\n",
       "    </tr>\n",
       "  </thead>\n",
       "  <tbody>\n",
       "    <tr>\n",
       "      <th>0</th>\n",
       "      <td>ireland</td>\n",
       "      <td>2020.0</td>\n",
       "      <td>-1.0</td>\n",
       "      <td>2020.0</td>\n",
       "      <td>dairy_cows</td>\n",
       "      <td>1.511850e+06</td>\n",
       "      <td>14.953000</td>\n",
       "      <td>538.000</td>\n",
       "      <td>irish_grass</td>\n",
       "      <td>pasture</td>\n",
       "      <td>concentrate</td>\n",
       "      <td>2.992828</td>\n",
       "      <td>0.0</td>\n",
       "      <td>13.589041</td>\n",
       "      <td>10.410959</td>\n",
       "      <td>0.0</td>\n",
       "      <td>tank liquid</td>\n",
       "      <td>broadcast</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>1</th>\n",
       "      <td>ireland</td>\n",
       "      <td>2020.0</td>\n",
       "      <td>-1.0</td>\n",
       "      <td>2020.0</td>\n",
       "      <td>suckler_cows</td>\n",
       "      <td>9.530000e+05</td>\n",
       "      <td>1.410959</td>\n",
       "      <td>600.000</td>\n",
       "      <td>irish_grass</td>\n",
       "      <td>pasture</td>\n",
       "      <td>concentrate</td>\n",
       "      <td>0.842752</td>\n",
       "      <td>0.0</td>\n",
       "      <td>12.219178</td>\n",
       "      <td>11.780822</td>\n",
       "      <td>0.0</td>\n",
       "      <td>tank liquid</td>\n",
       "      <td>broadcast</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>2</th>\n",
       "      <td>ireland</td>\n",
       "      <td>2020.0</td>\n",
       "      <td>-1.0</td>\n",
       "      <td>2020.0</td>\n",
       "      <td>bulls</td>\n",
       "      <td>5.095000e+04</td>\n",
       "      <td>0.000000</td>\n",
       "      <td>773.000</td>\n",
       "      <td>irish_grass</td>\n",
       "      <td>pasture</td>\n",
       "      <td>concentrate</td>\n",
       "      <td>0.654141</td>\n",
       "      <td>0.0</td>\n",
       "      <td>11.506849</td>\n",
       "      <td>12.493151</td>\n",
       "      <td>0.0</td>\n",
       "      <td>tank liquid</td>\n",
       "      <td>broadcast</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>3</th>\n",
       "      <td>ireland</td>\n",
       "      <td>2020.0</td>\n",
       "      <td>-1.0</td>\n",
       "      <td>2020.0</td>\n",
       "      <td>DxD_calves_m</td>\n",
       "      <td>2.486034e+05</td>\n",
       "      <td>0.000000</td>\n",
       "      <td>171.400</td>\n",
       "      <td>irish_grass</td>\n",
       "      <td>pasture</td>\n",
       "      <td>concentrate</td>\n",
       "      <td>1.000000</td>\n",
       "      <td>0.0</td>\n",
       "      <td>7.890411</td>\n",
       "      <td>16.109589</td>\n",
       "      <td>0.0</td>\n",
       "      <td>tank liquid</td>\n",
       "      <td>broadcast</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>4</th>\n",
       "      <td>ireland</td>\n",
       "      <td>2020.0</td>\n",
       "      <td>-1.0</td>\n",
       "      <td>2020.0</td>\n",
       "      <td>DxD_calves_f</td>\n",
       "      <td>3.537366e+05</td>\n",
       "      <td>0.000000</td>\n",
       "      <td>138.550</td>\n",
       "      <td>irish_grass</td>\n",
       "      <td>pasture</td>\n",
       "      <td>concentrate</td>\n",
       "      <td>1.000000</td>\n",
       "      <td>0.0</td>\n",
       "      <td>7.890411</td>\n",
       "      <td>16.109589</td>\n",
       "      <td>0.0</td>\n",
       "      <td>tank liquid</td>\n",
       "      <td>broadcast</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>5</th>\n",
       "      <td>ireland</td>\n",
       "      <td>2020.0</td>\n",
       "      <td>-1.0</td>\n",
       "      <td>2020.0</td>\n",
       "      <td>DxB_calves_m</td>\n",
       "      <td>3.201623e+05</td>\n",
       "      <td>0.000000</td>\n",
       "      <td>176.875</td>\n",
       "      <td>irish_grass</td>\n",
       "      <td>pasture</td>\n",
       "      <td>concentrate</td>\n",
       "      <td>1.000000</td>\n",
       "      <td>0.0</td>\n",
       "      <td>7.890411</td>\n",
       "      <td>16.109589</td>\n",
       "      <td>0.0</td>\n",
       "      <td>tank liquid</td>\n",
       "      <td>broadcast</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>6</th>\n",
       "      <td>ireland</td>\n",
       "      <td>2020.0</td>\n",
       "      <td>-1.0</td>\n",
       "      <td>2020.0</td>\n",
       "      <td>DxB_calves_f</td>\n",
       "      <td>3.189379e+05</td>\n",
       "      <td>0.000000</td>\n",
       "      <td>158.625</td>\n",
       "      <td>irish_grass</td>\n",
       "      <td>pasture</td>\n",
       "      <td>concentrate</td>\n",
       "      <td>1.000000</td>\n",
       "      <td>0.0</td>\n",
       "      <td>7.890411</td>\n",
       "      <td>16.109589</td>\n",
       "      <td>0.0</td>\n",
       "      <td>tank liquid</td>\n",
       "      <td>broadcast</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>7</th>\n",
       "      <td>ireland</td>\n",
       "      <td>2020.0</td>\n",
       "      <td>-1.0</td>\n",
       "      <td>2020.0</td>\n",
       "      <td>BxB_calves_m</td>\n",
       "      <td>4.059952e+05</td>\n",
       "      <td>0.000000</td>\n",
       "      <td>211.550</td>\n",
       "      <td>irish_grass</td>\n",
       "      <td>pasture</td>\n",
       "      <td>concentrate</td>\n",
       "      <td>1.000000</td>\n",
       "      <td>0.0</td>\n",
       "      <td>7.890411</td>\n",
       "      <td>16.109589</td>\n",
       "      <td>0.0</td>\n",
       "      <td>tank liquid</td>\n",
       "      <td>broadcast</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>8</th>\n",
       "      <td>ireland</td>\n",
       "      <td>2020.0</td>\n",
       "      <td>-1.0</td>\n",
       "      <td>2020.0</td>\n",
       "      <td>BxB_calves_f</td>\n",
       "      <td>4.041513e+05</td>\n",
       "      <td>0.000000</td>\n",
       "      <td>175.050</td>\n",
       "      <td>irish_grass</td>\n",
       "      <td>pasture</td>\n",
       "      <td>concentrate</td>\n",
       "      <td>1.000000</td>\n",
       "      <td>0.0</td>\n",
       "      <td>7.890411</td>\n",
       "      <td>16.109589</td>\n",
       "      <td>0.0</td>\n",
       "      <td>tank liquid</td>\n",
       "      <td>broadcast</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>9</th>\n",
       "      <td>ireland</td>\n",
       "      <td>2020.0</td>\n",
       "      <td>-1.0</td>\n",
       "      <td>2020.0</td>\n",
       "      <td>DxD_heifers_less_2_yr</td>\n",
       "      <td>3.465882e+05</td>\n",
       "      <td>0.000000</td>\n",
       "      <td>335.650</td>\n",
       "      <td>irish_grass</td>\n",
       "      <td>pasture</td>\n",
       "      <td>concentrate</td>\n",
       "      <td>0.000000</td>\n",
       "      <td>0.0</td>\n",
       "      <td>11.506849</td>\n",
       "      <td>12.493151</td>\n",
       "      <td>0.0</td>\n",
       "      <td>tank liquid</td>\n",
       "      <td>broadcast</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>10</th>\n",
       "      <td>ireland</td>\n",
       "      <td>2020.0</td>\n",
       "      <td>-1.0</td>\n",
       "      <td>2020.0</td>\n",
       "      <td>DxD_steers_less_2_yr</td>\n",
       "      <td>2.394694e+05</td>\n",
       "      <td>0.000000</td>\n",
       "      <td>434.200</td>\n",
       "      <td>irish_grass</td>\n",
       "      <td>pasture</td>\n",
       "      <td>concentrate</td>\n",
       "      <td>0.000000</td>\n",
       "      <td>0.0</td>\n",
       "      <td>18.465753</td>\n",
       "      <td>5.534247</td>\n",
       "      <td>0.0</td>\n",
       "      <td>tank liquid</td>\n",
       "      <td>broadcast</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>11</th>\n",
       "      <td>ireland</td>\n",
       "      <td>2020.0</td>\n",
       "      <td>-1.0</td>\n",
       "      <td>2020.0</td>\n",
       "      <td>DxB_heifers_less_2_yr</td>\n",
       "      <td>3.083219e+05</td>\n",
       "      <td>0.000000</td>\n",
       "      <td>395.875</td>\n",
       "      <td>irish_grass</td>\n",
       "      <td>pasture</td>\n",
       "      <td>concentrate</td>\n",
       "      <td>0.000000</td>\n",
       "      <td>0.0</td>\n",
       "      <td>11.506849</td>\n",
       "      <td>12.493151</td>\n",
       "      <td>0.0</td>\n",
       "      <td>tank liquid</td>\n",
       "      <td>broadcast</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>12</th>\n",
       "      <td>ireland</td>\n",
       "      <td>2020.0</td>\n",
       "      <td>-1.0</td>\n",
       "      <td>2020.0</td>\n",
       "      <td>DxB_steers_less_2_yr</td>\n",
       "      <td>3.092563e+05</td>\n",
       "      <td>0.000000</td>\n",
       "      <td>450.625</td>\n",
       "      <td>irish_grass</td>\n",
       "      <td>pasture</td>\n",
       "      <td>concentrate</td>\n",
       "      <td>0.000000</td>\n",
       "      <td>0.0</td>\n",
       "      <td>18.465753</td>\n",
       "      <td>5.534247</td>\n",
       "      <td>0.0</td>\n",
       "      <td>tank liquid</td>\n",
       "      <td>broadcast</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>13</th>\n",
       "      <td>ireland</td>\n",
       "      <td>2020.0</td>\n",
       "      <td>-1.0</td>\n",
       "      <td>2020.0</td>\n",
       "      <td>BxB_heifers_less_2_yr</td>\n",
       "      <td>3.908495e+05</td>\n",
       "      <td>0.000000</td>\n",
       "      <td>445.150</td>\n",
       "      <td>irish_grass</td>\n",
       "      <td>pasture</td>\n",
       "      <td>concentrate</td>\n",
       "      <td>0.000000</td>\n",
       "      <td>0.0</td>\n",
       "      <td>11.506849</td>\n",
       "      <td>12.493151</td>\n",
       "      <td>0.0</td>\n",
       "      <td>tank liquid</td>\n",
       "      <td>broadcast</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>14</th>\n",
       "      <td>ireland</td>\n",
       "      <td>2020.0</td>\n",
       "      <td>-1.0</td>\n",
       "      <td>2020.0</td>\n",
       "      <td>BxB_steers_less_2_yr</td>\n",
       "      <td>3.923286e+05</td>\n",
       "      <td>0.000000</td>\n",
       "      <td>554.650</td>\n",
       "      <td>irish_grass</td>\n",
       "      <td>pasture</td>\n",
       "      <td>concentrate</td>\n",
       "      <td>0.000000</td>\n",
       "      <td>0.0</td>\n",
       "      <td>18.465753</td>\n",
       "      <td>5.534247</td>\n",
       "      <td>0.0</td>\n",
       "      <td>tank liquid</td>\n",
       "      <td>broadcast</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>15</th>\n",
       "      <td>ireland</td>\n",
       "      <td>2020.0</td>\n",
       "      <td>-1.0</td>\n",
       "      <td>2020.0</td>\n",
       "      <td>DxD_heifers_more_2_yr</td>\n",
       "      <td>1.046702e+04</td>\n",
       "      <td>0.000000</td>\n",
       "      <td>481.650</td>\n",
       "      <td>irish_grass</td>\n",
       "      <td>pasture</td>\n",
       "      <td>concentrate</td>\n",
       "      <td>0.000000</td>\n",
       "      <td>0.0</td>\n",
       "      <td>12.765090</td>\n",
       "      <td>11.234910</td>\n",
       "      <td>0.0</td>\n",
       "      <td>tank liquid</td>\n",
       "      <td>broadcast</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>16</th>\n",
       "      <td>ireland</td>\n",
       "      <td>2020.0</td>\n",
       "      <td>-1.0</td>\n",
       "      <td>2020.0</td>\n",
       "      <td>DxD_steers_more_2_yr</td>\n",
       "      <td>5.823657e+04</td>\n",
       "      <td>0.000000</td>\n",
       "      <td>592.975</td>\n",
       "      <td>irish_grass</td>\n",
       "      <td>pasture</td>\n",
       "      <td>concentrate</td>\n",
       "      <td>0.000000</td>\n",
       "      <td>0.0</td>\n",
       "      <td>18.465753</td>\n",
       "      <td>5.534247</td>\n",
       "      <td>0.0</td>\n",
       "      <td>tank liquid</td>\n",
       "      <td>broadcast</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>17</th>\n",
       "      <td>ireland</td>\n",
       "      <td>2020.0</td>\n",
       "      <td>-1.0</td>\n",
       "      <td>2020.0</td>\n",
       "      <td>DxB_heifers_more_2_yr</td>\n",
       "      <td>2.021479e+04</td>\n",
       "      <td>0.000000</td>\n",
       "      <td>527.275</td>\n",
       "      <td>irish_grass</td>\n",
       "      <td>pasture</td>\n",
       "      <td>concentrate</td>\n",
       "      <td>0.000000</td>\n",
       "      <td>0.0</td>\n",
       "      <td>12.765090</td>\n",
       "      <td>11.234910</td>\n",
       "      <td>0.0</td>\n",
       "      <td>tank liquid</td>\n",
       "      <td>broadcast</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>18</th>\n",
       "      <td>ireland</td>\n",
       "      <td>2020.0</td>\n",
       "      <td>-1.0</td>\n",
       "      <td>2020.0</td>\n",
       "      <td>DxB_steers_more_2_yr</td>\n",
       "      <td>7.121757e+04</td>\n",
       "      <td>0.000000</td>\n",
       "      <td>624.000</td>\n",
       "      <td>irish_grass</td>\n",
       "      <td>pasture</td>\n",
       "      <td>concentrate</td>\n",
       "      <td>0.000000</td>\n",
       "      <td>0.0</td>\n",
       "      <td>18.465753</td>\n",
       "      <td>5.534247</td>\n",
       "      <td>0.0</td>\n",
       "      <td>tank liquid</td>\n",
       "      <td>broadcast</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>19</th>\n",
       "      <td>ireland</td>\n",
       "      <td>2020.0</td>\n",
       "      <td>-1.0</td>\n",
       "      <td>2020.0</td>\n",
       "      <td>BxB_heifers_more_2_yr</td>\n",
       "      <td>5.656799e+04</td>\n",
       "      <td>0.000000</td>\n",
       "      <td>603.925</td>\n",
       "      <td>irish_grass</td>\n",
       "      <td>pasture</td>\n",
       "      <td>concentrate</td>\n",
       "      <td>0.000000</td>\n",
       "      <td>0.0</td>\n",
       "      <td>12.765090</td>\n",
       "      <td>11.234910</td>\n",
       "      <td>0.0</td>\n",
       "      <td>tank liquid</td>\n",
       "      <td>broadcast</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>20</th>\n",
       "      <td>ireland</td>\n",
       "      <td>2020.0</td>\n",
       "      <td>-1.0</td>\n",
       "      <td>2020.0</td>\n",
       "      <td>BxB_steers_more_2_yr</td>\n",
       "      <td>6.266367e+04</td>\n",
       "      <td>0.000000</td>\n",
       "      <td>751.750</td>\n",
       "      <td>irish_grass</td>\n",
       "      <td>pasture</td>\n",
       "      <td>concentrate</td>\n",
       "      <td>0.000000</td>\n",
       "      <td>0.0</td>\n",
       "      <td>18.465753</td>\n",
       "      <td>5.534247</td>\n",
       "      <td>0.0</td>\n",
       "      <td>tank liquid</td>\n",
       "      <td>broadcast</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>21</th>\n",
       "      <td>ireland</td>\n",
       "      <td>2020.0</td>\n",
       "      <td>-1.0</td>\n",
       "      <td>2020.0</td>\n",
       "      <td>ewes</td>\n",
       "      <td>2.137040e+06</td>\n",
       "      <td>0.000000</td>\n",
       "      <td>68.000</td>\n",
       "      <td>average</td>\n",
       "      <td>flat_pasture</td>\n",
       "      <td>concentrate</td>\n",
       "      <td>0.000000</td>\n",
       "      <td>4.5</td>\n",
       "      <td>21.360000</td>\n",
       "      <td>2.640000</td>\n",
       "      <td>0.0</td>\n",
       "      <td>solid</td>\n",
       "      <td>broadcast</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>22</th>\n",
       "      <td>ireland</td>\n",
       "      <td>2020.0</td>\n",
       "      <td>-1.0</td>\n",
       "      <td>2020.0</td>\n",
       "      <td>ewes</td>\n",
       "      <td>5.342600e+05</td>\n",
       "      <td>0.000000</td>\n",
       "      <td>68.000</td>\n",
       "      <td>average</td>\n",
       "      <td>hilly_pasture</td>\n",
       "      <td>concentrate</td>\n",
       "      <td>0.000000</td>\n",
       "      <td>4.5</td>\n",
       "      <td>21.360000</td>\n",
       "      <td>2.640000</td>\n",
       "      <td>0.0</td>\n",
       "      <td>solid</td>\n",
       "      <td>broadcast</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>23</th>\n",
       "      <td>ireland</td>\n",
       "      <td>2020.0</td>\n",
       "      <td>-1.0</td>\n",
       "      <td>2020.0</td>\n",
       "      <td>lamb_less_1_yr</td>\n",
       "      <td>9.603852e+05</td>\n",
       "      <td>0.000000</td>\n",
       "      <td>37.000</td>\n",
       "      <td>average</td>\n",
       "      <td>flat_pasture</td>\n",
       "      <td>concentrate</td>\n",
       "      <td>0.000000</td>\n",
       "      <td>4.5</td>\n",
       "      <td>21.360000</td>\n",
       "      <td>2.640000</td>\n",
       "      <td>0.0</td>\n",
       "      <td>solid</td>\n",
       "      <td>broadcast</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>24</th>\n",
       "      <td>ireland</td>\n",
       "      <td>2020.0</td>\n",
       "      <td>-1.0</td>\n",
       "      <td>2020.0</td>\n",
       "      <td>lamb_less_1_yr</td>\n",
       "      <td>2.400963e+05</td>\n",
       "      <td>0.000000</td>\n",
       "      <td>37.000</td>\n",
       "      <td>average</td>\n",
       "      <td>hilly_pasture</td>\n",
       "      <td>concentrate</td>\n",
       "      <td>0.000000</td>\n",
       "      <td>4.5</td>\n",
       "      <td>21.360000</td>\n",
       "      <td>2.640000</td>\n",
       "      <td>0.0</td>\n",
       "      <td>solid</td>\n",
       "      <td>broadcast</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>25</th>\n",
       "      <td>ireland</td>\n",
       "      <td>2020.0</td>\n",
       "      <td>-1.0</td>\n",
       "      <td>2020.0</td>\n",
       "      <td>lamb_more_1_yr</td>\n",
       "      <td>1.027081e+05</td>\n",
       "      <td>0.000000</td>\n",
       "      <td>68.000</td>\n",
       "      <td>average</td>\n",
       "      <td>flat_pasture</td>\n",
       "      <td>concentrate</td>\n",
       "      <td>0.000000</td>\n",
       "      <td>4.5</td>\n",
       "      <td>21.360000</td>\n",
       "      <td>2.640000</td>\n",
       "      <td>0.0</td>\n",
       "      <td>solid</td>\n",
       "      <td>broadcast</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>26</th>\n",
       "      <td>ireland</td>\n",
       "      <td>2020.0</td>\n",
       "      <td>-1.0</td>\n",
       "      <td>2020.0</td>\n",
       "      <td>lamb_more_1_yr</td>\n",
       "      <td>2.567702e+04</td>\n",
       "      <td>0.000000</td>\n",
       "      <td>68.000</td>\n",
       "      <td>average</td>\n",
       "      <td>hilly_pasture</td>\n",
       "      <td>concentrate</td>\n",
       "      <td>0.000000</td>\n",
       "      <td>4.5</td>\n",
       "      <td>21.360000</td>\n",
       "      <td>2.640000</td>\n",
       "      <td>0.0</td>\n",
       "      <td>solid</td>\n",
       "      <td>broadcast</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>27</th>\n",
       "      <td>ireland</td>\n",
       "      <td>2020.0</td>\n",
       "      <td>-1.0</td>\n",
       "      <td>2020.0</td>\n",
       "      <td>male_less_1_yr</td>\n",
       "      <td>2.400963e+05</td>\n",
       "      <td>0.000000</td>\n",
       "      <td>37.000</td>\n",
       "      <td>average</td>\n",
       "      <td>hilly_pasture</td>\n",
       "      <td>concentrate</td>\n",
       "      <td>0.000000</td>\n",
       "      <td>4.5</td>\n",
       "      <td>21.360000</td>\n",
       "      <td>2.640000</td>\n",
       "      <td>0.0</td>\n",
       "      <td>solid</td>\n",
       "      <td>broadcast</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>28</th>\n",
       "      <td>ireland</td>\n",
       "      <td>2020.0</td>\n",
       "      <td>-1.0</td>\n",
       "      <td>2020.0</td>\n",
       "      <td>male_less_1_yr</td>\n",
       "      <td>9.603852e+05</td>\n",
       "      <td>0.000000</td>\n",
       "      <td>37.000</td>\n",
       "      <td>average</td>\n",
       "      <td>flat_pasture</td>\n",
       "      <td>concentrate</td>\n",
       "      <td>0.000000</td>\n",
       "      <td>4.5</td>\n",
       "      <td>21.360000</td>\n",
       "      <td>2.640000</td>\n",
       "      <td>0.0</td>\n",
       "      <td>solid</td>\n",
       "      <td>broadcast</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>29</th>\n",
       "      <td>ireland</td>\n",
       "      <td>2020.0</td>\n",
       "      <td>-1.0</td>\n",
       "      <td>2020.0</td>\n",
       "      <td>ram</td>\n",
       "      <td>6.876000e+04</td>\n",
       "      <td>0.000000</td>\n",
       "      <td>86.000</td>\n",
       "      <td>average</td>\n",
       "      <td>flat_pasture</td>\n",
       "      <td>concentrate</td>\n",
       "      <td>0.000000</td>\n",
       "      <td>4.5</td>\n",
       "      <td>21.360000</td>\n",
       "      <td>2.640000</td>\n",
       "      <td>0.0</td>\n",
       "      <td>solid</td>\n",
       "      <td>broadcast</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>30</th>\n",
       "      <td>ireland</td>\n",
       "      <td>2020.0</td>\n",
       "      <td>-1.0</td>\n",
       "      <td>2020.0</td>\n",
       "      <td>ram</td>\n",
       "      <td>1.719000e+04</td>\n",
       "      <td>0.000000</td>\n",
       "      <td>86.000</td>\n",
       "      <td>average</td>\n",
       "      <td>hilly_pasture</td>\n",
       "      <td>concentrate</td>\n",
       "      <td>0.000000</td>\n",
       "      <td>4.5</td>\n",
       "      <td>21.360000</td>\n",
       "      <td>2.640000</td>\n",
       "      <td>0.0</td>\n",
       "      <td>solid</td>\n",
       "      <td>broadcast</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0.0</td>\n",
       "    </tr>\n",
       "  </tbody>\n",
       "</table>\n",
       "</div>"
      ],
      "text/plain": [
       "   ef_country  farm_id  Scenarios    year                 cohort  \\\n",
       "0     ireland   2020.0       -1.0  2020.0             dairy_cows   \n",
       "1     ireland   2020.0       -1.0  2020.0           suckler_cows   \n",
       "2     ireland   2020.0       -1.0  2020.0                  bulls   \n",
       "3     ireland   2020.0       -1.0  2020.0           DxD_calves_m   \n",
       "4     ireland   2020.0       -1.0  2020.0           DxD_calves_f   \n",
       "5     ireland   2020.0       -1.0  2020.0           DxB_calves_m   \n",
       "6     ireland   2020.0       -1.0  2020.0           DxB_calves_f   \n",
       "7     ireland   2020.0       -1.0  2020.0           BxB_calves_m   \n",
       "8     ireland   2020.0       -1.0  2020.0           BxB_calves_f   \n",
       "9     ireland   2020.0       -1.0  2020.0  DxD_heifers_less_2_yr   \n",
       "10    ireland   2020.0       -1.0  2020.0   DxD_steers_less_2_yr   \n",
       "11    ireland   2020.0       -1.0  2020.0  DxB_heifers_less_2_yr   \n",
       "12    ireland   2020.0       -1.0  2020.0   DxB_steers_less_2_yr   \n",
       "13    ireland   2020.0       -1.0  2020.0  BxB_heifers_less_2_yr   \n",
       "14    ireland   2020.0       -1.0  2020.0   BxB_steers_less_2_yr   \n",
       "15    ireland   2020.0       -1.0  2020.0  DxD_heifers_more_2_yr   \n",
       "16    ireland   2020.0       -1.0  2020.0   DxD_steers_more_2_yr   \n",
       "17    ireland   2020.0       -1.0  2020.0  DxB_heifers_more_2_yr   \n",
       "18    ireland   2020.0       -1.0  2020.0   DxB_steers_more_2_yr   \n",
       "19    ireland   2020.0       -1.0  2020.0  BxB_heifers_more_2_yr   \n",
       "20    ireland   2020.0       -1.0  2020.0   BxB_steers_more_2_yr   \n",
       "21    ireland   2020.0       -1.0  2020.0                   ewes   \n",
       "22    ireland   2020.0       -1.0  2020.0                   ewes   \n",
       "23    ireland   2020.0       -1.0  2020.0         lamb_less_1_yr   \n",
       "24    ireland   2020.0       -1.0  2020.0         lamb_less_1_yr   \n",
       "25    ireland   2020.0       -1.0  2020.0         lamb_more_1_yr   \n",
       "26    ireland   2020.0       -1.0  2020.0         lamb_more_1_yr   \n",
       "27    ireland   2020.0       -1.0  2020.0         male_less_1_yr   \n",
       "28    ireland   2020.0       -1.0  2020.0         male_less_1_yr   \n",
       "29    ireland   2020.0       -1.0  2020.0                    ram   \n",
       "30    ireland   2020.0       -1.0  2020.0                    ram   \n",
       "\n",
       "             pop  daily_milk   weight       forage        grazing  \\\n",
       "0   1.511850e+06   14.953000  538.000  irish_grass        pasture   \n",
       "1   9.530000e+05    1.410959  600.000  irish_grass        pasture   \n",
       "2   5.095000e+04    0.000000  773.000  irish_grass        pasture   \n",
       "3   2.486034e+05    0.000000  171.400  irish_grass        pasture   \n",
       "4   3.537366e+05    0.000000  138.550  irish_grass        pasture   \n",
       "5   3.201623e+05    0.000000  176.875  irish_grass        pasture   \n",
       "6   3.189379e+05    0.000000  158.625  irish_grass        pasture   \n",
       "7   4.059952e+05    0.000000  211.550  irish_grass        pasture   \n",
       "8   4.041513e+05    0.000000  175.050  irish_grass        pasture   \n",
       "9   3.465882e+05    0.000000  335.650  irish_grass        pasture   \n",
       "10  2.394694e+05    0.000000  434.200  irish_grass        pasture   \n",
       "11  3.083219e+05    0.000000  395.875  irish_grass        pasture   \n",
       "12  3.092563e+05    0.000000  450.625  irish_grass        pasture   \n",
       "13  3.908495e+05    0.000000  445.150  irish_grass        pasture   \n",
       "14  3.923286e+05    0.000000  554.650  irish_grass        pasture   \n",
       "15  1.046702e+04    0.000000  481.650  irish_grass        pasture   \n",
       "16  5.823657e+04    0.000000  592.975  irish_grass        pasture   \n",
       "17  2.021479e+04    0.000000  527.275  irish_grass        pasture   \n",
       "18  7.121757e+04    0.000000  624.000  irish_grass        pasture   \n",
       "19  5.656799e+04    0.000000  603.925  irish_grass        pasture   \n",
       "20  6.266367e+04    0.000000  751.750  irish_grass        pasture   \n",
       "21  2.137040e+06    0.000000   68.000      average   flat_pasture   \n",
       "22  5.342600e+05    0.000000   68.000      average  hilly_pasture   \n",
       "23  9.603852e+05    0.000000   37.000      average   flat_pasture   \n",
       "24  2.400963e+05    0.000000   37.000      average  hilly_pasture   \n",
       "25  1.027081e+05    0.000000   68.000      average   flat_pasture   \n",
       "26  2.567702e+04    0.000000   68.000      average  hilly_pasture   \n",
       "27  2.400963e+05    0.000000   37.000      average  hilly_pasture   \n",
       "28  9.603852e+05    0.000000   37.000      average   flat_pasture   \n",
       "29  6.876000e+04    0.000000   86.000      average   flat_pasture   \n",
       "30  1.719000e+04    0.000000   86.000      average  hilly_pasture   \n",
       "\n",
       "       con_type  con_amount  wool  t_outdoors  t_indoors  t_stabled  \\\n",
       "0   concentrate    2.992828   0.0   13.589041  10.410959        0.0   \n",
       "1   concentrate    0.842752   0.0   12.219178  11.780822        0.0   \n",
       "2   concentrate    0.654141   0.0   11.506849  12.493151        0.0   \n",
       "3   concentrate    1.000000   0.0    7.890411  16.109589        0.0   \n",
       "4   concentrate    1.000000   0.0    7.890411  16.109589        0.0   \n",
       "5   concentrate    1.000000   0.0    7.890411  16.109589        0.0   \n",
       "6   concentrate    1.000000   0.0    7.890411  16.109589        0.0   \n",
       "7   concentrate    1.000000   0.0    7.890411  16.109589        0.0   \n",
       "8   concentrate    1.000000   0.0    7.890411  16.109589        0.0   \n",
       "9   concentrate    0.000000   0.0   11.506849  12.493151        0.0   \n",
       "10  concentrate    0.000000   0.0   18.465753   5.534247        0.0   \n",
       "11  concentrate    0.000000   0.0   11.506849  12.493151        0.0   \n",
       "12  concentrate    0.000000   0.0   18.465753   5.534247        0.0   \n",
       "13  concentrate    0.000000   0.0   11.506849  12.493151        0.0   \n",
       "14  concentrate    0.000000   0.0   18.465753   5.534247        0.0   \n",
       "15  concentrate    0.000000   0.0   12.765090  11.234910        0.0   \n",
       "16  concentrate    0.000000   0.0   18.465753   5.534247        0.0   \n",
       "17  concentrate    0.000000   0.0   12.765090  11.234910        0.0   \n",
       "18  concentrate    0.000000   0.0   18.465753   5.534247        0.0   \n",
       "19  concentrate    0.000000   0.0   12.765090  11.234910        0.0   \n",
       "20  concentrate    0.000000   0.0   18.465753   5.534247        0.0   \n",
       "21  concentrate    0.000000   4.5   21.360000   2.640000        0.0   \n",
       "22  concentrate    0.000000   4.5   21.360000   2.640000        0.0   \n",
       "23  concentrate    0.000000   4.5   21.360000   2.640000        0.0   \n",
       "24  concentrate    0.000000   4.5   21.360000   2.640000        0.0   \n",
       "25  concentrate    0.000000   4.5   21.360000   2.640000        0.0   \n",
       "26  concentrate    0.000000   4.5   21.360000   2.640000        0.0   \n",
       "27  concentrate    0.000000   4.5   21.360000   2.640000        0.0   \n",
       "28  concentrate    0.000000   4.5   21.360000   2.640000        0.0   \n",
       "29  concentrate    0.000000   4.5   21.360000   2.640000        0.0   \n",
       "30  concentrate    0.000000   4.5   21.360000   2.640000        0.0   \n",
       "\n",
       "     mm_storage daily_spreading  n_sold  n_bought  \n",
       "0   tank liquid       broadcast     0.0       0.0  \n",
       "1   tank liquid       broadcast     0.0       0.0  \n",
       "2   tank liquid       broadcast     0.0       0.0  \n",
       "3   tank liquid       broadcast     0.0       0.0  \n",
       "4   tank liquid       broadcast     0.0       0.0  \n",
       "5   tank liquid       broadcast     0.0       0.0  \n",
       "6   tank liquid       broadcast     0.0       0.0  \n",
       "7   tank liquid       broadcast     0.0       0.0  \n",
       "8   tank liquid       broadcast     0.0       0.0  \n",
       "9   tank liquid       broadcast     0.0       0.0  \n",
       "10  tank liquid       broadcast     0.0       0.0  \n",
       "11  tank liquid       broadcast     0.0       0.0  \n",
       "12  tank liquid       broadcast     0.0       0.0  \n",
       "13  tank liquid       broadcast     0.0       0.0  \n",
       "14  tank liquid       broadcast     0.0       0.0  \n",
       "15  tank liquid       broadcast     0.0       0.0  \n",
       "16  tank liquid       broadcast     0.0       0.0  \n",
       "17  tank liquid       broadcast     0.0       0.0  \n",
       "18  tank liquid       broadcast     0.0       0.0  \n",
       "19  tank liquid       broadcast     0.0       0.0  \n",
       "20  tank liquid       broadcast     0.0       0.0  \n",
       "21        solid       broadcast     0.0       0.0  \n",
       "22        solid       broadcast     0.0       0.0  \n",
       "23        solid       broadcast     0.0       0.0  \n",
       "24        solid       broadcast     0.0       0.0  \n",
       "25        solid       broadcast     0.0       0.0  \n",
       "26        solid       broadcast     0.0       0.0  \n",
       "27        solid       broadcast     0.0       0.0  \n",
       "28        solid       broadcast     0.0       0.0  \n",
       "29        solid       broadcast     0.0       0.0  \n",
       "30        solid       broadcast     0.0       0.0  "
      ]
     },
     "execution_count": 7,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "baseline_data"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 8,
   "metadata": {},
   "outputs": [],
   "source": [
    "# create dataframe for scenarios animals\n",
    "scenario_data = animal_class.create_animal_dataframe()"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 9,
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/html": [
       "<div>\n",
       "<style scoped>\n",
       "    .dataframe tbody tr th:only-of-type {\n",
       "        vertical-align: middle;\n",
       "    }\n",
       "\n",
       "    .dataframe tbody tr th {\n",
       "        vertical-align: top;\n",
       "    }\n",
       "\n",
       "    .dataframe thead th {\n",
       "        text-align: right;\n",
       "    }\n",
       "</style>\n",
       "<table border=\"1\" class=\"dataframe\">\n",
       "  <thead>\n",
       "    <tr style=\"text-align: right;\">\n",
       "      <th></th>\n",
       "      <th>ef_country</th>\n",
       "      <th>farm_id</th>\n",
       "      <th>Scenarios</th>\n",
       "      <th>year</th>\n",
       "      <th>cohort</th>\n",
       "      <th>pop</th>\n",
       "      <th>daily_milk</th>\n",
       "      <th>weight</th>\n",
       "      <th>forage</th>\n",
       "      <th>grazing</th>\n",
       "      <th>con_type</th>\n",
       "      <th>con_amount</th>\n",
       "      <th>wool</th>\n",
       "      <th>t_outdoors</th>\n",
       "      <th>t_indoors</th>\n",
       "      <th>t_stabled</th>\n",
       "      <th>mm_storage</th>\n",
       "      <th>daily_spreading</th>\n",
       "      <th>n_sold</th>\n",
       "      <th>n_bought</th>\n",
       "    </tr>\n",
       "  </thead>\n",
       "  <tbody>\n",
       "    <tr>\n",
       "      <th>0</th>\n",
       "      <td>ireland</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0.0</td>\n",
       "      <td>2050.0</td>\n",
       "      <td>dairy_cows</td>\n",
       "      <td>1.060000e+06</td>\n",
       "      <td>13.800000</td>\n",
       "      <td>538.00</td>\n",
       "      <td>irish_grass</td>\n",
       "      <td>pasture</td>\n",
       "      <td>concentrate</td>\n",
       "      <td>2.762056</td>\n",
       "      <td>0.0</td>\n",
       "      <td>13.589041</td>\n",
       "      <td>10.410959</td>\n",
       "      <td>0.0</td>\n",
       "      <td>tank liquid</td>\n",
       "      <td>broadcast</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>1</th>\n",
       "      <td>ireland</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0.0</td>\n",
       "      <td>2050.0</td>\n",
       "      <td>suckler_cows</td>\n",
       "      <td>0.000000e+00</td>\n",
       "      <td>1.410959</td>\n",
       "      <td>600.00</td>\n",
       "      <td>irish_grass</td>\n",
       "      <td>pasture</td>\n",
       "      <td>concentrate</td>\n",
       "      <td>0.842752</td>\n",
       "      <td>0.0</td>\n",
       "      <td>12.219178</td>\n",
       "      <td>11.780822</td>\n",
       "      <td>0.0</td>\n",
       "      <td>tank liquid</td>\n",
       "      <td>broadcast</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>2</th>\n",
       "      <td>ireland</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0.0</td>\n",
       "      <td>2050.0</td>\n",
       "      <td>bulls</td>\n",
       "      <td>2.191087e+04</td>\n",
       "      <td>0.000000</td>\n",
       "      <td>773.00</td>\n",
       "      <td>irish_grass</td>\n",
       "      <td>pasture</td>\n",
       "      <td>concentrate</td>\n",
       "      <td>0.654141</td>\n",
       "      <td>0.0</td>\n",
       "      <td>11.506849</td>\n",
       "      <td>12.493151</td>\n",
       "      <td>0.0</td>\n",
       "      <td>tank liquid</td>\n",
       "      <td>broadcast</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>3</th>\n",
       "      <td>ireland</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0.0</td>\n",
       "      <td>2050.0</td>\n",
       "      <td>DxD_calves_m</td>\n",
       "      <td>1.743027e+05</td>\n",
       "      <td>0.000000</td>\n",
       "      <td>171.40</td>\n",
       "      <td>irish_grass</td>\n",
       "      <td>pasture</td>\n",
       "      <td>concentrate</td>\n",
       "      <td>1.000000</td>\n",
       "      <td>0.0</td>\n",
       "      <td>7.890411</td>\n",
       "      <td>16.109589</td>\n",
       "      <td>0.0</td>\n",
       "      <td>tank liquid</td>\n",
       "      <td>broadcast</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>4</th>\n",
       "      <td>ireland</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0.0</td>\n",
       "      <td>2050.0</td>\n",
       "      <td>DxD_calves_f</td>\n",
       "      <td>2.480145e+05</td>\n",
       "      <td>0.000000</td>\n",
       "      <td>138.55</td>\n",
       "      <td>irish_grass</td>\n",
       "      <td>pasture</td>\n",
       "      <td>concentrate</td>\n",
       "      <td>1.000000</td>\n",
       "      <td>0.0</td>\n",
       "      <td>7.890411</td>\n",
       "      <td>16.109589</td>\n",
       "      <td>0.0</td>\n",
       "      <td>tank liquid</td>\n",
       "      <td>broadcast</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0.0</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",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>119</th>\n",
       "      <td>ireland</td>\n",
       "      <td>7.0</td>\n",
       "      <td>1.0</td>\n",
       "      <td>2050.0</td>\n",
       "      <td>lamb_more_1_yr</td>\n",
       "      <td>0.000000e+00</td>\n",
       "      <td>0.000000</td>\n",
       "      <td>68.00</td>\n",
       "      <td>average</td>\n",
       "      <td>hilly_pasture</td>\n",
       "      <td>concentrate</td>\n",
       "      <td>0.000000</td>\n",
       "      <td>4.5</td>\n",
       "      <td>21.360000</td>\n",
       "      <td>2.640000</td>\n",
       "      <td>0.0</td>\n",
       "      <td>solid</td>\n",
       "      <td>broadcast</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>120</th>\n",
       "      <td>ireland</td>\n",
       "      <td>7.0</td>\n",
       "      <td>1.0</td>\n",
       "      <td>2050.0</td>\n",
       "      <td>male_less_1_yr</td>\n",
       "      <td>0.000000e+00</td>\n",
       "      <td>0.000000</td>\n",
       "      <td>37.00</td>\n",
       "      <td>average</td>\n",
       "      <td>hilly_pasture</td>\n",
       "      <td>concentrate</td>\n",
       "      <td>0.000000</td>\n",
       "      <td>4.5</td>\n",
       "      <td>21.360000</td>\n",
       "      <td>2.640000</td>\n",
       "      <td>0.0</td>\n",
       "      <td>solid</td>\n",
       "      <td>broadcast</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>121</th>\n",
       "      <td>ireland</td>\n",
       "      <td>7.0</td>\n",
       "      <td>1.0</td>\n",
       "      <td>2050.0</td>\n",
       "      <td>male_less_1_yr</td>\n",
       "      <td>0.000000e+00</td>\n",
       "      <td>0.000000</td>\n",
       "      <td>37.00</td>\n",
       "      <td>average</td>\n",
       "      <td>flat_pasture</td>\n",
       "      <td>concentrate</td>\n",
       "      <td>0.000000</td>\n",
       "      <td>4.5</td>\n",
       "      <td>21.360000</td>\n",
       "      <td>2.640000</td>\n",
       "      <td>0.0</td>\n",
       "      <td>solid</td>\n",
       "      <td>broadcast</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>122</th>\n",
       "      <td>ireland</td>\n",
       "      <td>7.0</td>\n",
       "      <td>1.0</td>\n",
       "      <td>2050.0</td>\n",
       "      <td>ram</td>\n",
       "      <td>0.000000e+00</td>\n",
       "      <td>0.000000</td>\n",
       "      <td>86.00</td>\n",
       "      <td>average</td>\n",
       "      <td>flat_pasture</td>\n",
       "      <td>concentrate</td>\n",
       "      <td>0.000000</td>\n",
       "      <td>4.5</td>\n",
       "      <td>21.360000</td>\n",
       "      <td>2.640000</td>\n",
       "      <td>0.0</td>\n",
       "      <td>solid</td>\n",
       "      <td>broadcast</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>123</th>\n",
       "      <td>ireland</td>\n",
       "      <td>7.0</td>\n",
       "      <td>1.0</td>\n",
       "      <td>2050.0</td>\n",
       "      <td>ram</td>\n",
       "      <td>0.000000e+00</td>\n",
       "      <td>0.000000</td>\n",
       "      <td>86.00</td>\n",
       "      <td>average</td>\n",
       "      <td>hilly_pasture</td>\n",
       "      <td>concentrate</td>\n",
       "      <td>0.000000</td>\n",
       "      <td>4.5</td>\n",
       "      <td>21.360000</td>\n",
       "      <td>2.640000</td>\n",
       "      <td>0.0</td>\n",
       "      <td>solid</td>\n",
       "      <td>broadcast</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0.0</td>\n",
       "    </tr>\n",
       "  </tbody>\n",
       "</table>\n",
       "<p>124 rows × 20 columns</p>\n",
       "</div>"
      ],
      "text/plain": [
       "    ef_country  farm_id  Scenarios    year          cohort           pop  \\\n",
       "0      ireland      0.0        0.0  2050.0      dairy_cows  1.060000e+06   \n",
       "1      ireland      0.0        0.0  2050.0    suckler_cows  0.000000e+00   \n",
       "2      ireland      0.0        0.0  2050.0           bulls  2.191087e+04   \n",
       "3      ireland      0.0        0.0  2050.0    DxD_calves_m  1.743027e+05   \n",
       "4      ireland      0.0        0.0  2050.0    DxD_calves_f  2.480145e+05   \n",
       "..         ...      ...        ...     ...             ...           ...   \n",
       "119    ireland      7.0        1.0  2050.0  lamb_more_1_yr  0.000000e+00   \n",
       "120    ireland      7.0        1.0  2050.0  male_less_1_yr  0.000000e+00   \n",
       "121    ireland      7.0        1.0  2050.0  male_less_1_yr  0.000000e+00   \n",
       "122    ireland      7.0        1.0  2050.0             ram  0.000000e+00   \n",
       "123    ireland      7.0        1.0  2050.0             ram  0.000000e+00   \n",
       "\n",
       "     daily_milk  weight       forage        grazing     con_type  con_amount  \\\n",
       "0     13.800000  538.00  irish_grass        pasture  concentrate    2.762056   \n",
       "1      1.410959  600.00  irish_grass        pasture  concentrate    0.842752   \n",
       "2      0.000000  773.00  irish_grass        pasture  concentrate    0.654141   \n",
       "3      0.000000  171.40  irish_grass        pasture  concentrate    1.000000   \n",
       "4      0.000000  138.55  irish_grass        pasture  concentrate    1.000000   \n",
       "..          ...     ...          ...            ...          ...         ...   \n",
       "119    0.000000   68.00      average  hilly_pasture  concentrate    0.000000   \n",
       "120    0.000000   37.00      average  hilly_pasture  concentrate    0.000000   \n",
       "121    0.000000   37.00      average   flat_pasture  concentrate    0.000000   \n",
       "122    0.000000   86.00      average   flat_pasture  concentrate    0.000000   \n",
       "123    0.000000   86.00      average  hilly_pasture  concentrate    0.000000   \n",
       "\n",
       "     wool  t_outdoors  t_indoors  t_stabled   mm_storage daily_spreading  \\\n",
       "0     0.0   13.589041  10.410959        0.0  tank liquid       broadcast   \n",
       "1     0.0   12.219178  11.780822        0.0  tank liquid       broadcast   \n",
       "2     0.0   11.506849  12.493151        0.0  tank liquid       broadcast   \n",
       "3     0.0    7.890411  16.109589        0.0  tank liquid       broadcast   \n",
       "4     0.0    7.890411  16.109589        0.0  tank liquid       broadcast   \n",
       "..    ...         ...        ...        ...          ...             ...   \n",
       "119   4.5   21.360000   2.640000        0.0        solid       broadcast   \n",
       "120   4.5   21.360000   2.640000        0.0        solid       broadcast   \n",
       "121   4.5   21.360000   2.640000        0.0        solid       broadcast   \n",
       "122   4.5   21.360000   2.640000        0.0        solid       broadcast   \n",
       "123   4.5   21.360000   2.640000        0.0        solid       broadcast   \n",
       "\n",
       "     n_sold  n_bought  \n",
       "0       0.0       0.0  \n",
       "1       0.0       0.0  \n",
       "2       0.0       0.0  \n",
       "3       0.0       0.0  \n",
       "4       0.0       0.0  \n",
       "..      ...       ...  \n",
       "119     0.0       0.0  \n",
       "120     0.0       0.0  \n",
       "121     0.0       0.0  \n",
       "122     0.0       0.0  \n",
       "123     0.0       0.0  \n",
       "\n",
       "[124 rows x 20 columns]"
      ]
     },
     "execution_count": 9,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "scenario_data"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 10,
   "metadata": {},
   "outputs": [],
   "source": [
    "# Create dataframes for beef and milk output\n",
    "beef_outputs = export_class.compute_system_protien_exports(scenario_data, baseline_data)\n",
    "\n",
    "dairy_outputs = export_class.compute_system_milk_exports(scenario_data, baseline_data)\n",
    "\n",
    "protein_outputs = export_class.compute_system_total_protein_exports(scenario_data, baseline_data)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 11,
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "     Scenarios carcass_weight_kg            DxD_m            DxD_f  \\\n",
      "-1.0      -1.0  652803287.105261    77397063.2663  95540884.900078   \n",
      " 0.0       0.0  256991209.428068  54265229.395958  66986366.368411   \n",
      " 1.0       1.0  256991209.428068  54265229.395958  66986366.368411   \n",
      "\n",
      "                 DxB_m            DxB_f             BxB_m             BxB_f  \n",
      "-1.0  106421442.081876  86751315.047085  158882596.689614  127809985.120308  \n",
      " 0.0   74615027.024366  60823754.969018      166718.35959     134113.310724  \n",
      " 1.0   74615027.024366  60823754.969018      166718.35959     134113.310724  \n"
     ]
    }
   ],
   "source": [
    "print(beef_outputs)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 12,
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "     Scenarios  total_milk_kg\n",
      "-1.0      -1.0  8251442963.25\n",
      " 0.0       0.0   5339220000.0\n",
      " 1.0       1.0   5339220000.0\n"
     ]
    }
   ],
   "source": [
    "print(dairy_outputs)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 13,
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "         total_protein     milk_protein     beef_protein\n",
      "-1.0   438945259.74796  288800503.71375  150144756.03421\n",
      " 0.0  245980678.168456      186872700.0  59107978.168456\n",
      " 1.0  245980678.168456      186872700.0  59107978.168456\n"
     ]
    }
   ],
   "source": [
    "print(protein_outputs)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": []
  }
 ],
 "metadata": {
  "kernelspec": {
   "display_name": ".venv",
   "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.12"
  }
 },
 "nbformat": 4,
 "nbformat_minor": 2
}