LHC-Variance-Contribution-Breakdown.ipynb

{
 "cells": [
  {
   "attachments": {},
   "cell_type": "markdown",
   "id": "physical-design",
   "metadata": {},
   "source": [
    "# Variance contribution figures\n",
    "\n",
    "In this notebook, I attribute the variance seen in prices to different parameters. The gameplan is to perform regression on prices against parameter at each point in time, and plot the ratio of the variance attributable (i.e., the $r^2$ value of each regression) in time. The parameter with the highest $r^2$ value at each time is said to \"drive\" the prices at that time.\n",
    "\n",
    "This notebook is a little sloppy, but I tried to clean it up the best I could."
   ]
  },
  {
   "cell_type": "markdown",
   "id": "automated-chase",
   "metadata": {},
   "source": [
    "## Import necessary packages and set plotting params"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "raising-potter",
   "metadata": {},
   "outputs": [],
   "source": [
    "import sys\n",
    "import datetime\n",
    "import os\n",
    "\n",
    "import numpy as np\n",
    "import matplotlib.pyplot as plt \n",
    "import matplotlib as mpl\n",
    "import seaborn as sns\n",
    "\n",
    "# plotting parameters\n",
    "color_list = ['#000000', '#E69F00', '#56B4E9', '#009E73', '#F0E442', '#0072B2', '#D55E00', '#CC79A7'] * 2\n",
    "marker_list = ['o', 's', 'P', '*', '<', 'v', '>', '^', 'p']\n",
    "markersize = 6\n",
    "linestyle_list = ['solid', 'solid', 'solid', 'solid'] * 4\n",
    "linewidth = 2\n",
    "node_times = [2020, 2030, 2060, 2100, 2150, 2200]\n",
    "x_label = 'Year'\n",
    "\n",
    "%matplotlib inline\n",
    "%config InlineBackend.figure_format = 'retina' \n",
    "cdds_params={'axes.linewidth': 3,\n",
    " 'axes.axisbelow': False,\n",
    " 'axes.edgecolor': 'black',\n",
    " 'axes.facecolor': 'None',\n",
    " 'axes.grid': False,\n",
    " 'axes.labelcolor': 'black',\n",
    " 'axes.spines.right': False,\n",
    " 'axes.spines.top': False,\n",
    " 'axes.titlesize': 20,\n",
    " 'axes.labelsize': 20,\n",
    " 'axes.titlelocation': 'left',\n",
    " 'figure.facecolor': 'white',\n",
    " 'figure.figsize': (18, 10),\n",
    " 'lines.solid_capstyle': 'round',\n",
    " 'lines.linewidth': 2.5,\n",
    " 'patch.edgecolor': 'w',\n",
    " 'patch.force_edgecolor': True,\n",
    " 'text.color': 'black',\n",
    " 'legend.frameon': False,\n",
    " 'xtick.bottom': True,\n",
    " 'xtick.major.width': 3,\n",
    " 'xtick.major.size': 6,\n",
    " 'xtick.color': 'black',\n",
    " 'xtick.direction': 'out',\n",
    " 'xtick.top': False,\n",
    " 'ytick.color': 'black',\n",
    " 'ytick.direction': 'out',\n",
    " 'ytick.left': True,\n",
    " 'ytick.right': False,\n",
    " 'ytick.color' : 'black',\n",
    " 'ytick.major.width': 3, \n",
    " 'ytick.major.size': 6,\n",
    " 'axes.prop_cycle': plt.cycler(linestyle=linestyle_list),\n",
    " 'font.size': 16,\n",
    " 'font.family': 'serif'}\n",
    "plt.rcParams.update(cdds_params)\n",
    "\n",
    "# make base filename\n",
    "today = datetime.datetime.now()\n",
    "year = str(today.year)\n",
    "day = str(today.day)\n",
    "month = str(today.month)\n",
    "\n",
    "basefile = ''.join([month, '-', day, '-', year, '-'])\n",
    "\n",
    "ssp5_cind = 0\n",
    "ssp4_cind = 6\n",
    "ssp3_cind = 7\n",
    "ssp2_cind = 2\n",
    "ssp1_cind = 3\n",
    "alpha = 0.2"
   ]
  },
  {
   "cell_type": "markdown",
   "id": "greatest-delaware",
   "metadata": {},
   "source": [
    "# Save figs?!"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "imported-country",
   "metadata": {},
   "outputs": [],
   "source": [
    "save_fig = False"
   ]
  },
  {
   "cell_type": "markdown",
   "id": "thirty-portland",
   "metadata": {},
   "source": [
    "## Import data\n",
    "\n",
    "Also, remove `nan` values and runs with too high utility (these tend to be unphysical)."
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "adjustable-advocacy",
   "metadata": {},
   "outputs": [],
   "source": [
    "def remove_nans(ssp_paths_and_util, exp_m, temp, co2, dam, params):\n",
    "    c2020, c2030, c2060, c2100, c2150, c2200, util = ssp_paths_and_util.T\n",
    "    paths = ssp_paths_and_util[:, :-1]\n",
    "    unphys_util_vals = util > 15\n",
    "    c2020 = c2020[~unphys_util_vals]\n",
    "    c2030 = c2030[~unphys_util_vals]\n",
    "    c2060 = c2060[~unphys_util_vals]\n",
    "    c2100 = c2100[~unphys_util_vals]\n",
    "    c2150 = c2150[~unphys_util_vals]\n",
    "    c2200 = c2200[~unphys_util_vals]\n",
    "    util = util[~unphys_util_vals]\n",
    "    paths = paths[~unphys_util_vals]\n",
    "    params = params[~unphys_util_vals]\n",
    "    exp_m = exp_m[~unphys_util_vals]\n",
    "    temp = temp[~unphys_util_vals]\n",
    "    co2 = co2[~unphys_util_vals]\n",
    "    dam = dam[~unphys_util_vals]\n",
    "    return paths[~np.isnan(util)], c2020[~np.isnan(util)], c2030[~np.isnan(util)], c2060[~np.isnan(util)], c2100[~np.isnan(util)], c2150[~np.isnan(util)], c2200[~np.isnan(util)], util[~np.isnan(util)], params[~np.isnan(util)], exp_m[~np.isnan(util)], temp[~np.isnan(util)], co2[~np.isnan(util)], dam[~np.isnan(util)]\n",
    "\n",
    "path = os.getcwd()\n",
    "\n",
    "# sampled parameter values\n",
    "params = np.genfromtxt(''.join([path, '/data/LHC_samples_N1000_DIMS5.csv']), delimiter=',')\n",
    "\n",
    "# pull data from diff files \n",
    "ssp5_paths_and_util = np.genfromtxt(''.join([path, '/data/lhc_sampling_paths_ssp5_N1000.csv']), delimiter=',')\n",
    "ssp5_exp_m = np.genfromtxt(''.join([path, '/data/lhc_sampling_exp_m_ssp5_N1000.csv']), delimiter=',')\n",
    "ssp5_exp_t = np.genfromtxt(''.join([path, '/data/lhc_sampling_exp_T_ssp5_N1000.csv']), delimiter=',')\n",
    "ssp5_exp_co2 = np.genfromtxt(''.join([path, '/data/lhc_sampling_exp_conc_ssp5_N1000.csv']), delimiter=',')\n",
    "ssp5_exp_dam = np.genfromtxt(''.join([path, '/data/lhc_sampling_exp_dam_ssp5_N1000.csv']), delimiter=',')\n",
    "ssp5_paths, ssp5_c2020, ssp5_c2030, ssp5_c2060, ssp5_c2100, ssp5_c2150, ssp5_c2200, ssp5_util,\\\n",
    "    ssp5_params, ssp5_exp_m, ssp5_exp_t, ssp5_exp_co2, ssp5_exp_dam = remove_nans(ssp5_paths_and_util, \n",
    "                                                                                  ssp5_exp_m, ssp5_exp_t, \n",
    "                                                                                  ssp5_exp_co2, ssp5_exp_dam, \n",
    "                                                                                  params)\n",
    "ssp5_ra, ssp5_eis, ssp5_ex, ssp5_end, ssp5_prtp = ssp5_params.T\n",
    "\n",
    "ssp4_paths_and_util = np.genfromtxt(''.join([path, '/data/lhc_sampling_paths_ssp4_N1000.csv']), delimiter=',')\n",
    "ssp4_exp_m = np.genfromtxt(''.join([path, '/data/lhc_sampling_exp_m_ssp4_N1000.csv']), delimiter=',')\n",
    "ssp4_exp_t = np.genfromtxt(''.join([path, '/data/lhc_sampling_exp_T_ssp4_N1000.csv']), delimiter=',')\n",
    "ssp4_exp_co2 = np.genfromtxt(''.join([path, '/data/lhc_sampling_exp_conc_ssp4_N1000.csv']), delimiter=',')\n",
    "ssp4_exp_dam = np.genfromtxt(''.join([path, '/data/lhc_sampling_exp_dam_ssp4_N1000.csv']), delimiter=',')\n",
    "ssp4_paths, ssp4_c2020, ssp4_c2030, ssp4_c2060, ssp4_c2100, ssp4_c2150, ssp4_c2200, ssp4_util,\\\n",
    "    ssp4_params, ssp4_exp_m, ssp4_exp_t, ssp4_exp_co2, ssp4_exp_dam = remove_nans(ssp4_paths_and_util, \n",
    "                                                                                  ssp4_exp_m, ssp4_exp_t, \n",
    "                                                                                  ssp4_exp_co2, ssp4_exp_dam, \n",
    "                                                                                  params)\n",
    "ssp4_ra, ssp4_eis, ssp4_ex, ssp4_end, ssp4_prtp = ssp4_params.T\n",
    "\n",
    "ssp3_paths_and_util = np.genfromtxt(''.join([path, '/data/lhc_sampling_paths_ssp3_N1000.csv']), delimiter=',')\n",
    "ssp3_exp_m = np.genfromtxt(''.join([path, '/data/lhc_sampling_exp_m_ssp3_N1000.csv']), delimiter=',')\n",
    "ssp3_exp_t = np.genfromtxt(''.join([path, '/data/lhc_sampling_exp_T_ssp3_N1000.csv']), delimiter=',')\n",
    "ssp3_exp_co2 = np.genfromtxt(''.join([path, '/data/lhc_sampling_exp_conc_ssp3_N1000.csv']), delimiter=',')\n",
    "ssp3_exp_dam = np.genfromtxt(''.join([path, '/data/lhc_sampling_exp_dam_ssp3_N1000.csv']), delimiter=',')\n",
    "ssp3_paths, ssp3_c2020, ssp3_c2030, ssp3_c2060, ssp3_c2100, ssp3_c2150, ssp3_c2200, ssp3_util,\\\n",
    "    ssp3_params, ssp3_exp_m, ssp3_exp_t, ssp3_exp_co2, ssp3_exp_dam = remove_nans(ssp3_paths_and_util, \n",
    "                                                                                  ssp3_exp_m, ssp3_exp_t, \n",
    "                                                                                  ssp3_exp_co2, ssp3_exp_dam, \n",
    "                                                                                  params)\n",
    "ssp3_ra, ssp3_eis, ssp3_ex, ssp3_end, ssp3_prtp = ssp3_params.T\n",
    "\n",
    "ssp2_paths_and_util = np.genfromtxt(''.join([path, '/data/lhc_sampling_paths_ssp2_N1000.csv']), delimiter=',')\n",
    "ssp2_exp_m = np.genfromtxt(''.join([path, '/data/lhc_sampling_exp_m_ssp2_N1000.csv']), delimiter=',')\n",
    "ssp2_exp_t = np.genfromtxt(''.join([path, '/data/lhc_sampling_exp_T_ssp2_N1000.csv']), delimiter=',')\n",
    "ssp2_exp_co2 = np.genfromtxt(''.join([path, '/data/lhc_sampling_exp_conc_ssp2_N1000.csv']), delimiter=',')\n",
    "ssp2_exp_dam = np.genfromtxt(''.join([path, '/data/lhc_sampling_exp_dam_ssp2_N1000.csv']), delimiter=',')\n",
    "ssp2_paths, ssp2_c2020, ssp2_c2030, ssp2_c2060, ssp2_c2100, ssp2_c2150, ssp2_c2200, ssp2_util,\\\n",
    "    ssp2_params, ssp2_exp_m, ssp2_exp_t, ssp2_exp_co2, ssp2_exp_dam = remove_nans(ssp2_paths_and_util, \n",
    "                                                                                  ssp2_exp_m, ssp2_exp_t, \n",
    "                                                                                  ssp2_exp_co2, ssp2_exp_dam, \n",
    "                                                                                  params)\n",
    "ssp2_ra, ssp2_eis, ssp2_ex, ssp2_end, ssp2_prtp = ssp2_params.T\n",
    "\n",
    "ssp1_paths_and_util = np.genfromtxt(''.join([path, '/data/lhc_sampling_paths_ssp1_N1000.csv']), delimiter=',')\n",
    "ssp1_exp_m = np.genfromtxt(''.join([path, '/data/lhc_sampling_exp_m_ssp1_N1000.csv']), delimiter=',')\n",
    "ssp1_exp_t = np.genfromtxt(''.join([path, '/data/lhc_sampling_exp_T_ssp1_N1000.csv']), delimiter=',')\n",
    "ssp1_exp_co2 = np.genfromtxt(''.join([path, '/data/lhc_sampling_exp_conc_ssp1_N1000.csv']), delimiter=',')\n",
    "ssp1_exp_dam = np.genfromtxt(''.join([path, '/data/lhc_sampling_exp_dam_ssp1_N1000.csv']), delimiter=',')\n",
    "ssp1_paths, ssp1_c2020, ssp1_c2030, ssp1_c2060, ssp1_c2100, ssp1_c2150, ssp1_c2200, ssp1_util,\\\n",
    "    ssp1_params, ssp1_exp_m, ssp1_exp_t, ssp1_exp_co2, ssp1_exp_dam = remove_nans(ssp1_paths_and_util, \n",
    "                                                                                  ssp1_exp_m, ssp1_exp_t, \n",
    "                                                                                  ssp1_exp_co2, ssp1_exp_dam, \n",
    "                                                                                  params)\n",
    "ssp1_ra, ssp1_eis, ssp1_ex, ssp1_end, ssp1_prtp = ssp1_params.T"
   ]
  },
  {
   "attachments": {},
   "cell_type": "markdown",
   "id": "4d8ff7d1",
   "metadata": {},
   "source": [
    "## Unpack temperature, CO<sub>2</sub> concentrations and damages data"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "a3f6d619",
   "metadata": {},
   "outputs": [],
   "source": [
    "ssp5_t2020, ssp5_t2030, ssp5_t2060, ssp5_t2100, ssp5_t2150, ssp5_t2200 = ssp5_exp_t.T\n",
    "ssp5_co2020, ssp5_co2030, ssp5_co2060, ssp5_co2100, ssp5_co2150, ssp5_co2200 = ssp5_exp_co2.T\n",
    "ssp5_d2020, ssp5_d2030, ssp5_d2060, ssp5_d2100, ssp5_d2150, ssp5_d2200 = ssp5_exp_dam.T\n",
    "\n",
    "ssp4_t2020, ssp4_t2030, ssp4_t2060, ssp4_t2100, ssp4_t2150, ssp4_t2200 = ssp4_exp_t.T\n",
    "ssp4_co2020, ssp4_co2030, ssp4_co2060, ssp4_co2100, ssp4_co2150, ssp4_co2200 = ssp4_exp_co2.T\n",
    "ssp4_d2020, ssp4_d2030, ssp4_d2060, ssp4_d2100, ssp4_d2150, ssp4_d2200 = ssp4_exp_dam.T\n",
    "\n",
    "ssp3_t2020, ssp3_t2030, ssp3_t2060, ssp3_t2100, ssp3_t2150, ssp3_t2200 = ssp3_exp_t.T\n",
    "ssp3_co2020, ssp3_co2030, ssp3_co2060, ssp3_co2100, ssp3_co2150, ssp3_co2200 = ssp3_exp_co2.T\n",
    "ssp3_d2020, ssp3_d2030, ssp3_d2060, ssp3_d2100, ssp3_d2150, ssp3_d2200 = ssp3_exp_dam.T\n",
    "\n",
    "ssp2_t2020, ssp2_t2030, ssp2_t2060, ssp2_t2100, ssp2_t2150, ssp2_t2200 = ssp2_exp_t.T\n",
    "ssp2_co2020, ssp2_co2030, ssp2_co2060, ssp2_co2100, ssp2_co2150, ssp2_co2200 = ssp2_exp_co2.T\n",
    "ssp2_d2020, ssp2_d2030, ssp2_d2060, ssp2_d2100, ssp2_d2150, ssp2_d2200 = ssp2_exp_dam.T\n",
    "\n",
    "ssp1_t2020, ssp1_t2030, ssp1_t2060, ssp1_t2100, ssp1_t2150, ssp1_t2200 = ssp1_exp_t.T\n",
    "ssp1_co2020, ssp1_co2030, ssp1_co2060, ssp1_co2100, ssp1_co2150, ssp1_co2200 = ssp1_exp_co2.T\n",
    "ssp1_d2020, ssp1_d2030, ssp1_d2060, ssp1_d2100, ssp1_d2150, ssp1_d2200 = ssp1_exp_dam.T"
   ]
  },
  {
   "cell_type": "markdown",
   "id": "supposed-prescription",
   "metadata": {},
   "source": [
    "## Make scatter and regression plot\n",
    "\n",
    "Scatter plots of price vs parameter at every point in time with regression."
   ]
  },
  {
   "cell_type": "markdown",
   "id": "genetic-dealing",
   "metadata": {},
   "source": [
    "### Combine tech change into one parameter: $\\varphi_{t} := \\varphi_{0} + \\varphi_{1} X_{t}$"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "indirect-diving",
   "metadata": {},
   "outputs": [],
   "source": [
    "# Making total tech change parameter\n",
    "\n",
    "ssp5_tottech = ssp5_ex[:, None] + ssp5_end[:, None] * ssp5_exp_m\n",
    "ssp4_tottech = ssp4_ex[:, None] + ssp4_end[:, None] * ssp4_exp_m\n",
    "ssp3_tottech = ssp3_ex[:, None] + ssp3_end[:, None] * ssp3_exp_m\n",
    "ssp2_tottech = ssp2_ex[:, None] + ssp2_end[:, None] * ssp2_exp_m\n",
    "ssp1_tottech = ssp1_ex[:, None] + ssp1_end[:, None] * ssp1_exp_m"
   ]
  },
  {
   "cell_type": "markdown",
   "id": "unknown-effect",
   "metadata": {},
   "source": [
    "### Make functions for plotting\n",
    "\n",
    "It's quite a lot, as I did a lot of different tests to see which regressions worked best. "
   ]
  },
  {
   "cell_type": "markdown",
   "id": "valued-natural",
   "metadata": {},
   "source": [
    "#### Basic linear regression panel"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "talented-norway",
   "metadata": {},
   "outputs": [],
   "source": [
    "from sklearn.linear_model import LinearRegression\n",
    "\n",
    "def get_regression(param, costs):\n",
    "    LR = LinearRegression()\n",
    "    reg = LR.fit(param.reshape(-1,1), costs)\n",
    "    a, b = reg.coef_[0], reg.intercept_\n",
    "    score = reg.score(param.reshape(-1,1), costs)\n",
    "    return a, b, score\n",
    "\n",
    "def make_panel_w_regression(ax, grid_x, grid_y, ssp5_param, ssp4_param, ssp3_param, ssp2_param, ssp1_param, ssp5_prices, ssp4_prices, ssp3_prices, ssp2_prices, ssp1_prices,\n",
    "              x_label, y_label):\n",
    "    ssp5_a, ssp5_b, ssp5_r2 = get_regression(ssp5_param, ssp5_prices)\n",
    "    ssp4_a, ssp4_b, ssp4_r2 = get_regression(ssp4_param, ssp4_prices)\n",
    "    ssp3_a, ssp3_b, ssp3_r2 = get_regression(ssp3_param, ssp3_prices)\n",
    "    ssp2_a, ssp2_b, ssp2_r2 = get_regression(ssp2_param, ssp2_prices)\n",
    "    ssp1_a, ssp1_b, ssp1_r2 = get_regression(ssp1_param, ssp1_prices)\n",
    "    \n",
    "    ssp5_param_range = np.arange(min(ssp5_param), max(ssp5_param), 0.01)\n",
    "    ssp4_param_range = np.arange(min(ssp4_param), max(ssp4_param), 0.01)\n",
    "    ssp3_param_range = np.arange(min(ssp3_param), max(ssp3_param), 0.01)\n",
    "    ssp2_param_range = np.arange(min(ssp2_param), max(ssp2_param), 0.01)\n",
    "    ssp1_param_range = np.arange(min(ssp1_param), max(ssp1_param), 0.01)\n",
    "    \n",
    "    alpha = 0.2\n",
    "    ax[grid_x, grid_y].scatter(ssp5_param, ssp5_prices, c=color_list[ssp5_cind], marker=marker_list[0], alpha=alpha)\n",
    "    ax[grid_x, grid_y].scatter(ssp4_param, ssp4_prices, c=color_list[ssp4_cind], marker=marker_list[1], alpha=alpha)\n",
    "    ax[grid_x, grid_y].scatter(ssp3_param, ssp3_prices, c=color_list[ssp3_cind], marker=marker_list[2], alpha=alpha)\n",
    "    ax[grid_x, grid_y].scatter(ssp2_param, ssp2_prices, c=color_list[ssp2_cind], marker=marker_list[3], alpha=alpha)\n",
    "    ax[grid_x, grid_y].scatter(ssp1_param, ssp1_prices, c=color_list[ssp1_cind], marker=marker_list[4], alpha=alpha)\n",
    "    \n",
    "    ax[grid_x, grid_y].plot(ssp5_param_range, ssp5_a * ssp5_param_range + ssp5_b, c=color_list[ssp5_cind],\n",
    "                            linewidth=3, label=''.join([\"SSP5, r$^{2} =$\", \"{:0.2f}\".format(ssp5_r2)]))\n",
    "    ax[grid_x, grid_y].plot(ssp4_param_range, ssp4_a * ssp4_param_range + ssp4_b, c=color_list[ssp4_cind],\n",
    "                            linewidth=3, label=''.join([\"SSP4, r$^{2} =$\", \"{:0.2f}\".format(ssp4_r2)]))\n",
    "    ax[grid_x, grid_y].plot(ssp3_param_range, ssp3_a * ssp3_param_range + ssp3_b, c=color_list[ssp3_cind],\n",
    "                            linewidth=3, label=''.join([\"SSP3, r$^{2} =$\", \"{:0.2f}\".format(ssp3_r2)]))\n",
    "    ax[grid_x, grid_y].plot(ssp2_param_range, ssp2_a * ssp2_param_range + ssp2_b, c=color_list[ssp2_cind],\n",
    "                            linewidth=3, label=''.join([\"SSP2, r$^{2} =$\", \"{:0.2f}\".format(ssp2_r2)]))\n",
    "    ax[grid_x, grid_y].plot(ssp1_param_range, ssp1_a * ssp1_param_range + ssp1_b, c=color_list[ssp1_cind],\n",
    "                            linewidth=3, label=''.join([\"SSP1, r$^{2} =$\", \"{:0.2f}\".format(ssp1_r2)]))\n",
    "    \n",
    "    ax[grid_x, grid_y].legend(fontsize=15, frameon=True, facecolor='white', framealpha=0.9)\n",
    "    ax[grid_x, grid_y].set_xlabel(x_label)\n",
    "    ax[grid_x, grid_y].set_ylabel(y_label)\n",
    "    if x_label == \"Risk Aversion\":\n",
    "        ax[grid_x, grid_y].set_xticks([3, 9, 15])\n",
    "    elif x_label == \"EIS\":\n",
    "        ax[grid_x, grid_y].set_xticks([0.5, 0.8, 1.1])\n",
    "    elif x_label == r\"$\\delta$ (%)\":\n",
    "        ax[grid_x, grid_y].set_xticks([0.1, 0.45, 0.8, 1.15, 1.5])"
   ]
  },
  {
   "cell_type": "markdown",
   "id": "acting-theory",
   "metadata": {},
   "source": [
    "#### Quadratic regression panel"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "funded-unemployment",
   "metadata": {},
   "outputs": [],
   "source": [
    "from sklearn.metrics import r2_score\n",
    "\n",
    "def get_quad_regression(param, costs):\n",
    "    a2, a1, a0 = np.polyfit(param, costs, deg=2)\n",
    "    r2 = r2_score(costs, a2 * param**2 + a1 * param + a0)\n",
    "    return a2, a1, a0, r2\n",
    "\n",
    "def make_panel_w_quad_regression(ax, grid_x, grid_y, ssp5_param, ssp4_param, ssp3_param, ssp2_param, ssp1_param, ssp5_prices, ssp4_prices, ssp3_prices, ssp2_prices, ssp1_prices,\n",
    "              x_label, y_label):\n",
    "    ssp5_a2, ssp5_a1, ssp5_a0, ssp5_r2 = get_quad_regression(ssp5_param, ssp5_prices)\n",
    "    ssp4_a2, ssp4_a1, ssp4_a0, ssp4_r2 = get_quad_regression(ssp4_param, ssp4_prices)\n",
    "    ssp3_a2, ssp3_a1, ssp3_a0, ssp3_r2 = get_quad_regression(ssp3_param, ssp3_prices)\n",
    "    ssp2_a2, ssp2_a1, ssp2_a0, ssp2_r2 = get_quad_regression(ssp2_param, ssp2_prices)\n",
    "    ssp1_a2, ssp1_a1, ssp1_a0, ssp1_r2 = get_quad_regression(ssp1_param, ssp1_prices)\n",
    "    \n",
    "    ssp5_param_range = np.arange(min(ssp5_param), max(ssp5_param), 0.01)\n",
    "    ssp4_param_range = np.arange(min(ssp4_param), max(ssp4_param), 0.01)\n",
    "    ssp3_param_range = np.arange(min(ssp3_param), max(ssp3_param), 0.01)\n",
    "    ssp2_param_range = np.arange(min(ssp2_param), max(ssp2_param), 0.01)\n",
    "    ssp1_param_range = np.arange(min(ssp1_param), max(ssp1_param), 0.01)\n",
    "    \n",
    "    alpha = 0.2\n",
    "    ax[grid_x, grid_y].scatter(ssp5_param, ssp5_prices, c=color_list[ssp5_cind], marker=marker_list[0], alpha=alpha)\n",
    "    ax[grid_x, grid_y].scatter(ssp4_param, ssp4_prices, c=color_list[ssp4_cind], marker=marker_list[1], alpha=alpha)\n",
    "    ax[grid_x, grid_y].scatter(ssp3_param, ssp3_prices, c=color_list[ssp3_cind], marker=marker_list[2], alpha=alpha)\n",
    "    ax[grid_x, grid_y].scatter(ssp2_param, ssp2_prices, c=color_list[ssp2_cind], marker=marker_list[3], alpha=alpha)\n",
    "    ax[grid_x, grid_y].scatter(ssp1_param, ssp1_prices, c=color_list[ssp1_cind], marker=marker_list[4], alpha=alpha)\n",
    "    \n",
    "    ax[grid_x, grid_y].plot(ssp5_param_range, ssp5_a2 * ssp5_param_range**2 + ssp5_a1 * ssp5_param_range + ssp5_a0, c=color_list[ssp5_cind],\n",
    "                            linewidth=3, label=''.join([\"SSP5, r$^{2} =$\", \"{:0.2f}\".format(ssp5_r2)]))\n",
    "    ax[grid_x, grid_y].plot(ssp4_param_range, ssp4_a2 * ssp4_param_range**2 + ssp4_a1 * ssp4_param_range + ssp4_a0, c=color_list[ssp4_cind],\n",
    "                            linewidth=3, label=''.join([\"SSP4, r$^{2} =$\", \"{:0.2f}\".format(ssp4_r2)]))\n",
    "    ax[grid_x, grid_y].plot(ssp3_param_range, ssp3_a2 * ssp3_param_range**2 + ssp3_a1 * ssp3_param_range + ssp3_a0, c=color_list[ssp3_cind],\n",
    "                            linewidth=3, label=''.join([\"SSP3, r$^{2} =$\", \"{:0.2f}\".format(ssp3_r2)]))\n",
    "    ax[grid_x, grid_y].plot(ssp2_param_range, ssp2_a2 * ssp2_param_range**2 + ssp2_a1 * ssp2_param_range + ssp2_a0, c=color_list[ssp2_cind],\n",
    "                            linewidth=3, label=''.join([\"SSP2, r$^{2} =$\", \"{:0.2f}\".format(ssp2_r2)]))\n",
    "    ax[grid_x, grid_y].plot(ssp1_param_range, ssp1_a2 * ssp1_param_range**2 + ssp1_a1 * ssp1_param_range + ssp1_a0, c=color_list[ssp1_cind],\n",
    "                            linewidth=3, label=''.join([\"SSP1, r$^{2} =$\", \"{:0.2f}\".format(ssp1_r2)]))\n",
    "    \n",
    "    ax[grid_x, grid_y].legend(fontsize=15, frameon=True, facecolor='white', framealpha=0.9)\n",
    "    ax[grid_x, grid_y].set_xlabel(x_label)\n",
    "    ax[grid_x, grid_y].set_ylabel(y_label)"
   ]
  },
  {
   "cell_type": "markdown",
   "id": "cleared-outreach",
   "metadata": {},
   "source": [
    "### Make plot!"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "actual-explanation",
   "metadata": {},
   "outputs": [],
   "source": [
    "fig, ax = plt.subplots(4, 6, figsize=(30, 20))\n",
    "\n",
    "ax[0,0].set_title(\"2020\")\n",
    "ax[0,1].set_title(\"2030\")\n",
    "ax[0,2].set_title(\"2060\")\n",
    "ax[0,3].set_title(\"2100\")\n",
    "ax[0,4].set_title(\"2150\")\n",
    "ax[0,5].set_title(\"2200\")\n",
    "\n",
    "###########################\n",
    "## RISK AVERSION ROW\n",
    "###########################\n",
    "make_panel_w_regression(ax, 0, 0, ssp5_ra, ssp4_ra, ssp3_ra, ssp2_ra, ssp1_ra, ssp5_c2020, ssp4_c2020, ssp3_c2020, ssp2_c2020, ssp1_c2020, \n",
    "           \"Risk Aversion\", \"Cost\\n(2020 $USD)\")\n",
    "\n",
    "make_panel_w_regression(ax, 0, 1, ssp5_ra, ssp4_ra, ssp3_ra, ssp2_ra, ssp1_ra, ssp5_c2030, ssp4_c2030, ssp3_c2030, ssp2_c2030, ssp1_c2030, \n",
    "           \"Risk Aversion\", None)\n",
    "\n",
    "make_panel_w_regression(ax, 0, 2, ssp5_ra, ssp4_ra, ssp3_ra, ssp2_ra, ssp1_ra, ssp5_c2060, ssp4_c2060, ssp3_c2060, ssp2_c2060, ssp1_c2060, \n",
    "           \"Risk Aversion\", None)\n",
    "\n",
    "make_panel_w_regression(ax, 0, 3, ssp5_ra, ssp4_ra, ssp3_ra, ssp2_ra, ssp1_ra, ssp5_c2100, ssp4_c2100, ssp3_c2100, ssp2_c2100, ssp1_c2100, \n",
    "           \"Risk Aversion\", None)\n",
    "\n",
    "make_panel_w_regression(ax, 0, 4, ssp5_ra, ssp4_ra, ssp3_ra, ssp2_ra, ssp1_ra, ssp5_c2150, ssp4_c2150, ssp3_c2150, ssp2_c2150, ssp1_c2150, \n",
    "           \"Risk Aversion\", None)\n",
    "\n",
    "make_panel_w_regression(ax, 0, 5, ssp5_ra, ssp4_ra, ssp3_ra, ssp2_ra, ssp1_ra, ssp5_c2200, ssp4_c2200, ssp3_c2200, ssp2_c2200, ssp1_c2200, \n",
    "           \"Risk Aversion\", None)\n",
    "\n",
    "###########################\n",
    "# EIS ROW\n",
    "###########################\n",
    "make_panel_w_regression(ax, 1, 0, ssp5_eis, ssp4_eis, ssp3_eis, ssp2_eis, ssp1_eis, ssp5_c2020, ssp4_c2020, ssp3_c2020, ssp2_c2020, ssp1_c2020, \n",
    "           \"EIS\", \"Cost\\n(2020 $USD)\")\n",
    "\n",
    "make_panel_w_regression(ax, 1, 1, ssp5_eis, ssp4_eis, ssp3_eis, ssp2_eis, ssp1_eis, ssp5_c2030, ssp4_c2030, ssp3_c2030, ssp2_c2030, ssp1_c2030, \n",
    "           \"EIS\", None)\n",
    "\n",
    "make_panel_w_regression(ax, 1, 2, ssp5_eis, ssp4_eis, ssp3_eis, ssp2_eis, ssp1_eis, ssp5_c2060, ssp4_c2060, ssp3_c2060, ssp2_c2060, ssp1_c2060, \n",
    "           \"EIS\", None)\n",
    "\n",
    "make_panel_w_regression(ax, 1, 3, ssp5_eis, ssp4_eis, ssp3_eis, ssp2_eis, ssp1_eis, ssp5_c2100, ssp4_c2100, ssp3_c2100, ssp2_c2100, ssp1_c2100, \n",
    "           \"EIS\", None)\n",
    "\n",
    "make_panel_w_regression(ax, 1, 4, ssp5_eis, ssp4_eis, ssp3_eis, ssp2_eis, ssp1_eis, ssp5_c2150, ssp4_c2150, ssp3_c2150, ssp2_c2150, ssp1_c2150, \n",
    "           \"EIS\", None)\n",
    "\n",
    "make_panel_w_regression(ax, 1, 5, ssp5_eis, ssp4_eis, ssp3_eis, ssp2_eis, ssp1_eis, ssp5_c2200, ssp4_c2200, ssp3_c2200, ssp2_c2200, ssp1_c2200, \n",
    "           \"EIS\", None)\n",
    "\n",
    "###########################\n",
    "# PRTP ROW\n",
    "###########################\n",
    "make_panel_w_regression(ax, 2, 0, ssp5_prtp * 100, ssp4_prtp * 100, ssp3_prtp * 100, ssp2_prtp * 100, ssp1_prtp * 100, ssp5_c2020, ssp4_c2020, ssp3_c2020, ssp2_c2020, ssp1_c2020, \n",
    "           r\"$\\delta$ (%)\", \"Cost\\n(2020 $USD)\")\n",
    "\n",
    "make_panel_w_regression(ax, 2, 1, ssp5_prtp * 100, ssp4_prtp * 100, ssp3_prtp * 100, ssp2_prtp * 100, ssp1_prtp * 100, ssp5_c2030, ssp4_c2030, ssp3_c2030, ssp2_c2030, ssp1_c2030, \n",
    "           r\"$\\delta$ (%)\", None)\n",
    "\n",
    "make_panel_w_regression(ax, 2, 2, ssp5_prtp * 100, ssp4_prtp * 100, ssp3_prtp * 100, ssp2_prtp * 100, ssp1_prtp * 100, ssp5_c2060, ssp4_c2060, ssp3_c2060, ssp2_c2060, ssp1_c2060, \n",
    "           r\"$\\delta$ (%)\", None)\n",
    "\n",
    "make_panel_w_regression(ax, 2, 3, ssp5_prtp * 100, ssp4_prtp * 100, ssp3_prtp * 100, ssp2_prtp * 100, ssp1_prtp * 100, ssp5_c2100, ssp4_c2100, ssp3_c2100, ssp2_c2100, ssp1_c2100, \n",
    "           r\"$\\delta$ (%)\", None)\n",
    "\n",
    "make_panel_w_regression(ax, 2, 4, ssp5_prtp * 100, ssp4_prtp * 100, ssp3_prtp * 100, ssp2_prtp * 100, ssp1_prtp * 100, ssp5_c2150, ssp4_c2150, ssp3_c2150, ssp2_c2150, ssp1_c2150, \n",
    "           r\"$\\delta$ (%)\", None)\n",
    "\n",
    "make_panel_w_regression(ax, 2, 5, ssp5_prtp * 100, ssp4_prtp * 100, ssp3_prtp * 100, ssp2_prtp * 100, ssp1_prtp * 100, ssp5_c2200, ssp4_c2200, ssp3_c2200, ssp2_c2200, ssp1_c2200, \n",
    "           r\"$\\delta$ (%)\", None)\n",
    "\n",
    "###########################\n",
    "# EX TECH GROWTH ROW\n",
    "###########################\n",
    "make_panel_w_regression(ax, 3, 0, ssp5_tottech[:, 0], ssp4_tottech[:, 0], ssp3_tottech[:, 0],\n",
    "                        ssp2_tottech[:, 0], ssp1_tottech[:, 0], ssp5_c2020, ssp4_c2020, ssp3_c2020, \n",
    "                        ssp2_c2020, ssp1_c2020, \"Tech Growth (%)\", \"Cost\\n(2020 $USD)\")\n",
    "\n",
    "make_panel_w_regression(ax, 3, 1, ssp5_tottech[:, 1], ssp4_tottech[:, 1], ssp3_tottech[:, 1], \n",
    "                        ssp2_tottech[:, 1], ssp1_tottech[:, 1], ssp5_c2030, ssp4_c2030, ssp3_c2030, \n",
    "                        ssp2_c2030, ssp1_c2030, \"Tech Growth (%)\", None)\n",
    "\n",
    "make_panel_w_quad_regression(ax, 3, 2, ssp5_tottech[:, 2], ssp4_tottech[:, 2], ssp3_tottech[:, 2], \n",
    "                        ssp2_tottech[:, 2], ssp1_tottech[:, 2], ssp5_c2060, ssp4_c2060, ssp3_c2060, \n",
    "                        ssp2_c2060, ssp1_c2060, \"Tech Growth (%)\", None)\n",
    "\n",
    "make_panel_w_quad_regression(ax, 3, 3, ssp5_tottech[:, 3], ssp4_tottech[:, 3], ssp3_tottech[:, 3], \n",
    "                        ssp2_tottech[:, 3], ssp1_tottech[:, 3], ssp5_c2100, ssp4_c2100, ssp3_c2100, \n",
    "                        ssp2_c2100, ssp1_c2100, \"Tech Growth (%)\", None)\n",
    "\n",
    "make_panel_w_quad_regression(ax, 3, 4, ssp5_tottech[:, 4], ssp4_tottech[:, 4], ssp3_tottech[:, 4], \n",
    "                        ssp2_tottech[:, 4], ssp1_tottech[:, 4], ssp5_c2150, ssp4_c2150, ssp3_c2150, \n",
    "                        ssp2_c2150, ssp1_c2150, \"Tech Growth (%)\", None)\n",
    "\n",
    "make_panel_w_quad_regression(ax, 3, 5, ssp5_tottech[:, 5], ssp4_tottech[:, 5], ssp3_tottech[:, 5], \n",
    "                        ssp2_tottech[:, 5], ssp1_tottech[:, 5], ssp5_c2200, ssp4_c2200, ssp3_c2200, \n",
    "                        ssp2_c2200, ssp1_c2200, \"Tech Growth (%)\", None)\n",
    "\n",
    "ax[3,0].set_xticks([0,1,2,3])\n",
    "ax[3,0].set_xlim((0,3))\n",
    "ax[3,1].set_xticks([0,1,2,3,4,5])\n",
    "ax[3,2].set_xticks([0,1,2,3,4,5,6])\n",
    "ax[3,3].set_xticks([0,1,2,3,4,5,6])\n",
    "ax[3,4].set_xticks([0,1,2,3,4,5,6])\n",
    "ax[3,5].set_xticks([0,1,2,3,4,5,6])\n",
    "\n",
    "fig.tight_layout()\n",
    "sns.despine(trim=True, offset=10)\n",
    "\n",
    "if save_fig:\n",
    "    fig.savefig(''.join([basefile, 'regression-sticky-cost.png']), dpi=400)    "
   ]
  },
  {
   "cell_type": "markdown",
   "id": "comfortable-officer",
   "metadata": {},
   "source": [
    "## Parition $r^2$ sources\n",
    "\n",
    "We now partition the variance into different parameters using the $r^2$ calculated above."
   ]
  },
  {
   "cell_type": "markdown",
   "id": "willing-adventure",
   "metadata": {},
   "source": [
    "### $r^2$ curves in time functions\n",
    "\n",
    "The first is for all parameters that aren't tech change. Since tech change required two regressions in the later periods, it needs its own function where the $r^2$ of each curve is averaged and reported as the $r^2$ of that period."
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "ordinary-insurance",
   "metadata": {},
   "outputs": [],
   "source": [
    "# regressions\n",
    "\n",
    "def get_r2_in_time(param, paths):\n",
    "    c2020, c2030, c2060, c2100, c2150, c2200 = paths.T\n",
    "    _, _, r2_2020 = get_regression(param, c2020)\n",
    "    _, _, r2_2030 = get_regression(param, c2030)\n",
    "    _, _, r2_2060 = get_regression(param, c2060)\n",
    "    _, _, r2_2100 = get_regression(param, c2100)\n",
    "    _, _, r2_2150 = get_regression(param, c2150)\n",
    "    _, _, r2_2200 = get_regression(param, c2200)\n",
    "    \n",
    "    r2 = np.array([r2_2020, r2_2030, r2_2060, r2_2100, r2_2150, r2_2200])\n",
    "    return r2"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "convenient-qatar",
   "metadata": {},
   "outputs": [],
   "source": [
    "def get_tech_r2_in_time(tottech, paths, bp1_2150, bp2_2150, bp1_2200, bp2_2200, ssp1):\n",
    "    c2020, c2030, c2060, c2100, c2150, c2200 = paths.T\n",
    "    _, _, r2_2020 = get_regression(tottech[:,0], c2020)\n",
    "    _, _, r2_2030 = get_regression(tottech[:,1], c2030)\n",
    "    _, _, _, r2_2060 = get_quad_regression(tottech[:,2], c2060)\n",
    "    _, _, _, r2_2100 = get_quad_regression(tottech[:,3], c2100)\n",
    "    \n",
    "    ssp5_param_1, ssp5_param_2, ssp5_param_3, ssp5_prices_1, ssp5_prices_2, ssp5_prices_3 = get_double_partition(tottech[:,5], c2150, bp1_2150, bp2_2150)\n",
    "    \n",
    "    _, _, _, ssp5_r2_1 = get_quad_regression(ssp5_param_1, ssp5_prices_1)\n",
    "        \n",
    "    _, _, _, ssp5_r2_3 = get_quad_regression(ssp5_param_3, ssp5_prices_3)\n",
    "\n",
    "    if ssp1:\n",
    "        r2_2150 = np.mean([ssp5_r2_1])\n",
    "    else:\n",
    "        r2_2150 = np.mean([ssp5_r2_1, ssp5_r2_3])\n",
    "    \n",
    "    ssp5_param_1, ssp5_param_2, ssp5_param_3, ssp5_prices_1, ssp5_prices_2, ssp5_prices_3 = get_double_partition(tottech[:,5], c2200, bp2_2200, bp2_2200)\n",
    "    \n",
    "    _, _, _, ssp5_r2_1 = get_quad_regression(ssp5_param_1, ssp5_prices_1)\n",
    "        \n",
    "    _, _, _, ssp5_r2_3 = get_quad_regression(ssp5_param_3, ssp5_prices_3)\n",
    "\n",
    "    r2_2200 = np.mean([ssp5_r2_1, ssp5_r2_3])\n",
    "    \n",
    "    r2 = np.array([r2_2020, r2_2030, r2_2060, r2_2100, r2_2150, r2_2200])\n",
    "    return r2"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "canadian-choir",
   "metadata": {},
   "outputs": [],
   "source": [
    "ssp5_eis_r2 = get_r2_in_time(ssp5_eis, ssp5_paths)\n",
    "ssp5_ra_r2 = get_r2_in_time(ssp5_ra, ssp5_paths)\n",
    "ssp5_prtp_r2 = get_r2_in_time(ssp5_prtp, ssp5_paths)\n",
    "ssp5_tech_r2 = get_tech_r2_in_time(ssp5_tottech, ssp5_paths, bp1s_2150[0], bp2s_2150[0], \n",
    "                                   bp1s_2200[0], bp2s_2200[0], False)\n",
    "\n",
    "ssp4_eis_r2 = get_r2_in_time(ssp4_eis, ssp4_paths)\n",
    "ssp4_ra_r2 = get_r2_in_time(ssp4_ra, ssp4_paths)\n",
    "ssp4_prtp_r2 = get_r2_in_time(ssp4_prtp, ssp4_paths)\n",
    "ssp4_tech_r2 = get_tech_r2_in_time(ssp4_tottech, ssp4_paths, bp1s_2150[1], bp2s_2150[1], \n",
    "                                   bp1s_2200[1], bp2s_2200[1], False)\n",
    "\n",
    "ssp3_eis_r2 = get_r2_in_time(ssp3_eis, ssp3_paths)\n",
    "ssp3_ra_r2 = get_r2_in_time(ssp3_ra, ssp3_paths)\n",
    "ssp3_prtp_r2 = get_r2_in_time(ssp3_prtp, ssp3_paths)\n",
    "ssp3_tech_r2 = get_tech_r2_in_time(ssp3_tottech, ssp3_paths, bp1s_2150[2], bp2s_2150[2], \n",
    "                                   bp1s_2200[2], bp2s_2200[2], False)\n",
    "\n",
    "ssp2_eis_r2 = get_r2_in_time(ssp2_eis, ssp2_paths)\n",
    "ssp2_ra_r2 = get_r2_in_time(ssp2_ra, ssp2_paths)\n",
    "ssp2_prtp_r2 = get_r2_in_time(ssp2_prtp, ssp2_paths)\n",
    "ssp2_tech_r2 = get_tech_r2_in_time(ssp2_tottech, ssp2_paths, bp1s_2150[3], bp2s_2150[3], \n",
    "                                   bp1s_2200[3], bp2s_2200[3], False)\n",
    "\n",
    "ssp1_eis_r2 = get_r2_in_time(ssp1_eis, ssp1_paths)\n",
    "ssp1_ra_r2 = get_r2_in_time(ssp1_ra, ssp1_paths)\n",
    "ssp1_prtp_r2 = get_r2_in_time(ssp1_prtp, ssp1_paths)\n",
    "ssp1_tech_r2 = get_tech_r2_in_time(ssp1_tottech, ssp1_paths, bp1s_2150[4], bp2s_2150[4], \n",
    "                                   bp1s_2200[4], bp2s_2200[4], True)"
   ]
  },
  {
   "cell_type": "markdown",
   "id": "leading-horror",
   "metadata": {},
   "source": [
    "## Make plot of raw $r^2$ values"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "quiet-marshall",
   "metadata": {},
   "outputs": [],
   "source": [
    "fig, ax = plt.subplots(1,5, figsize=(20,20), sharey=True)\n",
    "\n",
    "ax[0].set_ylim((0,1))\n",
    "\n",
    "alpha = 0.8\n",
    "lw = 3\n",
    "l1cind = 0\n",
    "l2cind = 1\n",
    "l3cind = 2\n",
    "l4cind = 3\n",
    "l5cind = 4\n",
    "\n",
    "################################\n",
    "# SSP5\n",
    "################################\n",
    "ax[0].plot(node_times, ssp5_ra_r2, linestyle='dashed', linewidth=lw, \n",
    "        color=color_list[l1cind])\n",
    "ax[0].fill_between(node_times, np.zeros_like(node_times), ssp5_ra_r2, alpha=alpha, \n",
    "                color=color_list[l1cind], label=\"RA\")\n",
    "\n",
    "ax[0].plot(node_times, ssp5_eis_r2 + ssp5_ra_r2, linestyle='dashed', linewidth=lw, \n",
    "        color=color_list[l2cind])\n",
    "ax[0].fill_between(node_times, ssp5_ra_r2, ssp5_ra_r2 + ssp5_eis_r2, alpha=alpha, \n",
    "                color=color_list[l2cind], label=\"EIS\")\n",
    "\n",
    "ax[0].plot(node_times, ssp5_eis_r2 + ssp5_ra_r2 + ssp5_prtp_r2, linestyle='dashed', linewidth=lw, \n",
    "        color=color_list[l3cind])\n",
    "ax[0].fill_between(node_times, ssp5_eis_r2 + ssp5_ra_r2, ssp5_ra_r2 + ssp5_eis_r2 + ssp5_prtp_r2, alpha=alpha, \n",
    "                color=color_list[l3cind], label=\"PRTP\")\n",
    "\n",
    "ax[0].plot(node_times, ssp5_eis_r2 + ssp5_ra_r2 + ssp5_prtp_r2 + ssp5_tech_r2, linestyle='dashed', linewidth=lw, \n",
    "        color=color_list[l4cind])\n",
    "ax[0].fill_between(node_times, ssp5_eis_r2 + ssp5_ra_r2 + ssp5_prtp_r2, ssp5_ra_r2 + ssp5_eis_r2 + ssp5_prtp_r2 + ssp5_tech_r2, alpha=alpha, \n",
    "                color=color_list[l4cind], label=\"Tech. Growth\")\n",
    "\n",
    "\n",
    "################################\n",
    "# SSP4\n",
    "################################\n",
    "ax[1].plot(node_times, ssp4_ra_r2, linestyle='dashed', linewidth=lw, \n",
    "        color=color_list[l1cind])\n",
    "ax[1].fill_between(node_times, np.zeros_like(node_times), ssp4_ra_r2, alpha=alpha, \n",
    "                color=color_list[l1cind], label=\"RA\")\n",
    "\n",
    "ax[1].plot(node_times, ssp4_eis_r2 + ssp4_ra_r2, linestyle='dashed', linewidth=lw, \n",
    "        color=color_list[l2cind])\n",
    "ax[1].fill_between(node_times, ssp4_ra_r2, ssp5_ra_r2 + ssp4_eis_r2, alpha=alpha, \n",
    "                color=color_list[l2cind], label=\"EIS\")\n",
    "\n",
    "ax[1].plot(node_times, ssp4_eis_r2 + ssp4_ra_r2 + ssp4_prtp_r2, linestyle='dashed', linewidth=lw, \n",
    "        color=color_list[l3cind])\n",
    "ax[1].fill_between(node_times, ssp4_eis_r2 + ssp4_ra_r2, ssp4_ra_r2 + ssp4_eis_r2 + ssp4_prtp_r2, alpha=alpha, \n",
    "                color=color_list[l3cind], label=\"PRTP\")\n",
    "\n",
    "ax[1].plot(node_times, ssp4_eis_r2 + ssp4_ra_r2 + ssp4_prtp_r2 + ssp4_tech_r2, linestyle='dashed', linewidth=lw, \n",
    "        color=color_list[l4cind])\n",
    "ax[1].fill_between(node_times, ssp4_eis_r2 + ssp4_ra_r2 + ssp4_prtp_r2, ssp4_ra_r2 + ssp4_eis_r2 + ssp4_prtp_r2 + ssp4_tech_r2, alpha=alpha, \n",
    "                color=color_list[l4cind], label=\"Tech. Growth\")\n",
    "\n",
    "################################\n",
    "# SSP3\n",
    "################################\n",
    "ax[2].plot(node_times, ssp3_ra_r2, linestyle='dashed', linewidth=lw, \n",
    "        color=color_list[l1cind])\n",
    "ax[2].fill_between(node_times, np.zeros_like(node_times), ssp3_ra_r2, alpha=alpha, \n",
    "                color=color_list[l1cind], label=\"RA\")\n",
    "\n",
    "ax[2].plot(node_times, ssp3_eis_r2 + ssp3_ra_r2, linestyle='dashed', linewidth=lw, \n",
    "        color=color_list[l2cind])\n",
    "ax[2].fill_between(node_times, ssp3_ra_r2, ssp3_ra_r2 + ssp3_eis_r2, alpha=alpha, \n",
    "                color=color_list[l2cind], label=\"EIS\")\n",
    "\n",
    "ax[2].plot(node_times, ssp3_eis_r2 + ssp3_ra_r2 + ssp3_prtp_r2, linestyle='dashed', linewidth=lw, \n",
    "        color=color_list[l3cind])\n",
    "ax[2].fill_between(node_times, ssp3_eis_r2 + ssp3_ra_r2, ssp3_ra_r2 + ssp3_eis_r2 + ssp3_prtp_r2, alpha=alpha, \n",
    "                color=color_list[l3cind], label=\"PRTP\")\n",
    "\n",
    "ax[2].plot(node_times, ssp3_eis_r2 + ssp3_ra_r2 + ssp3_prtp_r2 + ssp3_tech_r2, linestyle='dashed', linewidth=lw, \n",
    "        color=color_list[l4cind])\n",
    "ax[2].fill_between(node_times, ssp3_eis_r2 + ssp3_ra_r2 + ssp3_prtp_r2, ssp3_ra_r2 + ssp3_eis_r2 + ssp3_prtp_r2 + ssp3_tech_r2, alpha=alpha, \n",
    "                color=color_list[l4cind], label=\"Tech. Growth\")\n",
    "\n",
    "################################\n",
    "# SSP2\n",
    "################################\n",
    "ax[3].plot(node_times, ssp2_ra_r2, linestyle='dashed', linewidth=lw, \n",
    "        color=color_list[l1cind])\n",
    "ax[3].fill_between(node_times, np.zeros_like(node_times), ssp2_ra_r2, alpha=alpha, \n",
    "                color=color_list[l1cind], label=\"RA\")\n",
    "\n",
    "ax[3].plot(node_times, ssp2_eis_r2 + ssp2_ra_r2, linestyle='dashed', linewidth=lw, \n",
    "        color=color_list[l2cind])\n",
    "ax[3].fill_between(node_times, ssp2_ra_r2, ssp2_ra_r2 + ssp2_eis_r2, alpha=alpha, \n",
    "                color=color_list[l2cind], label=\"EIS\")\n",
    "\n",
    "ax[3].plot(node_times, ssp2_eis_r2 + ssp2_ra_r2 + ssp2_prtp_r2, linestyle='dashed', linewidth=lw, \n",
    "        color=color_list[l3cind])\n",
    "ax[3].fill_between(node_times, ssp2_eis_r2 + ssp2_ra_r2, ssp2_ra_r2 + ssp2_eis_r2 + ssp2_prtp_r2, alpha=alpha, \n",
    "                color=color_list[l3cind], label=\"PRTP\")\n",
    "\n",
    "ax[3].plot(node_times, ssp2_eis_r2 + ssp2_ra_r2 + ssp2_prtp_r2 + ssp2_tech_r2, linestyle='dashed', linewidth=lw, \n",
    "        color=color_list[l4cind])\n",
    "ax[3].fill_between(node_times, ssp2_eis_r2 + ssp2_ra_r2 + ssp2_prtp_r2, ssp2_ra_r2 + ssp2_eis_r2 + ssp2_prtp_r2 + ssp2_tech_r2, alpha=alpha, \n",
    "                color=color_list[l4cind], label=\"Tech. Growth\")\n",
    "\n",
    "\n",
    "################################\n",
    "# SSP1\n",
    "################################\n",
    "ax[4].plot(node_times, ssp1_ra_r2, linestyle='dashed', linewidth=lw, \n",
    "        color=color_list[l1cind])\n",
    "ax[4].fill_between(node_times, np.zeros_like(node_times), ssp5_ra_r2, alpha=alpha, \n",
    "                color=color_list[l1cind], label=\"RA\")\n",
    "\n",
    "ax[4].plot(node_times, ssp1_eis_r2 + ssp1_ra_r2, linestyle='dashed', linewidth=lw, \n",
    "        color=color_list[l2cind])\n",
    "ax[4].fill_between(node_times, ssp1_ra_r2, ssp1_ra_r2 + ssp1_eis_r2, alpha=alpha, \n",
    "                color=color_list[l2cind], label=\"EIS\")\n",
    "\n",
    "ax[4].plot(node_times, ssp1_eis_r2 + ssp1_ra_r2 + ssp1_prtp_r2, linestyle='dashed', linewidth=lw, \n",
    "        color=color_list[l3cind])\n",
    "ax[4].fill_between(node_times, ssp1_eis_r2 + ssp1_ra_r2, ssp1_ra_r2 + ssp1_eis_r2 + ssp1_prtp_r2, alpha=alpha, \n",
    "                color=color_list[l3cind], label=\"PRTP\")\n",
    "\n",
    "ax[4].plot(node_times, ssp1_eis_r2 + ssp1_ra_r2 + ssp1_prtp_r2 + ssp1_tech_r2, linestyle='dashed', linewidth=lw, \n",
    "        color=color_list[l4cind])\n",
    "ax[4].fill_between(node_times, ssp1_eis_r2 + ssp1_ra_r2 + ssp1_prtp_r2, ssp1_ra_r2 + ssp1_eis_r2 + ssp1_prtp_r2 + ssp1_tech_r2, alpha=alpha, \n",
    "                color=color_list[l4cind], label=\"Tech. Growth\")\n",
    "\n",
    "ax[2].legend(loc='upper center', bbox_to_anchor=(0.5, -0.25), fancybox=True, shadow=True, ncol=5)\n",
    "\n",
    "ax[0].set_ylabel(\"r$^{2}$\")\n",
    "ax[0].set_title(\"SSP5\")\n",
    "ax[1].set_title(\"SSP4\")\n",
    "ax[2].set_title(\"SSP3\")\n",
    "ax[3].set_title(\"SSP2\")\n",
    "ax[4].set_title(\"SSP1\")\n",
    "\n",
    "for i in range(5):\n",
    "    ax[i].set_xlim((node_times[0], node_times[-1]))\n",
    "    ax[i].set_aspect(1.0/ax[i].get_data_ratio(), adjustable='box')\n",
    "    ax[i].set_xlabel(\"Year\")\n",
    "\n",
    "#fig.tight_layout()"
   ]
  },
  {
   "cell_type": "markdown",
   "id": "invalid-perfume",
   "metadata": {},
   "source": [
    "## Make plot using fraction of available $r^2$\n",
    "\n",
    "The missing $r^2$ in the above can be chalked up to either optimization variance or other variance (like in damage calculations, etc., but we expect the majority to be optimization error)."
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "israeli-taiwan",
   "metadata": {},
   "outputs": [],
   "source": [
    "fig, ax = plt.subplots(1,5, figsize=(20,20), sharey=True)\n",
    "\n",
    "ax[0].set_ylim((0,1))\n",
    "\n",
    "alpha = 0.8\n",
    "lw = 3\n",
    "l1cind = 0\n",
    "l2cind = 1\n",
    "l3cind = 2\n",
    "l4cind = 3\n",
    "l5cind = 4\n",
    "\n",
    "################################\n",
    "# SSP5\n",
    "################################\n",
    "tot_ssp5 = ssp5_ra_r2 + ssp5_eis_r2 + ssp5_prtp_r2 + ssp5_tech_r2\n",
    "ax[0].plot(node_times, ssp5_ra_r2 * tot_ssp5**(-1), linestyle='dashed', linewidth=lw, \n",
    "        color=color_list[l1cind])\n",
    "ax[0].fill_between(node_times, np.zeros_like(node_times), ssp5_ra_r2 * tot_ssp5**(-1), alpha=alpha, \n",
    "                color=color_list[l1cind], label=\"RA\")\n",
    "\n",
    "ax[0].plot(node_times, (ssp5_eis_r2 + ssp5_ra_r2)  * tot_ssp5**(-1), linestyle='dashed', linewidth=lw, \n",
    "        color=color_list[l2cind])\n",
    "ax[0].fill_between(node_times, ssp5_ra_r2  * tot_ssp5**(-1), \n",
    "                   (ssp5_ra_r2 + ssp5_eis_r2) * tot_ssp5**(-1), alpha=alpha, \n",
    "                color=color_list[l2cind], label=\"EIS\")\n",
    "\n",
    "ax[0].plot(node_times, (ssp5_eis_r2 + ssp5_ra_r2 + ssp5_prtp_r2)  * tot_ssp5**(-1), linestyle='dashed', linewidth=lw, \n",
    "        color=color_list[l3cind])\n",
    "ax[0].fill_between(node_times, (ssp5_eis_r2 + ssp5_ra_r2) * tot_ssp5**(-1), \n",
    "                  (ssp5_ra_r2 + ssp5_eis_r2 + ssp5_prtp_r2) * tot_ssp5**(-1), alpha=alpha, \n",
    "                color=color_list[l3cind], label=\"PRTP\")\n",
    "\n",
    "ax[0].plot(node_times, (ssp5_eis_r2 + ssp5_ra_r2 + ssp5_prtp_r2 + ssp5_tech_r2) * tot_ssp5**(-1), \n",
    "           linestyle='dashed', linewidth=lw, color=color_list[l4cind])\n",
    "ax[0].fill_between(node_times, (ssp5_eis_r2 + ssp5_ra_r2 + ssp5_prtp_r2) * tot_ssp5**(-1), \n",
    "                   (ssp5_ra_r2 + ssp5_eis_r2 + ssp5_prtp_r2 + ssp5_tech_r2) * tot_ssp5**(-1), alpha=alpha, \n",
    "                color=color_list[l4cind], label=\"Tech. Growth\")\n",
    "\n",
    "\n",
    "################################\n",
    "# SSP4\n",
    "################################\n",
    "tot_ssp4 = ssp4_ra_r2 + ssp4_eis_r2 + ssp4_prtp_r2 + ssp4_tech_r2\n",
    "ax[1].plot(node_times, ssp4_ra_r2 * tot_ssp4**(-1), linestyle='dashed', linewidth=lw, \n",
    "        color=color_list[l1cind])\n",
    "ax[1].fill_between(node_times, np.zeros_like(node_times), ssp4_ra_r2 * tot_ssp4**(-1), alpha=alpha, \n",
    "                color=color_list[l1cind], label=\"RA\")\n",
    "\n",
    "ax[1].plot(node_times, (ssp4_eis_r2 + ssp4_ra_r2) * tot_ssp4**(-1), linestyle='dashed', linewidth=lw, \n",
    "        color=color_list[l2cind])\n",
    "ax[1].fill_between(node_times, ssp4_ra_r2 * tot_ssp4**(-1), \n",
    "                   (ssp4_ra_r2 + ssp4_eis_r2) * tot_ssp4**(-1), alpha=alpha, \n",
    "                color=color_list[l2cind], label=\"EIS\")\n",
    "\n",
    "ax[1].plot(node_times, (ssp4_eis_r2 + ssp4_ra_r2 + ssp4_prtp_r2) * tot_ssp4**(-1), linestyle='dashed', linewidth=lw, \n",
    "        color=color_list[l3cind])\n",
    "ax[1].fill_between(node_times, (ssp4_eis_r2 + ssp4_ra_r2) * tot_ssp4**(-1), \n",
    "                  (ssp4_ra_r2 + ssp4_eis_r2 + ssp4_prtp_r2) * tot_ssp4**(-1), alpha=alpha, \n",
    "                color=color_list[l3cind], label=\"PRTP\")\n",
    "\n",
    "ax[1].plot(node_times, (ssp4_eis_r2 + ssp4_ra_r2 + ssp4_prtp_r2 + ssp4_tech_r2) * tot_ssp4**(-1), \n",
    "           linestyle='dashed', linewidth=lw, color=color_list[l4cind])\n",
    "ax[1].fill_between(node_times, (ssp4_eis_r2 + ssp4_ra_r2 + ssp4_prtp_r2) * tot_ssp4**(-1), \n",
    "                   (ssp4_ra_r2 + ssp4_eis_r2 + ssp4_prtp_r2 + ssp4_tech_r2) * tot_ssp4**(-1), alpha=alpha, \n",
    "                color=color_list[l4cind], label=\"Tech. Growth\")\n",
    "\n",
    "################################\n",
    "# SSP3\n",
    "################################\n",
    "ssp3_tot = ssp3_ra_r2 + ssp3_eis_r2 + ssp3_prtp_r2 + ssp3_tech_r2\n",
    "ax[2].plot(node_times, ssp3_ra_r2 * ssp3_tot**(-1), linestyle='dashed', linewidth=lw, \n",
    "        color=color_list[l1cind])\n",
    "ax[2].fill_between(node_times, np.zeros_like(node_times), ssp3_ra_r2 * ssp3_tot**(-1), alpha=alpha, \n",
    "                color=color_list[l1cind], label=\"RA\")\n",
    "\n",
    "ax[2].plot(node_times, (ssp3_eis_r2 + ssp3_ra_r2) * ssp3_tot**(-1), linestyle='dashed', linewidth=lw, \n",
    "        color=color_list[l2cind])\n",
    "ax[2].fill_between(node_times, ssp3_ra_r2 * ssp3_tot**(-1), \n",
    "                   (ssp3_ra_r2 + ssp3_eis_r2) * ssp3_tot**(-1), alpha=alpha, \n",
    "                color=color_list[l2cind], label=\"EIS\")\n",
    "\n",
    "ax[2].plot(node_times, (ssp3_eis_r2 + ssp3_ra_r2 + ssp3_prtp_r2) * ssp3_tot**(-1), linestyle='dashed', linewidth=lw, \n",
    "        color=color_list[l3cind])\n",
    "ax[2].fill_between(node_times, (ssp3_eis_r2 + ssp3_ra_r2) * ssp3_tot**(-1), \n",
    "                   (ssp3_ra_r2 + ssp3_eis_r2 + ssp3_prtp_r2) * ssp3_tot**(-1), alpha=alpha, \n",
    "                color=color_list[l3cind], label=\"PRTP\")\n",
    "\n",
    "ax[2].plot(node_times, (ssp3_eis_r2 + ssp3_ra_r2 + ssp3_prtp_r2 + ssp3_tech_r2) * ssp3_tot**(-1), linestyle='dashed', linewidth=lw, \n",
    "        color=color_list[l4cind])\n",
    "ax[2].fill_between(node_times, (ssp3_eis_r2 + ssp3_ra_r2 + ssp3_prtp_r2) * ssp3_tot**(-1), \n",
    "                   (ssp3_ra_r2 + ssp3_eis_r2 + ssp3_prtp_r2 + ssp3_tech_r2) * ssp3_tot**(-1), alpha=alpha, \n",
    "                color=color_list[l4cind], label=\"Tech. Growth\")\n",
    "\n",
    "################################\n",
    "# SSP2\n",
    "################################\n",
    "ssp2_tot = ssp2_ra_r2 + ssp2_eis_r2 + ssp2_prtp_r2 + ssp2_tech_r2\n",
    "ax[3].plot(node_times, ssp2_ra_r2 * ssp2_tot**(-1), linestyle='dashed', linewidth=lw, \n",
    "        color=color_list[l1cind])\n",
    "ax[3].fill_between(node_times, np.zeros_like(node_times), ssp2_ra_r2* ssp2_tot**(-1), alpha=alpha, \n",
    "                color=color_list[l1cind], label=\"RA\")\n",
    "\n",
    "ax[3].plot(node_times, (ssp2_eis_r2 + ssp2_ra_r2)* ssp2_tot**(-1), linestyle='dashed', linewidth=lw, \n",
    "        color=color_list[l2cind])\n",
    "ax[3].fill_between(node_times, ssp2_ra_r2 * ssp2_tot**(-1), \n",
    "                   (ssp2_ra_r2 + ssp2_eis_r2) * ssp2_tot**(-1), alpha=alpha, \n",
    "                color=color_list[l2cind], label=\"EIS\")\n",
    "\n",
    "ax[3].plot(node_times, (ssp2_eis_r2 + ssp2_ra_r2 + ssp2_prtp_r2) * ssp2_tot**(-1), linestyle='dashed', linewidth=lw, \n",
    "        color=color_list[l3cind])\n",
    "ax[3].fill_between(node_times, (ssp2_eis_r2 + ssp2_ra_r2) * ssp2_tot**(-1), \n",
    "                   (ssp2_ra_r2 + ssp2_eis_r2 + ssp2_prtp_r2) * ssp2_tot**(-1), alpha=alpha, \n",
    "                color=color_list[l3cind], label=\"PRTP\")\n",
    "\n",
    "ax[3].plot(node_times, (ssp2_eis_r2 + ssp2_ra_r2 + ssp2_prtp_r2 + ssp2_tech_r2) * ssp2_tot**(-1), linestyle='dashed', linewidth=lw, \n",
    "        color=color_list[l4cind])\n",
    "ax[3].fill_between(node_times, (ssp2_eis_r2 + ssp2_ra_r2 + ssp2_prtp_r2) * ssp2_tot**(-1), \n",
    "                   (ssp2_ra_r2 + ssp2_eis_r2 + ssp2_prtp_r2 + ssp2_tech_r2) * ssp2_tot**(-1), alpha=alpha, \n",
    "                color=color_list[l4cind], label=\"Tech. Growth\")\n",
    "\n",
    "\n",
    "################################\n",
    "# SSP1\n",
    "################################\n",
    "ssp1_tot = ssp1_ra_r2 + ssp1_eis_r2 + ssp1_prtp_r2 + ssp1_tech_r2\n",
    "ax[4].plot(node_times, ssp1_ra_r2 * ssp1_tot**(-1), linestyle='dashed', linewidth=lw, \n",
    "        color=color_list[l1cind])\n",
    "ax[4].fill_between(node_times, np.zeros_like(node_times), ssp5_ra_r2 * ssp1_tot**(-1), alpha=alpha, \n",
    "                color=color_list[l1cind], label=\"RA\")\n",
    "\n",
    "ax[4].plot(node_times, (ssp1_eis_r2 + ssp1_ra_r2) * ssp1_tot**(-1), linestyle='dashed', linewidth=lw, \n",
    "        color=color_list[l2cind])\n",
    "ax[4].fill_between(node_times, ssp1_ra_r2 * ssp1_tot**(-1), \n",
    "                   (ssp1_ra_r2 + ssp1_eis_r2) * ssp1_tot**(-1), alpha=alpha, \n",
    "                    color=color_list[l2cind], label=\"EIS\")\n",
    "\n",
    "ax[4].plot(node_times, (ssp1_eis_r2 + ssp1_ra_r2 + ssp1_prtp_r2) * ssp1_tot**(-1), linestyle='dashed', linewidth=lw, \n",
    "        color=color_list[l3cind])\n",
    "ax[4].fill_between(node_times, (ssp1_eis_r2 + ssp1_ra_r2) * ssp1_tot**(-1), \n",
    "                   (ssp1_ra_r2 + ssp1_eis_r2 + ssp1_prtp_r2) * ssp1_tot**(-1), alpha=alpha, \n",
    "                    color=color_list[l3cind], label=\"PRTP\")\n",
    "\n",
    "ax[4].plot(node_times, (ssp1_eis_r2 + ssp1_ra_r2 + ssp1_prtp_r2 + ssp1_tech_r2) * ssp1_tot**(-1), linestyle='dashed', linewidth=lw, \n",
    "        color=color_list[l4cind])\n",
    "ax[4].fill_between(node_times, (ssp1_eis_r2 + ssp1_ra_r2 + ssp1_prtp_r2) * ssp1_tot**(-1), \n",
    "                   (ssp1_ra_r2 + ssp1_eis_r2 + ssp1_prtp_r2 + ssp1_tech_r2) * ssp1_tot**(-1), alpha=alpha, \n",
    "                color=color_list[l4cind], label=\"Tech. Growth\")\n",
    "\n",
    "ax[2].legend(loc='upper center', bbox_to_anchor=(0.5, -0.25), fancybox=True, shadow=True, ncol=5, fontsize=20)\n",
    "\n",
    "ax[0].set_ylabel(\"$r^{2} / r_{tot}^{2}$\")\n",
    "ax[0].set_title(\"SSP5\")\n",
    "ax[1].set_title(\"SSP4\")\n",
    "ax[2].set_title(\"SSP3\")\n",
    "ax[3].set_title(\"SSP2\")\n",
    "ax[4].set_title(\"SSP1\")\n",
    "\n",
    "panel_label = [\"A\", \"B\", \"C\", \"D\", \"E\"]\n",
    "for i in range(5):\n",
    "    ax[i].set_xlim((node_times[0], node_times[-1]))\n",
    "    ax[i].set_aspect(1.0/ax[i].get_data_ratio(), adjustable='box')\n",
    "    ax[i].set_xlabel(\"Year\")\n",
    "    ax[i].text(2168, 0.88, panel_label[i], fontsize=22, color='white', fontweight='bold')\n",
    "    ax[i].set_xticks([2020,2100,2200])\n",
    "    ax[i].set_xticklabels(['2020', '2100', '2200'])\n",
    "    ax[i].tick_params(axis='both', labelsize=14)\n",
    "\n",
    "sns.despine(trim=True, offset=10)\n",
    "\n",
    "if save_fig:\n",
    "    fig.savefig(''.join([basefile, 'rsquared.png']), dpi=400)"
   ]
  },
  {
   "cell_type": "markdown",
   "id": "415fef9a",
   "metadata": {},
   "source": [
    "# And now, temperature"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "0ea79f05",
   "metadata": {},
   "outputs": [],
   "source": [
    "fig, ax = plt.subplots(4, 5, figsize=(30, 20))\n",
    "\n",
    "ax[0,0].set_title(\"2030\")\n",
    "ax[0,1].set_title(\"2060\")\n",
    "ax[0,2].set_title(\"2100\")\n",
    "ax[0,3].set_title(\"2150\")\n",
    "ax[0,4].set_title(\"2200\")\n",
    "\n",
    "###########################\n",
    "## RISK AVERSION ROW\n",
    "###########################\n",
    "make_panel_w_regression(ax, 0, 0, ssp5_ra, ssp4_ra, ssp3_ra, ssp2_ra, ssp1_ra, ssp5_t2030, ssp4_t2030, ssp3_t2030, ssp2_t2030, ssp1_t2030, \n",
    "           \"Risk Aversion\", \"Temperature ($^{\\circ}$C)\")\n",
    "\n",
    "make_panel_w_regression(ax, 0, 1, ssp5_ra, ssp4_ra, ssp3_ra, ssp2_ra, ssp1_ra, ssp5_t2060, ssp4_t2060, ssp3_t2060, ssp2_t2060, ssp1_t2060, \n",
    "           \"Risk Aversion\", None)\n",
    "\n",
    "make_panel_w_regression(ax, 0, 2, ssp5_ra, ssp4_ra, ssp3_ra, ssp2_ra, ssp1_ra, ssp5_t2100, ssp4_t2100, ssp3_t2100, ssp2_t2100, ssp1_t2100, \n",
    "           \"Risk Aversion\", None)\n",
    "\n",
    "make_panel_w_regression(ax, 0, 3, ssp5_ra, ssp4_ra, ssp3_ra, ssp2_ra, ssp1_ra, ssp5_t2150, ssp4_t2150, ssp3_t2150, ssp2_t2150, ssp1_t2150, \n",
    "           \"Risk Aversion\", None)\n",
    "\n",
    "make_panel_w_regression(ax, 0, 4, ssp5_ra, ssp4_ra, ssp3_ra, ssp2_ra, ssp1_ra, ssp5_t2200, ssp4_t2200, ssp3_t2200, ssp2_t2200, ssp1_t2200, \n",
    "           \"Risk Aversion\", None)\n",
    "\n",
    "###########################\n",
    "# EIS ROW\n",
    "###########################\n",
    "make_panel_w_regression(ax, 1, 0, ssp5_eis, ssp4_eis, ssp3_eis, ssp2_eis, ssp1_eis, ssp5_t2030, ssp4_t2030, ssp3_t2030, ssp2_t2030, ssp1_t2030, \n",
    "           \"EIS\", \"Temperature ($^{\\circ}$C)\")\n",
    "\n",
    "make_panel_w_regression(ax, 1, 1, ssp5_eis, ssp4_eis, ssp3_eis, ssp2_eis, ssp1_eis, ssp5_t2060, ssp4_t2060, ssp3_t2060, ssp2_t2060, ssp1_t2060, \n",
    "           \"EIS\", None)\n",
    "\n",
    "make_panel_w_regression(ax, 1, 2, ssp5_eis, ssp4_eis, ssp3_eis, ssp2_eis, ssp1_eis, ssp5_t2100, ssp4_t2100, ssp3_t2100, ssp2_t2100, ssp1_t2100, \n",
    "           \"EIS\", None)\n",
    "\n",
    "make_panel_w_regression(ax, 1, 3, ssp5_eis, ssp4_eis, ssp3_eis, ssp2_eis, ssp1_eis, ssp5_t2150, ssp4_t2150, ssp3_t2150, ssp2_t2150, ssp1_t2150, \n",
    "           \"EIS\", None)\n",
    "\n",
    "make_panel_w_regression(ax, 1, 4, ssp5_eis, ssp4_eis, ssp3_eis, ssp2_eis, ssp1_eis, ssp5_t2200, ssp4_t2200, ssp3_t2200, ssp2_t2200, ssp1_t2200, \n",
    "           \"EIS\", None)\n",
    "\n",
    "###########################\n",
    "# PRTP ROW\n",
    "###########################\n",
    "make_panel_w_regression(ax, 2, 0, ssp5_prtp * 100, ssp4_prtp * 100, ssp3_prtp * 100, ssp2_prtp * 100, ssp1_prtp * 100, ssp5_t2030, ssp4_t2030, ssp3_t2030, ssp2_t2030, ssp1_t2030, \n",
    "           r\"$\\delta$ (%)\", \"Temperature ($^{\\circ}$C)\")\n",
    "\n",
    "make_panel_w_regression(ax, 2, 1, ssp5_prtp * 100, ssp4_prtp * 100, ssp3_prtp * 100, ssp2_prtp * 100, ssp1_prtp * 100, ssp5_t2060, ssp4_t2060, ssp3_t2060, ssp2_t2060, ssp1_t2060, \n",
    "           r\"$\\delta$ (%)\", None)\n",
    "\n",
    "make_panel_w_regression(ax, 2, 2, ssp5_prtp * 100, ssp4_prtp * 100, ssp3_prtp * 100, ssp2_prtp * 100, ssp1_prtp * 100, ssp5_t2100, ssp4_t2100, ssp3_t2100, ssp2_t2100, ssp1_t2100, \n",
    "           r\"$\\delta$ (%)\", None)\n",
    "\n",
    "make_panel_w_regression(ax, 2, 3, ssp5_prtp * 100, ssp4_prtp * 100, ssp3_prtp * 100, ssp2_prtp * 100, ssp1_prtp * 100, ssp5_t2150, ssp4_t2150, ssp3_t2150, ssp2_t2150, ssp1_t2150, \n",
    "           r\"$\\delta$ (%)\", None)\n",
    "\n",
    "make_panel_w_regression(ax, 2, 4, ssp5_prtp * 100, ssp4_prtp * 100, ssp3_prtp * 100, ssp2_prtp * 100, ssp1_prtp * 100, ssp5_t2200, ssp4_t2200, ssp3_t2200, ssp2_t2200, ssp1_t2200, \n",
    "           r\"$\\delta$ (%)\", None)\n",
    "\n",
    "###########################\n",
    "# EX TECH GROWTH ROW\n",
    "###########################\n",
    "make_panel_w_regression(ax, 3, 0, ssp5_tottech[:, 1], ssp4_tottech[:, 1], ssp3_tottech[:, 1], \n",
    "                        ssp2_tottech[:, 1], ssp1_tottech[:, 1], ssp5_t2030, ssp4_t2030, ssp3_t2030, \n",
    "                        ssp2_t2030, ssp1_t2030, \"Tech Growth (%)\", \"Temperature ($^{\\circ}$C)\")\n",
    "\n",
    "make_panel_w_regression(ax, 3, 1, ssp5_tottech[:, 2], ssp4_tottech[:, 2], ssp3_tottech[:, 2], \n",
    "                        ssp2_tottech[:, 2], ssp1_tottech[:, 2], ssp5_t2060, ssp4_t2060, ssp3_t2060, \n",
    "                        ssp2_t2060, ssp1_t2060, \"Tech Growth (%)\", None)\n",
    "\n",
    "make_panel_w_regression(ax, 3, 2, ssp5_tottech[:, 3], ssp4_tottech[:, 3], ssp3_tottech[:, 3], \n",
    "                        ssp2_tottech[:, 3], ssp1_tottech[:, 3], ssp5_t2100, ssp4_t2100, ssp3_t2100, \n",
    "                        ssp2_t2100, ssp1_t2100, \"Tech Growth (%)\", None)\n",
    "\n",
    "make_panel_w_regression(ax, 3, 3, ssp5_tottech[:, 4], ssp4_tottech[:, 4], ssp3_tottech[:, 4], \n",
    "                        ssp2_tottech[:, 4], ssp1_tottech[:, 4], ssp5_t2150, ssp4_t2150, ssp3_t2150, \n",
    "                        ssp2_t2150, ssp1_t2150, \"Tech Growth (%)\", None)\n",
    "\n",
    "make_panel_w_regression(ax, 3, 4, ssp5_tottech[:, 5], ssp4_tottech[:, 5], ssp3_tottech[:, 5], \n",
    "                        ssp2_tottech[:, 5], ssp1_tottech[:, 5], ssp5_t2200, ssp4_t2200, ssp3_t2200, \n",
    "                        ssp2_t2200, ssp1_t2200, \"Tech Growth (%)\", None)\n",
    "\n",
    "ax[3,0].set_xticks([0,1,2,3])\n",
    "ax[3,0].set_xlim((0,3))\n",
    "ax[3,1].set_xticks([0,1,2,3,4,5])\n",
    "ax[3,2].set_xticks([0,1,2,3,4,5,6])\n",
    "ax[3,3].set_xticks([0,1,2,3,4,5,6])\n",
    "ax[3,4].set_xticks([0,1,2,3,4,5,6])\n",
    "\n",
    "fig.tight_layout()\n",
    "sns.despine(trim=True, offset=10)\n",
    "\n",
    "if save_fig:\n",
    "    fig.savefig(''.join([basefile, 'regression-sticky-t.png']), dpi=400)    "
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "3a361d05",
   "metadata": {},
   "outputs": [],
   "source": [
    "def get_tech_r2_in_time_alllin(tottech, paths):\n",
    "    c2020, c2030, c2060, c2100, c2150, c2200 = paths.T\n",
    "    _, _, r2_2020 = get_regression(tottech[:,0], c2020)\n",
    "    _, _, r2_2030 = get_regression(tottech[:,1], c2030)\n",
    "    _, _, r2_2060 = get_regression(tottech[:,2], c2060)\n",
    "    _, _, r2_2100 = get_regression(tottech[:,3], c2100)\n",
    "    _, _, r2_2150 = get_regression(tottech[:,4], c2150)\n",
    "    _, _, r2_2200 = get_regression(tottech[:,5], c2200)\n",
    "    \n",
    "    r2 = np.array([r2_2020, r2_2030, r2_2060, r2_2100, r2_2150, r2_2200])\n",
    "    return r2"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "09bebff1",
   "metadata": {},
   "outputs": [],
   "source": [
    "ssp5_eis_r2_t = get_r2_in_time(ssp5_eis, ssp5_exp_t)\n",
    "ssp5_ra_r2_t = get_r2_in_time(ssp5_ra, ssp5_exp_t)\n",
    "ssp5_prtp_r2_t = get_r2_in_time(ssp5_prtp, ssp5_exp_t)\n",
    "ssp5_tech_r2_t = get_tech_r2_in_time_alllin(ssp5_tottech, ssp5_exp_t)\n",
    "\n",
    "ssp4_eis_r2_t = get_r2_in_time(ssp4_eis, ssp4_exp_t)\n",
    "ssp4_ra_r2_t = get_r2_in_time(ssp4_ra, ssp4_exp_t)\n",
    "ssp4_prtp_r2_t = get_r2_in_time(ssp4_prtp, ssp4_exp_t)\n",
    "ssp4_tech_r2_t = get_tech_r2_in_time_alllin(ssp4_tottech, ssp4_exp_t)\n",
    "\n",
    "ssp3_eis_r2_t = get_r2_in_time(ssp3_eis, ssp3_exp_t)\n",
    "ssp3_ra_r2_t = get_r2_in_time(ssp3_ra, ssp3_exp_t)\n",
    "ssp3_prtp_r2_t = get_r2_in_time(ssp3_prtp, ssp3_exp_t)\n",
    "ssp3_tech_r2_t = get_tech_r2_in_time_alllin(ssp3_tottech, ssp3_exp_t)\n",
    "\n",
    "ssp2_eis_r2_t = get_r2_in_time(ssp2_eis, ssp2_exp_t)\n",
    "ssp2_ra_r2_t = get_r2_in_time(ssp2_ra, ssp2_exp_t)\n",
    "ssp2_prtp_r2_t = get_r2_in_time(ssp2_prtp, ssp2_exp_t)\n",
    "ssp2_tech_r2_t = get_tech_r2_in_time_alllin(ssp2_tottech, ssp2_exp_t)\n",
    "\n",
    "ssp1_eis_r2_t = get_r2_in_time(ssp1_eis, ssp1_exp_t)\n",
    "ssp1_ra_r2_t = get_r2_in_time(ssp1_ra, ssp1_exp_t)\n",
    "ssp1_prtp_r2_t = get_r2_in_time(ssp1_prtp, ssp1_exp_t)\n",
    "ssp1_tech_r2_t = get_tech_r2_in_time_alllin(ssp1_tottech, ssp1_exp_t)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "fe0531f2",
   "metadata": {},
   "outputs": [],
   "source": [
    "fig, ax = plt.subplots(1,5, figsize=(20,20), sharey=True)\n",
    "\n",
    "ax[0].set_ylim((0,1))\n",
    "\n",
    "alpha = 0.8\n",
    "lw = 3\n",
    "l1cind = 0\n",
    "l2cind = 1\n",
    "l3cind = 2\n",
    "l4cind = 3\n",
    "l5cind = 4\n",
    "\n",
    "################################\n",
    "# SSP5\n",
    "################################\n",
    "ax[0].plot(node_times[1:], ssp5_ra_r2_t[1:], linestyle='dashed', linewidth=lw, \n",
    "        color=color_list[l1cind])\n",
    "ax[0].fill_between(node_times[1:], np.zeros_like(node_times[1:]), ssp5_ra_r2_t[1:], alpha=alpha, \n",
    "                color=color_list[l1cind], label=\"RA\")\n",
    "\n",
    "ax[0].plot(node_times[1:], ssp5_eis_r2_t[1:] + ssp5_ra_r2_t[1:], linestyle='dashed', linewidth=lw, \n",
    "        color=color_list[l2cind])\n",
    "ax[0].fill_between(node_times[1:], ssp5_ra_r2_t[1:], ssp5_ra_r2_t[1:] + ssp5_eis_r2_t[1:], alpha=alpha, \n",
    "                color=color_list[l2cind], label=\"EIS\")\n",
    "\n",
    "ax[0].plot(node_times[1:], ssp5_eis_r2_t[1:] + ssp5_ra_r2_t[1:] + ssp5_prtp_r2_t[1:], linestyle='dashed', linewidth=lw, \n",
    "        color=color_list[l3cind])\n",
    "ax[0].fill_between(node_times[1:], ssp5_eis_r2_t[1:] + ssp5_ra_r2_t[1:], ssp5_ra_r2_t[1:] + ssp5_eis_r2_t[1:] + ssp5_prtp_r2_t[1:], alpha=alpha, \n",
    "                color=color_list[l3cind], label=\"PRTP\")\n",
    "\n",
    "ax[0].plot(node_times[1:], ssp5_eis_r2_t[1:] + ssp5_ra_r2_t[1:] + ssp5_prtp_r2_t[1:] + ssp5_tech_r2_t[1:], linestyle='dashed', linewidth=lw, \n",
    "        color=color_list[l4cind])\n",
    "ax[0].fill_between(node_times[1:], ssp5_eis_r2_t[1:] + ssp5_ra_r2_t[1:] + ssp5_prtp_r2_t[1:], ssp5_ra_r2_t[1:] + ssp5_eis_r2_t[1:] + ssp5_prtp_r2_t[1:] + ssp5_tech_r2_t[1:], alpha=alpha, \n",
    "                color=color_list[l4cind], label=\"Tech. Growth\")\n",
    "\n",
    "\n",
    "################################\n",
    "# SSP4\n",
    "################################\n",
    "ax[1].plot(node_times[1:], ssp4_ra_r2_t[1:], linestyle='dashed', linewidth=lw, \n",
    "        color=color_list[l1cind])\n",
    "ax[1].fill_between(node_times[1:], np.zeros_like(node_times[1:]), ssp4_ra_r2_t[1:], alpha=alpha, \n",
    "                color=color_list[l1cind], label=\"RA\")\n",
    "\n",
    "ax[1].plot(node_times[1:], ssp4_eis_r2_t[1:] + ssp4_ra_r2_t[1:], linestyle='dashed', linewidth=lw, \n",
    "        color=color_list[l2cind])\n",
    "ax[1].fill_between(node_times[1:], ssp4_ra_r2_t[1:], ssp5_ra_r2_t[1:] + ssp4_eis_r2_t[1:], alpha=alpha, \n",
    "                color=color_list[l2cind], label=\"EIS\")\n",
    "\n",
    "ax[1].plot(node_times[1:], ssp4_eis_r2_t[1:] + ssp4_ra_r2_t[1:] + ssp4_prtp_r2_t[1:], linestyle='dashed', linewidth=lw, \n",
    "        color=color_list[l3cind])\n",
    "ax[1].fill_between(node_times[1:], ssp4_eis_r2_t[1:] + ssp4_ra_r2_t[1:], ssp4_ra_r2_t[1:] + ssp4_eis_r2_t[1:] + ssp4_prtp_r2_t[1:], alpha=alpha, \n",
    "                color=color_list[l3cind], label=\"PRTP\")\n",
    "\n",
    "ax[1].plot(node_times[1:], ssp4_eis_r2_t[1:] + ssp4_ra_r2_t[1:] + ssp4_prtp_r2_t[1:] + ssp4_tech_r2_t[1:], linestyle='dashed', linewidth=lw, \n",
    "        color=color_list[l4cind])\n",
    "ax[1].fill_between(node_times[1:], ssp4_eis_r2_t[1:] + ssp4_ra_r2_t[1:] + ssp4_prtp_r2_t[1:], ssp4_ra_r2_t[1:] + ssp4_eis_r2_t[1:] + ssp4_prtp_r2_t[1:] + ssp4_tech_r2_t[1:], alpha=alpha, \n",
    "                color=color_list[l4cind], label=\"Tech. Growth\")\n",
    "\n",
    "################################\n",
    "# SSP3\n",
    "################################\n",
    "ax[2].plot(node_times[1:], ssp3_ra_r2_t[1:], linestyle='dashed', linewidth=lw, \n",
    "        color=color_list[l1cind])\n",
    "ax[2].fill_between(node_times[1:], np.zeros_like(node_times[1:]), ssp3_ra_r2_t[1:], alpha=alpha, \n",
    "                color=color_list[l1cind], label=\"RA\")\n",
    "\n",
    "ax[2].plot(node_times[1:], ssp3_eis_r2_t[1:] + ssp3_ra_r2_t[1:], linestyle='dashed', linewidth=lw, \n",
    "        color=color_list[l2cind])\n",
    "ax[2].fill_between(node_times[1:], ssp3_ra_r2_t[1:], ssp3_ra_r2_t[1:] + ssp3_eis_r2_t[1:], alpha=alpha, \n",
    "                color=color_list[l2cind], label=\"EIS\")\n",
    "\n",
    "ax[2].plot(node_times[1:], ssp3_eis_r2_t[1:] + ssp3_ra_r2_t[1:] + ssp3_prtp_r2_t[1:], linestyle='dashed', linewidth=lw, \n",
    "        color=color_list[l3cind])\n",
    "ax[2].fill_between(node_times[1:], ssp3_eis_r2_t[1:] + ssp3_ra_r2_t[1:], ssp3_ra_r2_t[1:] + ssp3_eis_r2_t[1:] + ssp3_prtp_r2_t[1:], alpha=alpha, \n",
    "                color=color_list[l3cind], label=\"PRTP\")\n",
    "\n",
    "ax[2].plot(node_times[1:], ssp3_eis_r2_t[1:] + ssp3_ra_r2_t[1:] + ssp3_prtp_r2_t[1:] + ssp3_tech_r2_t[1:], linestyle='dashed', linewidth=lw, \n",
    "        color=color_list[l4cind])\n",
    "ax[2].fill_between(node_times[1:], ssp3_eis_r2_t[1:] + ssp3_ra_r2_t[1:] + ssp3_prtp_r2_t[1:], ssp3_ra_r2_t[1:] + ssp3_eis_r2_t[1:] + ssp3_prtp_r2_t[1:] + ssp3_tech_r2_t[1:], alpha=alpha, \n",
    "                color=color_list[l4cind], label=\"Tech. Growth\")\n",
    "\n",
    "################################\n",
    "# SSP2\n",
    "################################\n",
    "ax[3].plot(node_times[1:], ssp2_ra_r2_t[1:], linestyle='dashed', linewidth=lw, \n",
    "        color=color_list[l1cind])\n",
    "ax[3].fill_between(node_times[1:], np.zeros_like(node_times[1:]), ssp2_ra_r2_t[1:], alpha=alpha, \n",
    "                color=color_list[l1cind], label=\"RA\")\n",
    "\n",
    "ax[3].plot(node_times[1:], ssp2_eis_r2_t[1:] + ssp2_ra_r2_t[1:], linestyle='dashed', linewidth=lw, \n",
    "        color=color_list[l2cind])\n",
    "ax[3].fill_between(node_times[1:], ssp2_ra_r2_t[1:], ssp2_ra_r2_t[1:] + ssp2_eis_r2_t[1:], alpha=alpha, \n",
    "                color=color_list[l2cind], label=\"EIS\")\n",
    "\n",
    "ax[3].plot(node_times[1:], ssp2_eis_r2_t[1:] + ssp2_ra_r2_t[1:] + ssp2_prtp_r2_t[1:], linestyle='dashed', linewidth=lw, \n",
    "        color=color_list[l3cind])\n",
    "ax[3].fill_between(node_times[1:], ssp2_eis_r2_t[1:] + ssp2_ra_r2_t[1:], ssp2_ra_r2_t[1:] + ssp2_eis_r2_t[1:] + ssp2_prtp_r2_t[1:], alpha=alpha, \n",
    "                color=color_list[l3cind], label=\"PRTP\")\n",
    "\n",
    "ax[3].plot(node_times[1:], ssp2_eis_r2_t[1:] + ssp2_ra_r2_t[1:] + ssp2_prtp_r2_t[1:] + ssp2_tech_r2_t[1:], linestyle='dashed', linewidth=lw, \n",
    "        color=color_list[l4cind])\n",
    "ax[3].fill_between(node_times[1:], ssp2_eis_r2_t[1:] + ssp2_ra_r2_t[1:] + ssp2_prtp_r2_t[1:], ssp2_ra_r2_t[1:] + ssp2_eis_r2_t[1:] + ssp2_prtp_r2_t[1:] + ssp2_tech_r2_t[1:], alpha=alpha, \n",
    "                color=color_list[l4cind], label=\"Tech. Growth\")\n",
    "\n",
    "\n",
    "################################\n",
    "# SSP1\n",
    "################################\n",
    "ax[4].plot(node_times[1:], ssp1_ra_r2_t[1:], linestyle='dashed', linewidth=lw, \n",
    "        color=color_list[l1cind])\n",
    "ax[4].fill_between(node_times[1:], np.zeros_like(node_times[1:]), ssp5_ra_r2_t[1:], alpha=alpha, \n",
    "                color=color_list[l1cind], label=\"RA\")\n",
    "\n",
    "ax[4].plot(node_times[1:], ssp1_eis_r2_t[1:] + ssp1_ra_r2_t[1:], linestyle='dashed', linewidth=lw, \n",
    "        color=color_list[l2cind])\n",
    "ax[4].fill_between(node_times[1:], ssp1_ra_r2_t[1:], ssp1_ra_r2_t[1:] + ssp1_eis_r2_t[1:], alpha=alpha, \n",
    "                color=color_list[l2cind], label=\"EIS\")\n",
    "\n",
    "ax[4].plot(node_times[1:], ssp1_eis_r2_t[1:] + ssp1_ra_r2_t[1:] + ssp1_prtp_r2_t[1:], linestyle='dashed', linewidth=lw, \n",
    "        color=color_list[l3cind])\n",
    "ax[4].fill_between(node_times[1:], ssp1_eis_r2_t[1:] + ssp1_ra_r2_t[1:], ssp1_ra_r2_t[1:] + ssp1_eis_r2_t[1:] + ssp1_prtp_r2_t[1:], alpha=alpha, \n",
    "                color=color_list[l3cind], label=\"PRTP\")\n",
    "\n",
    "ax[4].plot(node_times[1:], ssp1_eis_r2_t[1:] + ssp1_ra_r2_t[1:] + ssp1_prtp_r2_t[1:] + ssp1_tech_r2_t[1:], linestyle='dashed', linewidth=lw, \n",
    "        color=color_list[l4cind])\n",
    "ax[4].fill_between(node_times[1:], ssp1_eis_r2_t[1:] + ssp1_ra_r2_t[1:] + ssp1_prtp_r2_t[1:], ssp1_ra_r2_t[1:] + ssp1_eis_r2_t[1:] + ssp1_prtp_r2_t[1:] + ssp1_tech_r2_t[1:], alpha=alpha, \n",
    "                color=color_list[l4cind], label=\"Tech. Growth\")\n",
    "\n",
    "ax[2].legend(loc='upper center', bbox_to_anchor=(0.5, -0.25), fancybox=True, shadow=True, ncol=5)\n",
    "\n",
    "ax[0].set_ylabel(\"r$^{2}$\")\n",
    "ax[0].set_title(\"SSP5\")\n",
    "ax[1].set_title(\"SSP4\")\n",
    "ax[2].set_title(\"SSP3\")\n",
    "ax[3].set_title(\"SSP2\")\n",
    "ax[4].set_title(\"SSP1\")\n",
    "\n",
    "for i in range(5):\n",
    "    ax[i].set_xlim((node_times[1], node_times[-1]))\n",
    "    ax[i].set_aspect(1.0/ax[i].get_data_ratio(), adjustable='box')\n",
    "    ax[i].set_xlabel(\"Year\")\n",
    "\n",
    "#fig.tight_layout()"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "daebee2d",
   "metadata": {},
   "outputs": [],
   "source": [
    "fig, ax = plt.subplots(1,5, figsize=(20,20), sharey=True)\n",
    "\n",
    "ax[0].set_ylim((0,1))\n",
    "\n",
    "alpha = 0.8\n",
    "lw = 3\n",
    "l1cind = 0\n",
    "l2cind = 1\n",
    "l3cind = 2\n",
    "l4cind = 3\n",
    "l5cind = 4\n",
    "\n",
    "################################\n",
    "# SSP5\n",
    "################################\n",
    "tot_ssp5 = ssp5_ra_r2_t[1:] + ssp5_eis_r2_t[1:] + ssp5_prtp_r2_t[1:] + ssp5_tech_r2_t[1:]\n",
    "ax[0].plot(node_times[1:], ssp5_ra_r2_t[1:] * tot_ssp5**(-1), linestyle='dashed', linewidth=lw, \n",
    "        color=color_list[l1cind])\n",
    "ax[0].fill_between(node_times[1:], np.zeros_like(node_times[1:]), ssp5_ra_r2_t[1:] * tot_ssp5**(-1), alpha=alpha, \n",
    "                color=color_list[l1cind], label=\"RA\")\n",
    "\n",
    "ax[0].plot(node_times[1:], (ssp5_eis_r2_t[1:] + ssp5_ra_r2_t[1:])  * tot_ssp5**(-1), linestyle='dashed', linewidth=lw, \n",
    "        color=color_list[l2cind])\n",
    "ax[0].fill_between(node_times[1:], ssp5_ra_r2_t[1:]  * tot_ssp5**(-1), \n",
    "                   (ssp5_ra_r2_t[1:] + ssp5_eis_r2_t[1:]) * tot_ssp5**(-1), alpha=alpha, \n",
    "                color=color_list[l2cind], label=\"EIS\")\n",
    "\n",
    "ax[0].plot(node_times[1:], (ssp5_eis_r2_t[1:] + ssp5_ra_r2_t[1:] + ssp5_prtp_r2_t[1:])  * tot_ssp5**(-1), linestyle='dashed', linewidth=lw, \n",
    "        color=color_list[l3cind])\n",
    "ax[0].fill_between(node_times[1:], (ssp5_eis_r2_t[1:] + ssp5_ra_r2_t[1:]) * tot_ssp5**(-1), \n",
    "                  (ssp5_ra_r2_t[1:] + ssp5_eis_r2_t[1:] + ssp5_prtp_r2_t[1:]) * tot_ssp5**(-1), alpha=alpha, \n",
    "                color=color_list[l3cind], label=\"PRTP\")\n",
    "\n",
    "ax[0].plot(node_times[1:], (ssp5_eis_r2_t[1:] + ssp5_ra_r2_t[1:] + ssp5_prtp_r2_t[1:] + ssp5_tech_r2_t[1:]) * tot_ssp5**(-1), \n",
    "           linestyle='dashed', linewidth=lw, color=color_list[l4cind])\n",
    "ax[0].fill_between(node_times[1:], (ssp5_eis_r2_t[1:] + ssp5_ra_r2_t[1:] + ssp5_prtp_r2_t[1:]) * tot_ssp5**(-1), \n",
    "                   (ssp5_ra_r2_t[1:] + ssp5_eis_r2_t[1:] + ssp5_prtp_r2_t[1:] + ssp5_tech_r2_t[1:]) * tot_ssp5**(-1), alpha=alpha, \n",
    "                color=color_list[l4cind], label=\"Tech. Growth\")\n",
    "\n",
    "\n",
    "################################\n",
    "# SSP4\n",
    "################################\n",
    "tot_ssp4 = ssp4_ra_r2_t[1:] + ssp4_eis_r2_t[1:] + ssp4_prtp_r2_t[1:] + ssp4_tech_r2_t[1:]\n",
    "ax[1].plot(node_times[1:], ssp4_ra_r2_t[1:] * tot_ssp4**(-1), linestyle='dashed', linewidth=lw, \n",
    "        color=color_list[l1cind])\n",
    "ax[1].fill_between(node_times[1:], np.zeros_like(node_times[1:]), ssp4_ra_r2_t[1:] * tot_ssp4**(-1), alpha=alpha, \n",
    "                color=color_list[l1cind], label=\"RA\")\n",
    "\n",
    "ax[1].plot(node_times[1:], (ssp4_eis_r2_t[1:] + ssp4_ra_r2_t[1:]) * tot_ssp4**(-1), linestyle='dashed', linewidth=lw, \n",
    "        color=color_list[l2cind])\n",
    "ax[1].fill_between(node_times[1:], ssp4_ra_r2_t[1:] * tot_ssp4**(-1), \n",
    "                   (ssp4_ra_r2_t[1:] + ssp4_eis_r2_t[1:]) * tot_ssp4**(-1), alpha=alpha, \n",
    "                color=color_list[l2cind], label=\"EIS\")\n",
    "\n",
    "ax[1].plot(node_times[1:], (ssp4_eis_r2_t[1:] + ssp4_ra_r2_t[1:] + ssp4_prtp_r2_t[1:]) * tot_ssp4**(-1), linestyle='dashed', linewidth=lw, \n",
    "        color=color_list[l3cind])\n",
    "ax[1].fill_between(node_times[1:], (ssp4_eis_r2_t[1:] + ssp4_ra_r2_t[1:]) * tot_ssp4**(-1), \n",
    "                  (ssp4_ra_r2_t[1:] + ssp4_eis_r2_t[1:] + ssp4_prtp_r2_t[1:]) * tot_ssp4**(-1), alpha=alpha, \n",
    "                color=color_list[l3cind], label=\"PRTP\")\n",
    "\n",
    "ax[1].plot(node_times[1:], (ssp4_eis_r2_t[1:] + ssp4_ra_r2_t[1:] + ssp4_prtp_r2_t[1:] + ssp4_tech_r2_t[1:]) * tot_ssp4**(-1), \n",
    "           linestyle='dashed', linewidth=lw, color=color_list[l4cind])\n",
    "ax[1].fill_between(node_times[1:], (ssp4_eis_r2_t[1:] + ssp4_ra_r2_t[1:] + ssp4_prtp_r2_t[1:]) * tot_ssp4**(-1), \n",
    "                   (ssp4_ra_r2_t[1:] + ssp4_eis_r2_t[1:] + ssp4_prtp_r2_t[1:] + ssp4_tech_r2_t[1:]) * tot_ssp4**(-1), alpha=alpha, \n",
    "                color=color_list[l4cind], label=\"Tech. Growth\")\n",
    "\n",
    "################################\n",
    "# SSP3\n",
    "################################\n",
    "ssp3_tot = ssp3_ra_r2_t[1:] + ssp3_eis_r2_t[1:] + ssp3_prtp_r2_t[1:] + ssp3_tech_r2_t[1:]\n",
    "ax[2].plot(node_times[1:], ssp3_ra_r2_t[1:] * ssp3_tot**(-1), linestyle='dashed', linewidth=lw, \n",
    "        color=color_list[l1cind])\n",
    "ax[2].fill_between(node_times[1:], np.zeros_like(node_times[1:]), ssp3_ra_r2_t[1:] * ssp3_tot**(-1), alpha=alpha, \n",
    "                color=color_list[l1cind], label=\"RA\")\n",
    "\n",
    "ax[2].plot(node_times[1:], (ssp3_eis_r2_t[1:] + ssp3_ra_r2_t[1:]) * ssp3_tot**(-1), linestyle='dashed', linewidth=lw, \n",
    "        color=color_list[l2cind])\n",
    "ax[2].fill_between(node_times[1:], ssp3_ra_r2_t[1:] * ssp3_tot**(-1), \n",
    "                   (ssp3_ra_r2_t[1:] + ssp3_eis_r2_t[1:]) * ssp3_tot**(-1), alpha=alpha, \n",
    "                color=color_list[l2cind], label=\"EIS\")\n",
    "\n",
    "ax[2].plot(node_times[1:], (ssp3_eis_r2_t[1:] + ssp3_ra_r2_t[1:] + ssp3_prtp_r2_t[1:]) * ssp3_tot**(-1), linestyle='dashed', linewidth=lw, \n",
    "        color=color_list[l3cind])\n",
    "ax[2].fill_between(node_times[1:], (ssp3_eis_r2_t[1:] + ssp3_ra_r2_t[1:]) * ssp3_tot**(-1), \n",
    "                   (ssp3_ra_r2_t[1:] + ssp3_eis_r2_t[1:] + ssp3_prtp_r2_t[1:]) * ssp3_tot**(-1), alpha=alpha, \n",
    "                color=color_list[l3cind], label=\"PRTP\")\n",
    "\n",
    "ax[2].plot(node_times[1:], (ssp3_eis_r2_t[1:] + ssp3_ra_r2_t[1:] + ssp3_prtp_r2_t[1:] + ssp3_tech_r2_t[1:]) * ssp3_tot**(-1), linestyle='dashed', linewidth=lw, \n",
    "        color=color_list[l4cind])\n",
    "ax[2].fill_between(node_times[1:], (ssp3_eis_r2_t[1:] + ssp3_ra_r2_t[1:] + ssp3_prtp_r2_t[1:]) * ssp3_tot**(-1), \n",
    "                   (ssp3_ra_r2_t[1:] + ssp3_eis_r2_t[1:] + ssp3_prtp_r2_t[1:] + ssp3_tech_r2_t[1:]) * ssp3_tot**(-1), alpha=alpha, \n",
    "                color=color_list[l4cind], label=\"Tech. Growth\")\n",
    "\n",
    "################################\n",
    "# SSP2\n",
    "################################\n",
    "ssp2_tot = ssp2_ra_r2_t[1:] + ssp2_eis_r2_t[1:] + ssp2_prtp_r2_t[1:] + ssp2_tech_r2_t[1:]\n",
    "ax[3].plot(node_times[1:], ssp2_ra_r2_t[1:] * ssp2_tot**(-1), linestyle='dashed', linewidth=lw, \n",
    "        color=color_list[l1cind])\n",
    "ax[3].fill_between(node_times[1:], np.zeros_like(node_times[1:]), ssp2_ra_r2_t[1:]* ssp2_tot**(-1), alpha=alpha, \n",
    "                color=color_list[l1cind], label=\"RA\")\n",
    "\n",
    "ax[3].plot(node_times[1:], (ssp2_eis_r2_t[1:] + ssp2_ra_r2_t[1:])* ssp2_tot**(-1), linestyle='dashed', linewidth=lw, \n",
    "        color=color_list[l2cind])\n",
    "ax[3].fill_between(node_times[1:], ssp2_ra_r2_t[1:] * ssp2_tot**(-1), \n",
    "                   (ssp2_ra_r2_t[1:] + ssp2_eis_r2_t[1:]) * ssp2_tot**(-1), alpha=alpha, \n",
    "                color=color_list[l2cind], label=\"EIS\")\n",
    "\n",
    "ax[3].plot(node_times[1:], (ssp2_eis_r2_t[1:] + ssp2_ra_r2_t[1:] + ssp2_prtp_r2_t[1:]) * ssp2_tot**(-1), linestyle='dashed', linewidth=lw, \n",
    "        color=color_list[l3cind])\n",
    "ax[3].fill_between(node_times[1:], (ssp2_eis_r2_t[1:] + ssp2_ra_r2_t[1:]) * ssp2_tot**(-1), \n",
    "                   (ssp2_ra_r2_t[1:] + ssp2_eis_r2_t[1:] + ssp2_prtp_r2_t[1:]) * ssp2_tot**(-1), alpha=alpha, \n",
    "                color=color_list[l3cind], label=\"PRTP\")\n",
    "\n",
    "ax[3].plot(node_times[1:], (ssp2_eis_r2_t[1:] + ssp2_ra_r2_t[1:] + ssp2_prtp_r2_t[1:] + ssp2_tech_r2_t[1:]) * ssp2_tot**(-1), linestyle='dashed', linewidth=lw, \n",
    "        color=color_list[l4cind])\n",
    "ax[3].fill_between(node_times[1:], (ssp2_eis_r2_t[1:] + ssp2_ra_r2_t[1:] + ssp2_prtp_r2_t[1:]) * ssp2_tot**(-1), \n",
    "                   (ssp2_ra_r2_t[1:] + ssp2_eis_r2_t[1:] + ssp2_prtp_r2_t[1:] + ssp2_tech_r2_t[1:]) * ssp2_tot**(-1), alpha=alpha, \n",
    "                color=color_list[l4cind], label=\"Tech. Growth\")\n",
    "\n",
    "\n",
    "################################\n",
    "# SSP1\n",
    "################################\n",
    "ssp1_tot = ssp1_ra_r2_t[1:] + ssp1_eis_r2_t[1:] + ssp1_prtp_r2_t[1:] + ssp1_tech_r2_t[1:]\n",
    "ax[4].plot(node_times[1:], ssp1_ra_r2_t[1:] * ssp1_tot**(-1), linestyle='dashed', linewidth=lw, \n",
    "        color=color_list[l1cind])\n",
    "ax[4].fill_between(node_times[1:], np.zeros_like(node_times[1:]), ssp5_ra_r2_t[1:] * ssp1_tot**(-1), alpha=alpha, \n",
    "                color=color_list[l1cind], label=\"RA\")\n",
    "\n",
    "ax[4].plot(node_times[1:], (ssp1_eis_r2_t[1:] + ssp1_ra_r2_t[1:]) * ssp1_tot**(-1), linestyle='dashed', linewidth=lw, \n",
    "        color=color_list[l2cind])\n",
    "ax[4].fill_between(node_times[1:], ssp1_ra_r2_t[1:] * ssp1_tot**(-1), \n",
    "                   (ssp1_ra_r2_t[1:] + ssp1_eis_r2_t[1:]) * ssp1_tot**(-1), alpha=alpha, \n",
    "                    color=color_list[l2cind], label=\"EIS\")\n",
    "\n",
    "ax[4].plot(node_times[1:], (ssp1_eis_r2_t[1:] + ssp1_ra_r2_t[1:] + ssp1_prtp_r2_t[1:]) * ssp1_tot**(-1), linestyle='dashed', linewidth=lw, \n",
    "        color=color_list[l3cind])\n",
    "ax[4].fill_between(node_times[1:], (ssp1_eis_r2_t[1:] + ssp1_ra_r2_t[1:]) * ssp1_tot**(-1), \n",
    "                   (ssp1_ra_r2_t[1:] + ssp1_eis_r2_t[1:] + ssp1_prtp_r2_t[1:]) * ssp1_tot**(-1), alpha=alpha, \n",
    "                    color=color_list[l3cind], label=\"PRTP\")\n",
    "\n",
    "ax[4].plot(node_times[1:], (ssp1_eis_r2_t[1:] + ssp1_ra_r2_t[1:] + ssp1_prtp_r2_t[1:] + ssp1_tech_r2_t[1:]) * ssp1_tot**(-1), linestyle='dashed', linewidth=lw, \n",
    "        color=color_list[l4cind])\n",
    "ax[4].fill_between(node_times[1:], (ssp1_eis_r2_t[1:] + ssp1_ra_r2_t[1:] + ssp1_prtp_r2_t[1:]) * ssp1_tot**(-1), \n",
    "                   (ssp1_ra_r2_t[1:] + ssp1_eis_r2_t[1:] + ssp1_prtp_r2_t[1:] + ssp1_tech_r2_t[1:]) * ssp1_tot**(-1), alpha=alpha, \n",
    "                color=color_list[l4cind], label=\"Tech. Growth\")\n",
    "\n",
    "ax[2].legend(loc='upper center', bbox_to_anchor=(0.5, -0.25), fancybox=True, shadow=True, ncol=5, fontsize=20)\n",
    "\n",
    "ax[0].set_ylabel(\"$r^{2} / r_{tot}^{2}$\")\n",
    "ax[0].set_title(\"SSP5\")\n",
    "ax[1].set_title(\"SSP4\")\n",
    "ax[2].set_title(\"SSP3\")\n",
    "ax[3].set_title(\"SSP2\")\n",
    "ax[4].set_title(\"SSP1\")\n",
    "\n",
    "panel_label = [\"A\", \"B\", \"C\", \"D\", \"E\"]\n",
    "for i in range(5):\n",
    "    ax[i].set_xlim((node_times[1], node_times[-1]))\n",
    "    ax[i].set_aspect(1.0/ax[i].get_data_ratio(), adjustable='box')\n",
    "    ax[i].set_xlabel(\"Year\")\n",
    "    ax[i].text(2168, 0.88, panel_label[i], fontsize=22, color='white', fontweight='bold')\n",
    "    ax[i].set_xticks([2030,2100,2200])\n",
    "    ax[i].set_xticklabels(['2030', '2100', '2200'])\n",
    "    ax[i].tick_params(axis='both', labelsize=14)\n",
    "\n",
    "sns.despine(trim=True, offset=10)\n",
    "\n",
    "if save_fig:\n",
    "    fig.savefig(''.join([basefile, 'rsquared-t.png']), dpi=400)"
   ]
  },
  {
   "cell_type": "markdown",
   "id": "f6bef07d",
   "metadata": {},
   "source": [
    "# Concentrations! "
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "69578eb0",
   "metadata": {},
   "outputs": [],
   "source": [
    "fig, ax = plt.subplots(4, 5, figsize=(30, 20))\n",
    "\n",
    "ax[0,0].set_title(\"2030\")\n",
    "ax[0,1].set_title(\"2060\")\n",
    "ax[0,2].set_title(\"2100\")\n",
    "ax[0,3].set_title(\"2150\")\n",
    "ax[0,4].set_title(\"2200\")\n",
    "\n",
    "###########################\n",
    "## RISK AVERSION ROW\n",
    "###########################\n",
    "make_panel_w_regression(ax, 0, 0, ssp5_ra, ssp4_ra, ssp3_ra, ssp2_ra, ssp1_ra, ssp5_co2030, ssp4_co2030, ssp3_co2030, ssp2_co2030, ssp1_co2030, \n",
    "           \"Risk Aversion\", \"CO$_2$ conc. (ppm)\")\n",
    "\n",
    "make_panel_w_regression(ax, 0, 1, ssp5_ra, ssp4_ra, ssp3_ra, ssp2_ra, ssp1_ra, ssp5_co2060, ssp4_co2060, ssp3_co2060, ssp2_co2060, ssp1_co2060, \n",
    "           \"Risk Aversion\", None)\n",
    "\n",
    "make_panel_w_regression(ax, 0, 2, ssp5_ra, ssp4_ra, ssp3_ra, ssp2_ra, ssp1_ra, ssp5_co2100, ssp4_co2100, ssp3_co2100, ssp2_co2100, ssp1_co2100, \n",
    "           \"Risk Aversion\", None)\n",
    "\n",
    "make_panel_w_regression(ax, 0, 3, ssp5_ra, ssp4_ra, ssp3_ra, ssp2_ra, ssp1_ra, ssp5_co2150, ssp4_co2150, ssp3_co2150, ssp2_co2150, ssp1_co2150, \n",
    "           \"Risk Aversion\", None)\n",
    "\n",
    "make_panel_w_regression(ax, 0, 4, ssp5_ra, ssp4_ra, ssp3_ra, ssp2_ra, ssp1_ra, ssp5_co2200, ssp4_co2200, ssp3_co2200, ssp2_co2200, ssp1_co2200, \n",
    "           \"Risk Aversion\", None)\n",
    "\n",
    "###########################\n",
    "# EIS ROW\n",
    "###########################\n",
    "make_panel_w_regression(ax, 1, 0, ssp5_eis, ssp4_eis, ssp3_eis, ssp2_eis, ssp1_eis, ssp5_co2030, ssp4_co2030, ssp3_co2030, ssp2_co2030, ssp1_co2030, \n",
    "           \"EIS\", \"CO$_2$ conc. (ppm)\")\n",
    "\n",
    "make_panel_w_regression(ax, 1, 1, ssp5_eis, ssp4_eis, ssp3_eis, ssp2_eis, ssp1_eis, ssp5_co2060, ssp4_co2060, ssp3_co2060, ssp2_co2060, ssp1_co2060, \n",
    "           \"EIS\", None)\n",
    "\n",
    "make_panel_w_regression(ax, 1, 2, ssp5_eis, ssp4_eis, ssp3_eis, ssp2_eis, ssp1_eis, ssp5_co2100, ssp4_co2100, ssp3_co2100, ssp2_co2100, ssp1_co2100, \n",
    "           \"EIS\", None)\n",
    "\n",
    "make_panel_w_regression(ax, 1, 3, ssp5_eis, ssp4_eis, ssp3_eis, ssp2_eis, ssp1_eis, ssp5_co2150, ssp4_co2150, ssp3_co2150, ssp2_co2150, ssp1_co2150, \n",
    "           \"EIS\", None)\n",
    "\n",
    "make_panel_w_regression(ax, 1, 4, ssp5_eis, ssp4_eis, ssp3_eis, ssp2_eis, ssp1_eis, ssp5_co2200, ssp4_co2200, ssp3_co2200, ssp2_co2200, ssp1_co2200, \n",
    "           \"EIS\", None)\n",
    "\n",
    "###########################\n",
    "# PRTP ROW\n",
    "###########################\n",
    "make_panel_w_regression(ax, 2, 0, ssp5_prtp * 100, ssp4_prtp * 100, ssp3_prtp * 100, ssp2_prtp * 100, ssp1_prtp * 100, ssp5_co2030, ssp4_co2030, ssp3_co2030, ssp2_co2030, ssp1_co2030, \n",
    "           r\"$\\delta$ (%)\", \"CO$_2$ conc. (ppm)\")\n",
    "\n",
    "make_panel_w_regression(ax, 2, 1, ssp5_prtp * 100, ssp4_prtp * 100, ssp3_prtp * 100, ssp2_prtp * 100, ssp1_prtp * 100, ssp5_co2060, ssp4_co2060, ssp3_co2060, ssp2_co2060, ssp1_co2060, \n",
    "           r\"$\\delta$ (%)\", None)\n",
    "\n",
    "make_panel_w_regression(ax, 2, 2, ssp5_prtp * 100, ssp4_prtp * 100, ssp3_prtp * 100, ssp2_prtp * 100, ssp1_prtp * 100, ssp5_co2100, ssp4_co2100, ssp3_co2100, ssp2_co2100, ssp1_co2100, \n",
    "           r\"$\\delta$ (%)\", None)\n",
    "\n",
    "make_panel_w_regression(ax, 2, 3, ssp5_prtp * 100, ssp4_prtp * 100, ssp3_prtp * 100, ssp2_prtp * 100, ssp1_prtp * 100, ssp5_co2150, ssp4_co2150, ssp3_co2150, ssp2_co2150, ssp1_co2150, \n",
    "           r\"$\\delta$ (%)\", None)\n",
    "\n",
    "make_panel_w_regression(ax, 2, 4, ssp5_prtp * 100, ssp4_prtp * 100, ssp3_prtp * 100, ssp2_prtp * 100, ssp1_prtp * 100, ssp5_co2200, ssp4_co2200, ssp3_co2200, ssp2_co2200, ssp1_co2200, \n",
    "           r\"$\\delta$ (%)\", None)\n",
    "\n",
    "###########################\n",
    "# EX TECH GROWTH ROW\n",
    "###########################\n",
    "make_panel_w_regression(ax, 3, 0, ssp5_tottech[:, 1], ssp4_tottech[:, 1], ssp3_tottech[:, 1], \n",
    "                        ssp2_tottech[:, 1], ssp1_tottech[:, 1], ssp5_co2030, ssp4_co2030, ssp3_co2030, \n",
    "                        ssp2_co2030, ssp1_co2030, \"Tech Growth (%)\", \"CO$_2$ conc. (ppm)\")\n",
    "\n",
    "make_panel_w_regression(ax, 3, 1, ssp5_tottech[:, 2], ssp4_tottech[:, 2], ssp3_tottech[:, 2], \n",
    "                        ssp2_tottech[:, 2], ssp1_tottech[:, 2], ssp5_co2060, ssp4_co2060, ssp3_co2060, \n",
    "                        ssp2_co2060, ssp1_co2060, \"Tech Growth (%)\", None)\n",
    "\n",
    "make_panel_w_regression(ax, 3, 2, ssp5_tottech[:, 3], ssp4_tottech[:, 3], ssp3_tottech[:, 3], \n",
    "                        ssp2_tottech[:, 3], ssp1_tottech[:, 3], ssp5_co2100, ssp4_co2100, ssp3_co2100, \n",
    "                        ssp2_co2100, ssp1_co2100, \"Tech Growth (%)\", None)\n",
    "\n",
    "make_panel_w_regression(ax, 3, 3, ssp5_tottech[:, 4], ssp4_tottech[:, 4], ssp3_tottech[:, 4], \n",
    "                        ssp2_tottech[:, 4], ssp1_tottech[:, 4], ssp5_co2150, ssp4_co2150, ssp3_co2150, \n",
    "                        ssp2_co2150, ssp1_co2150, \"Tech Growth (%)\", None)\n",
    "\n",
    "make_panel_w_regression(ax, 3, 4, ssp5_tottech[:, 5], ssp4_tottech[:, 5], ssp3_tottech[:, 5], \n",
    "                        ssp2_tottech[:, 5], ssp1_tottech[:, 5], ssp5_co2200, ssp4_co2200, ssp3_co2200, \n",
    "                        ssp2_co2200, ssp1_co2200, \"Tech Growth (%)\", None)\n",
    "\n",
    "ax[3,0].set_xticks([0,1,2,3])\n",
    "ax[3,0].set_xlim((0,3))\n",
    "ax[3,1].set_xticks([0,1,2,3,4,5])\n",
    "ax[3,2].set_xticks([0,1,2,3,4,5,6])\n",
    "ax[3,3].set_xticks([0,1,2,3,4,5,6])\n",
    "ax[3,4].set_xticks([0,1,2,3,4,5,6])\n",
    "\n",
    "fig.tight_layout()\n",
    "sns.despine(trim=True, offset=10)\n",
    "\n",
    "if save_fig:\n",
    "    fig.savefig(''.join([basefile, 'regression-sticky-co.png']), dpi=400)    "
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "e3099f44",
   "metadata": {},
   "outputs": [],
   "source": [
    "ssp5_eis_r2_co = get_r2_in_time(ssp5_eis, ssp5_exp_co2)\n",
    "ssp5_ra_r2_co = get_r2_in_time(ssp5_ra, ssp5_exp_co2)\n",
    "ssp5_prtp_r2_co = get_r2_in_time(ssp5_prtp, ssp5_exp_co2)\n",
    "ssp5_tech_r2_co = get_tech_r2_in_time_alllin(ssp5_tottech, ssp5_exp_co2)\n",
    "\n",
    "ssp4_eis_r2_co = get_r2_in_time(ssp4_eis, ssp4_exp_co2)\n",
    "ssp4_ra_r2_co = get_r2_in_time(ssp4_ra, ssp4_exp_co2)\n",
    "ssp4_prtp_r2_co = get_r2_in_time(ssp4_prtp, ssp4_exp_co2)\n",
    "ssp4_tech_r2_co = get_tech_r2_in_time_alllin(ssp4_tottech, ssp4_exp_co2)\n",
    "\n",
    "ssp3_eis_r2_co = get_r2_in_time(ssp3_eis, ssp3_exp_co2)\n",
    "ssp3_ra_r2_co = get_r2_in_time(ssp3_ra, ssp3_exp_co2)\n",
    "ssp3_prtp_r2_co = get_r2_in_time(ssp3_prtp, ssp3_exp_co2)\n",
    "ssp3_tech_r2_co = get_tech_r2_in_time_alllin(ssp3_tottech, ssp3_exp_co2)\n",
    "\n",
    "ssp2_eis_r2_co = get_r2_in_time(ssp2_eis, ssp2_exp_co2)\n",
    "ssp2_ra_r2_co = get_r2_in_time(ssp2_ra, ssp2_exp_co2)\n",
    "ssp2_prtp_r2_co = get_r2_in_time(ssp2_prtp, ssp2_exp_co2)\n",
    "ssp2_tech_r2_co = get_tech_r2_in_time_alllin(ssp2_tottech, ssp2_exp_co2)\n",
    "\n",
    "ssp1_eis_r2_co = get_r2_in_time(ssp1_eis, ssp1_exp_co2)\n",
    "ssp1_ra_r2_co = get_r2_in_time(ssp1_ra, ssp1_exp_co2)\n",
    "ssp1_prtp_r2_co = get_r2_in_time(ssp1_prtp, ssp1_exp_co2)\n",
    "ssp1_tech_r2_co = get_tech_r2_in_time_alllin(ssp1_tottech, ssp1_exp_co2)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "7ef45c0c",
   "metadata": {},
   "outputs": [],
   "source": [
    "fig, ax = plt.subplots(1,5, figsize=(20,20), sharey=True)\n",
    "\n",
    "ax[0].set_ylim((0,1))\n",
    "\n",
    "alpha = 0.8\n",
    "lw = 3\n",
    "l1cind = 0\n",
    "l2cind = 1\n",
    "l3cind = 2\n",
    "l4cind = 3\n",
    "l5cind = 4\n",
    "\n",
    "################################\n",
    "# SSP5\n",
    "################################\n",
    "ax[0].plot(node_times[1:], ssp5_ra_r2_co[1:], linestyle='dashed', linewidth=lw, \n",
    "        color=color_list[l1cind])\n",
    "ax[0].fill_between(node_times[1:], np.zeros_like(node_times[1:]), ssp5_ra_r2_co[1:], alpha=alpha, \n",
    "                color=color_list[l1cind], label=\"RA\")\n",
    "\n",
    "ax[0].plot(node_times[1:], ssp5_eis_r2_co[1:] + ssp5_ra_r2_co[1:], linestyle='dashed', linewidth=lw, \n",
    "        color=color_list[l2cind])\n",
    "ax[0].fill_between(node_times[1:], ssp5_ra_r2_co[1:], ssp5_ra_r2_co[1:] + ssp5_eis_r2_co[1:], alpha=alpha, \n",
    "                color=color_list[l2cind], label=\"EIS\")\n",
    "\n",
    "ax[0].plot(node_times[1:], ssp5_eis_r2_co[1:] + ssp5_ra_r2_co[1:] + ssp5_prtp_r2_co[1:], linestyle='dashed', linewidth=lw, \n",
    "        color=color_list[l3cind])\n",
    "ax[0].fill_between(node_times[1:], ssp5_eis_r2_co[1:] + ssp5_ra_r2_co[1:], ssp5_ra_r2_co[1:] + ssp5_eis_r2_co[1:] + ssp5_prtp_r2_co[1:], alpha=alpha, \n",
    "                color=color_list[l3cind], label=\"PRTP\")\n",
    "\n",
    "ax[0].plot(node_times[1:], ssp5_eis_r2_co[1:] + ssp5_ra_r2_co[1:] + ssp5_prtp_r2_co[1:] + ssp5_tech_r2_co[1:], linestyle='dashed', linewidth=lw, \n",
    "        color=color_list[l4cind])\n",
    "ax[0].fill_between(node_times[1:], ssp5_eis_r2_co[1:] + ssp5_ra_r2_co[1:] + ssp5_prtp_r2_co[1:], ssp5_ra_r2_co[1:] + ssp5_eis_r2_co[1:] + ssp5_prtp_r2_co[1:] + ssp5_tech_r2_co[1:], alpha=alpha, \n",
    "                color=color_list[l4cind], label=\"Tech. Growth\")\n",
    "\n",
    "\n",
    "################################\n",
    "# SSP4\n",
    "################################\n",
    "ax[1].plot(node_times[1:], ssp4_ra_r2_co[1:], linestyle='dashed', linewidth=lw, \n",
    "        color=color_list[l1cind])\n",
    "ax[1].fill_between(node_times[1:], np.zeros_like(node_times[1:]), ssp4_ra_r2_co[1:], alpha=alpha, \n",
    "                color=color_list[l1cind], label=\"RA\")\n",
    "\n",
    "ax[1].plot(node_times[1:], ssp4_eis_r2_co[1:] + ssp4_ra_r2_co[1:], linestyle='dashed', linewidth=lw, \n",
    "        color=color_list[l2cind])\n",
    "ax[1].fill_between(node_times[1:], ssp4_ra_r2_co[1:], ssp5_ra_r2_co[1:] + ssp4_eis_r2_co[1:], alpha=alpha, \n",
    "                color=color_list[l2cind], label=\"EIS\")\n",
    "\n",
    "ax[1].plot(node_times[1:], ssp4_eis_r2_co[1:] + ssp4_ra_r2_co[1:] + ssp4_prtp_r2_co[1:], linestyle='dashed', linewidth=lw, \n",
    "        color=color_list[l3cind])\n",
    "ax[1].fill_between(node_times[1:], ssp4_eis_r2_co[1:] + ssp4_ra_r2_co[1:], ssp4_ra_r2_co[1:] + ssp4_eis_r2_co[1:] + ssp4_prtp_r2_co[1:], alpha=alpha, \n",
    "                color=color_list[l3cind], label=\"PRTP\")\n",
    "\n",
    "ax[1].plot(node_times[1:], ssp4_eis_r2_co[1:] + ssp4_ra_r2_co[1:] + ssp4_prtp_r2_co[1:] + ssp4_tech_r2_co[1:], linestyle='dashed', linewidth=lw, \n",
    "        color=color_list[l4cind])\n",
    "ax[1].fill_between(node_times[1:], ssp4_eis_r2_co[1:] + ssp4_ra_r2_co[1:] + ssp4_prtp_r2_co[1:], ssp4_ra_r2_co[1:] + ssp4_eis_r2_co[1:] + ssp4_prtp_r2_co[1:] + ssp4_tech_r2_co[1:], alpha=alpha, \n",
    "                color=color_list[l4cind], label=\"Tech. Growth\")\n",
    "\n",
    "################################\n",
    "# SSP3\n",
    "################################\n",
    "ax[2].plot(node_times[1:], ssp3_ra_r2_co[1:], linestyle='dashed', linewidth=lw, \n",
    "        color=color_list[l1cind])\n",
    "ax[2].fill_between(node_times[1:], np.zeros_like(node_times[1:]), ssp3_ra_r2_co[1:], alpha=alpha, \n",
    "                color=color_list[l1cind], label=\"RA\")\n",
    "\n",
    "ax[2].plot(node_times[1:], ssp3_eis_r2_co[1:] + ssp3_ra_r2_co[1:], linestyle='dashed', linewidth=lw, \n",
    "        color=color_list[l2cind])\n",
    "ax[2].fill_between(node_times[1:], ssp3_ra_r2_co[1:], ssp3_ra_r2_co[1:] + ssp3_eis_r2_co[1:], alpha=alpha, \n",
    "                color=color_list[l2cind], label=\"EIS\")\n",
    "\n",
    "ax[2].plot(node_times[1:], ssp3_eis_r2_co[1:] + ssp3_ra_r2_co[1:] + ssp3_prtp_r2_co[1:], linestyle='dashed', linewidth=lw, \n",
    "        color=color_list[l3cind])\n",
    "ax[2].fill_between(node_times[1:], ssp3_eis_r2_co[1:] + ssp3_ra_r2_co[1:], ssp3_ra_r2_co[1:] + ssp3_eis_r2_co[1:] + ssp3_prtp_r2_co[1:], alpha=alpha, \n",
    "                color=color_list[l3cind], label=\"PRTP\")\n",
    "\n",
    "ax[2].plot(node_times[1:], ssp3_eis_r2_co[1:] + ssp3_ra_r2_co[1:] + ssp3_prtp_r2_co[1:] + ssp3_tech_r2_co[1:], linestyle='dashed', linewidth=lw, \n",
    "        color=color_list[l4cind])\n",
    "ax[2].fill_between(node_times[1:], ssp3_eis_r2_co[1:] + ssp3_ra_r2_co[1:] + ssp3_prtp_r2_co[1:], ssp3_ra_r2_co[1:] + ssp3_eis_r2_co[1:] + ssp3_prtp_r2_co[1:] + ssp3_tech_r2_co[1:], alpha=alpha, \n",
    "                color=color_list[l4cind], label=\"Tech. Growth\")\n",
    "\n",
    "################################\n",
    "# SSP2\n",
    "################################\n",
    "ax[3].plot(node_times[1:], ssp2_ra_r2_co[1:], linestyle='dashed', linewidth=lw, \n",
    "        color=color_list[l1cind])\n",
    "ax[3].fill_between(node_times[1:], np.zeros_like(node_times[1:]), ssp2_ra_r2_co[1:], alpha=alpha, \n",
    "                color=color_list[l1cind], label=\"RA\")\n",
    "\n",
    "ax[3].plot(node_times[1:], ssp2_eis_r2_co[1:] + ssp2_ra_r2_co[1:], linestyle='dashed', linewidth=lw, \n",
    "        color=color_list[l2cind])\n",
    "ax[3].fill_between(node_times[1:], ssp2_ra_r2_co[1:], ssp2_ra_r2_co[1:] + ssp2_eis_r2_co[1:], alpha=alpha, \n",
    "                color=color_list[l2cind], label=\"EIS\")\n",
    "\n",
    "ax[3].plot(node_times[1:], ssp2_eis_r2_co[1:] + ssp2_ra_r2_co[1:] + ssp2_prtp_r2_co[1:], linestyle='dashed', linewidth=lw, \n",
    "        color=color_list[l3cind])\n",
    "ax[3].fill_between(node_times[1:], ssp2_eis_r2_co[1:] + ssp2_ra_r2_co[1:], ssp2_ra_r2_co[1:] + ssp2_eis_r2_co[1:] + ssp2_prtp_r2_co[1:], alpha=alpha, \n",
    "                color=color_list[l3cind], label=\"PRTP\")\n",
    "\n",
    "ax[3].plot(node_times[1:], ssp2_eis_r2_co[1:] + ssp2_ra_r2_co[1:] + ssp2_prtp_r2_co[1:] + ssp2_tech_r2_co[1:], linestyle='dashed', linewidth=lw, \n",
    "        color=color_list[l4cind])\n",
    "ax[3].fill_between(node_times[1:], ssp2_eis_r2_co[1:] + ssp2_ra_r2_co[1:] + ssp2_prtp_r2_co[1:], ssp2_ra_r2_co[1:] + ssp2_eis_r2_co[1:] + ssp2_prtp_r2_co[1:] + ssp2_tech_r2_co[1:], alpha=alpha, \n",
    "                color=color_list[l4cind], label=\"Tech. Growth\")\n",
    "\n",
    "\n",
    "################################\n",
    "# SSP1\n",
    "################################\n",
    "ax[4].plot(node_times[1:], ssp1_ra_r2_co[1:], linestyle='dashed', linewidth=lw, \n",
    "        color=color_list[l1cind])\n",
    "ax[4].fill_between(node_times[1:], np.zeros_like(node_times[1:]), ssp5_ra_r2_co[1:], alpha=alpha, \n",
    "                color=color_list[l1cind], label=\"RA\")\n",
    "\n",
    "ax[4].plot(node_times[1:], ssp1_eis_r2_co[1:] + ssp1_ra_r2_co[1:], linestyle='dashed', linewidth=lw, \n",
    "        color=color_list[l2cind])\n",
    "ax[4].fill_between(node_times[1:], ssp1_ra_r2_co[1:], ssp1_ra_r2_co[1:] + ssp1_eis_r2_co[1:], alpha=alpha, \n",
    "                color=color_list[l2cind], label=\"EIS\")\n",
    "\n",
    "ax[4].plot(node_times[1:], ssp1_eis_r2_co[1:] + ssp1_ra_r2_co[1:] + ssp1_prtp_r2_co[1:], linestyle='dashed', linewidth=lw, \n",
    "        color=color_list[l3cind])\n",
    "ax[4].fill_between(node_times[1:], ssp1_eis_r2_co[1:] + ssp1_ra_r2_co[1:], ssp1_ra_r2_co[1:] + ssp1_eis_r2_co[1:] + ssp1_prtp_r2_co[1:], alpha=alpha, \n",
    "                color=color_list[l3cind], label=\"PRTP\")\n",
    "\n",
    "ax[4].plot(node_times[1:], ssp1_eis_r2_co[1:] + ssp1_ra_r2_co[1:] + ssp1_prtp_r2_co[1:] + ssp1_tech_r2_co[1:], linestyle='dashed', linewidth=lw, \n",
    "        color=color_list[l4cind])\n",
    "ax[4].fill_between(node_times[1:], ssp1_eis_r2_co[1:] + ssp1_ra_r2_co[1:] + ssp1_prtp_r2_co[1:], ssp1_ra_r2_co[1:] + ssp1_eis_r2_co[1:] + ssp1_prtp_r2_co[1:] + ssp1_tech_r2_co[1:], alpha=alpha, \n",
    "                color=color_list[l4cind], label=\"Tech. Growth\")\n",
    "\n",
    "ax[2].legend(loc='upper center', bbox_to_anchor=(0.5, -0.25), fancybox=True, shadow=True, ncol=5)\n",
    "\n",
    "ax[0].set_ylabel(\"r$^{2}$\")\n",
    "ax[0].set_title(\"SSP5\")\n",
    "ax[1].set_title(\"SSP4\")\n",
    "ax[2].set_title(\"SSP3\")\n",
    "ax[3].set_title(\"SSP2\")\n",
    "ax[4].set_title(\"SSP1\")\n",
    "\n",
    "for i in range(5):\n",
    "    ax[i].set_xlim((node_times[1], node_times[-1]))\n",
    "    ax[i].set_aspect(1.0/ax[i].get_data_ratio(), adjustable='box')\n",
    "    ax[i].set_xlabel(\"Year\")\n",
    "\n",
    "#fig.tight_layout()"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "bd642add",
   "metadata": {},
   "outputs": [],
   "source": [
    "fig, ax = plt.subplots(1,5, figsize=(20,20), sharey=True)\n",
    "\n",
    "ax[0].set_ylim((0,1))\n",
    "\n",
    "alpha = 0.8\n",
    "lw = 3\n",
    "l1cind = 0\n",
    "l2cind = 1\n",
    "l3cind = 2\n",
    "l4cind = 3\n",
    "l5cind = 4\n",
    "\n",
    "################################\n",
    "# SSP5\n",
    "################################\n",
    "tot_ssp5 = ssp5_ra_r2_co[1:] + ssp5_eis_r2_co[1:] + ssp5_prtp_r2_co[1:] + ssp5_tech_r2_co[1:]\n",
    "ax[0].plot(node_times[1:], ssp5_ra_r2_co[1:] * tot_ssp5**(-1), linestyle='dashed', linewidth=lw, \n",
    "        color=color_list[l1cind])\n",
    "ax[0].fill_between(node_times[1:], np.zeros_like(node_times[1:]), ssp5_ra_r2_co[1:] * tot_ssp5**(-1), alpha=alpha, \n",
    "                color=color_list[l1cind], label=\"RA\")\n",
    "\n",
    "ax[0].plot(node_times[1:], (ssp5_eis_r2_co[1:] + ssp5_ra_r2_co[1:])  * tot_ssp5**(-1), linestyle='dashed', linewidth=lw, \n",
    "        color=color_list[l2cind])\n",
    "ax[0].fill_between(node_times[1:], ssp5_ra_r2_co[1:]  * tot_ssp5**(-1), \n",
    "                   (ssp5_ra_r2_co[1:] + ssp5_eis_r2_co[1:]) * tot_ssp5**(-1), alpha=alpha, \n",
    "                color=color_list[l2cind], label=\"EIS\")\n",
    "\n",
    "ax[0].plot(node_times[1:], (ssp5_eis_r2_co[1:] + ssp5_ra_r2_co[1:] + ssp5_prtp_r2_co[1:])  * tot_ssp5**(-1), linestyle='dashed', linewidth=lw, \n",
    "        color=color_list[l3cind])\n",
    "ax[0].fill_between(node_times[1:], (ssp5_eis_r2_co[1:] + ssp5_ra_r2_co[1:]) * tot_ssp5**(-1), \n",
    "                  (ssp5_ra_r2_co[1:] + ssp5_eis_r2_co[1:] + ssp5_prtp_r2_co[1:]) * tot_ssp5**(-1), alpha=alpha, \n",
    "                color=color_list[l3cind], label=\"PRTP\")\n",
    "\n",
    "ax[0].plot(node_times[1:], (ssp5_eis_r2_co[1:] + ssp5_ra_r2_co[1:] + ssp5_prtp_r2_co[1:] + ssp5_tech_r2_co[1:]) * tot_ssp5**(-1), \n",
    "           linestyle='dashed', linewidth=lw, color=color_list[l4cind])\n",
    "ax[0].fill_between(node_times[1:], (ssp5_eis_r2_co[1:] + ssp5_ra_r2_co[1:] + ssp5_prtp_r2_co[1:]) * tot_ssp5**(-1), \n",
    "                   (ssp5_ra_r2_co[1:] + ssp5_eis_r2_co[1:] + ssp5_prtp_r2_co[1:] + ssp5_tech_r2_co[1:]) * tot_ssp5**(-1), alpha=alpha, \n",
    "                color=color_list[l4cind], label=\"Tech. Growth\")\n",
    "\n",
    "\n",
    "################################\n",
    "# SSP4\n",
    "################################\n",
    "tot_ssp4 = ssp4_ra_r2_co[1:] + ssp4_eis_r2_co[1:] + ssp4_prtp_r2_co[1:] + ssp4_tech_r2_co[1:]\n",
    "ax[1].plot(node_times[1:], ssp4_ra_r2_co[1:] * tot_ssp4**(-1), linestyle='dashed', linewidth=lw, \n",
    "        color=color_list[l1cind])\n",
    "ax[1].fill_between(node_times[1:], np.zeros_like(node_times[1:]), ssp4_ra_r2_co[1:] * tot_ssp4**(-1), alpha=alpha, \n",
    "                color=color_list[l1cind], label=\"RA\")\n",
    "\n",
    "ax[1].plot(node_times[1:], (ssp4_eis_r2_co[1:] + ssp4_ra_r2_co[1:]) * tot_ssp4**(-1), linestyle='dashed', linewidth=lw, \n",
    "        color=color_list[l2cind])\n",
    "ax[1].fill_between(node_times[1:], ssp4_ra_r2_co[1:] * tot_ssp4**(-1), \n",
    "                   (ssp4_ra_r2_co[1:] + ssp4_eis_r2_co[1:]) * tot_ssp4**(-1), alpha=alpha, \n",
    "                color=color_list[l2cind], label=\"EIS\")\n",
    "\n",
    "ax[1].plot(node_times[1:], (ssp4_eis_r2_co[1:] + ssp4_ra_r2_co[1:] + ssp4_prtp_r2_co[1:]) * tot_ssp4**(-1), linestyle='dashed', linewidth=lw, \n",
    "        color=color_list[l3cind])\n",
    "ax[1].fill_between(node_times[1:], (ssp4_eis_r2_co[1:] + ssp4_ra_r2_co[1:]) * tot_ssp4**(-1), \n",
    "                  (ssp4_ra_r2_co[1:] + ssp4_eis_r2_co[1:] + ssp4_prtp_r2_co[1:]) * tot_ssp4**(-1), alpha=alpha, \n",
    "                color=color_list[l3cind], label=\"PRTP\")\n",
    "\n",
    "ax[1].plot(node_times[1:], (ssp4_eis_r2_co[1:] + ssp4_ra_r2_co[1:] + ssp4_prtp_r2_co[1:] + ssp4_tech_r2_co[1:]) * tot_ssp4**(-1), \n",
    "           linestyle='dashed', linewidth=lw, color=color_list[l4cind])\n",
    "ax[1].fill_between(node_times[1:], (ssp4_eis_r2_co[1:] + ssp4_ra_r2_co[1:] + ssp4_prtp_r2_co[1:]) * tot_ssp4**(-1), \n",
    "                   (ssp4_ra_r2_co[1:] + ssp4_eis_r2_co[1:] + ssp4_prtp_r2_co[1:] + ssp4_tech_r2_co[1:]) * tot_ssp4**(-1), alpha=alpha, \n",
    "                color=color_list[l4cind], label=\"Tech. Growth\")\n",
    "\n",
    "################################\n",
    "# SSP3\n",
    "################################\n",
    "ssp3_tot = ssp3_ra_r2_co[1:] + ssp3_eis_r2_co[1:] + ssp3_prtp_r2_co[1:] + ssp3_tech_r2_co[1:]\n",
    "ax[2].plot(node_times[1:], ssp3_ra_r2_co[1:] * ssp3_tot**(-1), linestyle='dashed', linewidth=lw, \n",
    "        color=color_list[l1cind])\n",
    "ax[2].fill_between(node_times[1:], np.zeros_like(node_times[1:]), ssp3_ra_r2_co[1:] * ssp3_tot**(-1), alpha=alpha, \n",
    "                color=color_list[l1cind], label=\"RA\")\n",
    "\n",
    "ax[2].plot(node_times[1:], (ssp3_eis_r2_co[1:] + ssp3_ra_r2_co[1:]) * ssp3_tot**(-1), linestyle='dashed', linewidth=lw, \n",
    "        color=color_list[l2cind])\n",
    "ax[2].fill_between(node_times[1:], ssp3_ra_r2_co[1:] * ssp3_tot**(-1), \n",
    "                   (ssp3_ra_r2_co[1:] + ssp3_eis_r2_co[1:]) * ssp3_tot**(-1), alpha=alpha, \n",
    "                color=color_list[l2cind], label=\"EIS\")\n",
    "\n",
    "ax[2].plot(node_times[1:], (ssp3_eis_r2_co[1:] + ssp3_ra_r2_co[1:] + ssp3_prtp_r2_co[1:]) * ssp3_tot**(-1), linestyle='dashed', linewidth=lw, \n",
    "        color=color_list[l3cind])\n",
    "ax[2].fill_between(node_times[1:], (ssp3_eis_r2_co[1:] + ssp3_ra_r2_co[1:]) * ssp3_tot**(-1), \n",
    "                   (ssp3_ra_r2_co[1:] + ssp3_eis_r2_co[1:] + ssp3_prtp_r2_co[1:]) * ssp3_tot**(-1), alpha=alpha, \n",
    "                color=color_list[l3cind], label=\"PRTP\")\n",
    "\n",
    "ax[2].plot(node_times[1:], (ssp3_eis_r2_co[1:] + ssp3_ra_r2_co[1:] + ssp3_prtp_r2_co[1:] + ssp3_tech_r2_co[1:]) * ssp3_tot**(-1), linestyle='dashed', linewidth=lw, \n",
    "        color=color_list[l4cind])\n",
    "ax[2].fill_between(node_times[1:], (ssp3_eis_r2_co[1:] + ssp3_ra_r2_co[1:] + ssp3_prtp_r2_co[1:]) * ssp3_tot**(-1), \n",
    "                   (ssp3_ra_r2_co[1:] + ssp3_eis_r2_co[1:] + ssp3_prtp_r2_co[1:] + ssp3_tech_r2_co[1:]) * ssp3_tot**(-1), alpha=alpha, \n",
    "                color=color_list[l4cind], label=\"Tech. Growth\")\n",
    "\n",
    "################################\n",
    "# SSP2\n",
    "################################\n",
    "ssp2_tot = ssp2_ra_r2_co[1:] + ssp2_eis_r2_co[1:] + ssp2_prtp_r2_co[1:] + ssp2_tech_r2_co[1:]\n",
    "ax[3].plot(node_times[1:], ssp2_ra_r2_co[1:] * ssp2_tot**(-1), linestyle='dashed', linewidth=lw, \n",
    "        color=color_list[l1cind])\n",
    "ax[3].fill_between(node_times[1:], np.zeros_like(node_times[1:]), ssp2_ra_r2_co[1:]* ssp2_tot**(-1), alpha=alpha, \n",
    "                color=color_list[l1cind], label=\"RA\")\n",
    "\n",
    "ax[3].plot(node_times[1:], (ssp2_eis_r2_co[1:] + ssp2_ra_r2_co[1:])* ssp2_tot**(-1), linestyle='dashed', linewidth=lw, \n",
    "        color=color_list[l2cind])\n",
    "ax[3].fill_between(node_times[1:], ssp2_ra_r2_co[1:] * ssp2_tot**(-1), \n",
    "                   (ssp2_ra_r2_co[1:] + ssp2_eis_r2_co[1:]) * ssp2_tot**(-1), alpha=alpha, \n",
    "                color=color_list[l2cind], label=\"EIS\")\n",
    "\n",
    "ax[3].plot(node_times[1:], (ssp2_eis_r2_co[1:] + ssp2_ra_r2_co[1:] + ssp2_prtp_r2_co[1:]) * ssp2_tot**(-1), linestyle='dashed', linewidth=lw, \n",
    "        color=color_list[l3cind])\n",
    "ax[3].fill_between(node_times[1:], (ssp2_eis_r2_co[1:] + ssp2_ra_r2_co[1:]) * ssp2_tot**(-1), \n",
    "                   (ssp2_ra_r2_co[1:] + ssp2_eis_r2_co[1:] + ssp2_prtp_r2_co[1:]) * ssp2_tot**(-1), alpha=alpha, \n",
    "                color=color_list[l3cind], label=\"PRTP\")\n",
    "\n",
    "ax[3].plot(node_times[1:], (ssp2_eis_r2_co[1:] + ssp2_ra_r2_co[1:] + ssp2_prtp_r2_co[1:] + ssp2_tech_r2_co[1:]) * ssp2_tot**(-1), linestyle='dashed', linewidth=lw, \n",
    "        color=color_list[l4cind])\n",
    "ax[3].fill_between(node_times[1:], (ssp2_eis_r2_co[1:] + ssp2_ra_r2_co[1:] + ssp2_prtp_r2_co[1:]) * ssp2_tot**(-1), \n",
    "                   (ssp2_ra_r2_co[1:] + ssp2_eis_r2_co[1:] + ssp2_prtp_r2_co[1:] + ssp2_tech_r2_co[1:]) * ssp2_tot**(-1), alpha=alpha, \n",
    "                color=color_list[l4cind], label=\"Tech. Growth\")\n",
    "\n",
    "\n",
    "################################\n",
    "# SSP1\n",
    "################################\n",
    "ssp1_tot = ssp1_ra_r2_co[1:] + ssp1_eis_r2_co[1:] + ssp1_prtp_r2_co[1:] + ssp1_tech_r2_co[1:]\n",
    "ax[4].plot(node_times[1:], ssp1_ra_r2_co[1:] * ssp1_tot**(-1), linestyle='dashed', linewidth=lw, \n",
    "        color=color_list[l1cind])\n",
    "ax[4].fill_between(node_times[1:], np.zeros_like(node_times[1:]), ssp5_ra_r2_co[1:] * ssp1_tot**(-1), alpha=alpha, \n",
    "                color=color_list[l1cind], label=\"RA\")\n",
    "\n",
    "ax[4].plot(node_times[1:], (ssp1_eis_r2_co[1:] + ssp1_ra_r2_co[1:]) * ssp1_tot**(-1), linestyle='dashed', linewidth=lw, \n",
    "        color=color_list[l2cind])\n",
    "ax[4].fill_between(node_times[1:], ssp1_ra_r2_co[1:] * ssp1_tot**(-1), \n",
    "                   (ssp1_ra_r2_co[1:] + ssp1_eis_r2_co[1:]) * ssp1_tot**(-1), alpha=alpha, \n",
    "                    color=color_list[l2cind], label=\"EIS\")\n",
    "\n",
    "ax[4].plot(node_times[1:], (ssp1_eis_r2_co[1:] + ssp1_ra_r2_co[1:] + ssp1_prtp_r2_co[1:]) * ssp1_tot**(-1), linestyle='dashed', linewidth=lw, \n",
    "        color=color_list[l3cind])\n",
    "ax[4].fill_between(node_times[1:], (ssp1_eis_r2_co[1:] + ssp1_ra_r2_co[1:]) * ssp1_tot**(-1), \n",
    "                   (ssp1_ra_r2_co[1:] + ssp1_eis_r2_co[1:] + ssp1_prtp_r2_co[1:]) * ssp1_tot**(-1), alpha=alpha, \n",
    "                    color=color_list[l3cind], label=\"PRTP\")\n",
    "\n",
    "ax[4].plot(node_times[1:], (ssp1_eis_r2_co[1:] + ssp1_ra_r2_co[1:] + ssp1_prtp_r2_co[1:] + ssp1_tech_r2_co[1:]) * ssp1_tot**(-1), linestyle='dashed', linewidth=lw, \n",
    "        color=color_list[l4cind])\n",
    "ax[4].fill_between(node_times[1:], (ssp1_eis_r2_co[1:] + ssp1_ra_r2_co[1:] + ssp1_prtp_r2_co[1:]) * ssp1_tot**(-1), \n",
    "                   (ssp1_ra_r2_co[1:] + ssp1_eis_r2_co[1:] + ssp1_prtp_r2_co[1:] + ssp1_tech_r2_co[1:]) * ssp1_tot**(-1), alpha=alpha, \n",
    "                color=color_list[l4cind], label=\"Tech. Growth\")\n",
    "\n",
    "ax[2].legend(loc='upper center', bbox_to_anchor=(0.5, -0.25), fancybox=True, shadow=True, ncol=5, fontsize=20)\n",
    "\n",
    "ax[0].set_ylabel(\"$r^{2} / r_{tot}^{2}$\")\n",
    "ax[0].set_title(\"SSP5\")\n",
    "ax[1].set_title(\"SSP4\")\n",
    "ax[2].set_title(\"SSP3\")\n",
    "ax[3].set_title(\"SSP2\")\n",
    "ax[4].set_title(\"SSP1\")\n",
    "\n",
    "panel_label = [\"A\", \"B\", \"C\", \"D\", \"E\"]\n",
    "for i in range(5):\n",
    "    ax[i].set_xlim((node_times[1], node_times[-1]))\n",
    "    ax[i].set_aspect(1.0/ax[i].get_data_ratio(), adjustable='box')\n",
    "    ax[i].set_xlabel(\"Year\")\n",
    "    ax[i].text(2168, 0.88, panel_label[i], fontsize=22, color='white', fontweight='bold')\n",
    "    ax[i].set_xticks([2030,2100,2200])\n",
    "    ax[i].set_xticklabels(['2030', '2100', '2200'])\n",
    "    ax[i].tick_params(axis='both', labelsize=14)\n",
    "\n",
    "sns.despine(trim=True, offset=10)\n",
    "\n",
    "if save_fig:\n",
    "    fig.savefig(''.join([basefile, 'rsquared-conc.png']), dpi=400)"
   ]
  },
  {
   "cell_type": "markdown",
   "id": "372b3cef",
   "metadata": {},
   "source": [
    "# Damages!!"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "2163039f",
   "metadata": {},
   "outputs": [],
   "source": [
    "fig, ax = plt.subplots(4, 5, figsize=(30, 20))\n",
    "\n",
    "ax[0,0].set_title(\"2030\")\n",
    "ax[0,1].set_title(\"2060\")\n",
    "ax[0,2].set_title(\"2100\")\n",
    "ax[0,3].set_title(\"2150\")\n",
    "ax[0,4].set_title(\"2200\")\n",
    "\n",
    "###########################\n",
    "## RISK AVERSION ROW\n",
    "###########################\n",
    "make_panel_w_regression(ax, 0, 0, ssp5_ra, ssp4_ra, ssp3_ra, ssp2_ra, ssp1_ra, ssp5_d2030, ssp4_d2030, ssp3_d2030, ssp2_d2030, ssp1_d2030, \n",
    "           \"Risk Aversion\", \"Damage (frac GDP)\")\n",
    "\n",
    "make_panel_w_regression(ax, 0, 1, ssp5_ra, ssp4_ra, ssp3_ra, ssp2_ra, ssp1_ra, ssp5_d2060, ssp4_d2060, ssp3_d2060, ssp2_d2060, ssp1_d2060, \n",
    "           \"Risk Aversion\", None)\n",
    "\n",
    "make_panel_w_regression(ax, 0, 2, ssp5_ra, ssp4_ra, ssp3_ra, ssp2_ra, ssp1_ra, ssp5_d2100, ssp4_d2100, ssp3_d2100, ssp2_d2100, ssp1_d2100, \n",
    "           \"Risk Aversion\", None)\n",
    "\n",
    "make_panel_w_regression(ax, 0, 3, ssp5_ra, ssp4_ra, ssp3_ra, ssp2_ra, ssp1_ra, ssp5_d2150, ssp4_d2150, ssp3_d2150, ssp2_d2150, ssp1_d2150, \n",
    "           \"Risk Aversion\", None)\n",
    "\n",
    "make_panel_w_regression(ax, 0, 4, ssp5_ra, ssp4_ra, ssp3_ra, ssp2_ra, ssp1_ra, ssp5_d2200, ssp4_d2200, ssp3_d2200, ssp2_d2200, ssp1_d2200, \n",
    "           \"Risk Aversion\", None)\n",
    "\n",
    "###########################\n",
    "# EIS ROW\n",
    "###########################\n",
    "make_panel_w_regression(ax, 1, 0, ssp5_eis, ssp4_eis, ssp3_eis, ssp2_eis, ssp1_eis, ssp5_d2030, ssp4_d2030, ssp3_d2030, ssp2_d2030, ssp1_d2030, \n",
    "           \"EIS\", \"Damage (frac GDP)\")\n",
    "\n",
    "make_panel_w_regression(ax, 1, 1, ssp5_eis, ssp4_eis, ssp3_eis, ssp2_eis, ssp1_eis, ssp5_d2060, ssp4_d2060, ssp3_d2060, ssp2_d2060, ssp1_d2060, \n",
    "           \"EIS\", None)\n",
    "\n",
    "make_panel_w_regression(ax, 1, 2, ssp5_eis, ssp4_eis, ssp3_eis, ssp2_eis, ssp1_eis, ssp5_d2100, ssp4_d2100, ssp3_d2100, ssp2_d2100, ssp1_d2100, \n",
    "           \"EIS\", None)\n",
    "\n",
    "make_panel_w_regression(ax, 1, 3, ssp5_eis, ssp4_eis, ssp3_eis, ssp2_eis, ssp1_eis, ssp5_d2150, ssp4_d2150, ssp3_d2150, ssp2_d2150, ssp1_d2150, \n",
    "           \"EIS\", None)\n",
    "\n",
    "make_panel_w_regression(ax, 1, 4, ssp5_eis, ssp4_eis, ssp3_eis, ssp2_eis, ssp1_eis, ssp5_d2200, ssp4_d2200, ssp3_d2200, ssp2_d2200, ssp1_d2200, \n",
    "           \"EIS\", None)\n",
    "\n",
    "###########################\n",
    "# PRTP ROW\n",
    "###########################\n",
    "make_panel_w_regression(ax, 2, 0, ssp5_prtp * 100, ssp4_prtp * 100, ssp3_prtp * 100, ssp2_prtp * 100, ssp1_prtp * 100, ssp5_d2030, ssp4_d2030, ssp3_d2030, ssp2_d2030, ssp1_d2030, \n",
    "           r\"$\\delta$ (%)\", \"Damage (frac GDP)\")\n",
    "\n",
    "make_panel_w_regression(ax, 2, 1, ssp5_prtp * 100, ssp4_prtp * 100, ssp3_prtp * 100, ssp2_prtp * 100, ssp1_prtp * 100, ssp5_d2060, ssp4_d2060, ssp3_d2060, ssp2_d2060, ssp1_d2060, \n",
    "           r\"$\\delta$ (%)\", None)\n",
    "\n",
    "make_panel_w_regression(ax, 2, 2, ssp5_prtp * 100, ssp4_prtp * 100, ssp3_prtp * 100, ssp2_prtp * 100, ssp1_prtp * 100, ssp5_d2100, ssp4_d2100, ssp3_d2100, ssp2_d2100, ssp1_d2100, \n",
    "           r\"$\\delta$ (%)\", None)\n",
    "\n",
    "make_panel_w_regression(ax, 2, 3, ssp5_prtp * 100, ssp4_prtp * 100, ssp3_prtp * 100, ssp2_prtp * 100, ssp1_prtp * 100, ssp5_d2150, ssp4_d2150, ssp3_d2150, ssp2_d2150, ssp1_d2150, \n",
    "           r\"$\\delta$ (%)\", None)\n",
    "\n",
    "make_panel_w_regression(ax, 2, 4, ssp5_prtp * 100, ssp4_prtp * 100, ssp3_prtp * 100, ssp2_prtp * 100, ssp1_prtp * 100, ssp5_d2200, ssp4_d2200, ssp3_d2200, ssp2_d2200, ssp1_d2200, \n",
    "           r\"$\\delta$ (%)\", None)\n",
    "\n",
    "###########################\n",
    "# EX TECH GROWTH ROW\n",
    "###########################\n",
    "make_panel_w_regression(ax, 3, 0, ssp5_tottech[:, 1], ssp4_tottech[:, 1], ssp3_tottech[:, 1], \n",
    "                        ssp2_tottech[:, 1], ssp1_tottech[:, 1], ssp5_d2030, ssp4_d2030, ssp3_d2030, \n",
    "                        ssp2_d2030, ssp1_d2030, \"Tech Growth (%)\", \"Damage (frac GDP)\")\n",
    "\n",
    "make_panel_w_regression(ax, 3, 1, ssp5_tottech[:, 2], ssp4_tottech[:, 2], ssp3_tottech[:, 2], \n",
    "                        ssp2_tottech[:, 2], ssp1_tottech[:, 2], ssp5_d2060, ssp4_d2060, ssp3_d2060, \n",
    "                        ssp2_d2060, ssp1_d2060, \"Tech Growth (%)\", None)\n",
    "\n",
    "make_panel_w_regression(ax, 3, 2, ssp5_tottech[:, 3], ssp4_tottech[:, 3], ssp3_tottech[:, 3], \n",
    "                        ssp2_tottech[:, 3], ssp1_tottech[:, 3], ssp5_d2100, ssp4_d2100, ssp3_d2100, \n",
    "                        ssp2_d2100, ssp1_d2100, \"Tech Growth (%)\", None)\n",
    "\n",
    "make_panel_w_regression(ax, 3, 3, ssp5_tottech[:, 4], ssp4_tottech[:, 4], ssp3_tottech[:, 4], \n",
    "                        ssp2_tottech[:, 4], ssp1_tottech[:, 4], ssp5_d2150, ssp4_d2150, ssp3_d2150, \n",
    "                        ssp2_d2150, ssp1_d2150, \"Tech Growth (%)\", None)\n",
    "\n",
    "make_panel_w_regression(ax, 3, 4, ssp5_tottech[:, 5], ssp4_tottech[:, 5], ssp3_tottech[:, 5], \n",
    "                        ssp2_tottech[:, 5], ssp1_tottech[:, 5], ssp5_d2200, ssp4_d2200, ssp3_d2200, \n",
    "                        ssp2_d2200, ssp1_d2200, \"Tech Growth (%)\", None)\n",
    "\n",
    "ax[3,0].set_xticks([0,1,2,3])\n",
    "ax[3,0].set_xlim((0,3))\n",
    "ax[3,1].set_xticks([0,1,2,3,4,5])\n",
    "ax[3,2].set_xticks([0,1,2,3,4,5,6])\n",
    "ax[3,3].set_xticks([0,1,2,3,4,5,6])\n",
    "ax[3,4].set_xticks([0,1,2,3,4,5,6])\n",
    "\n",
    "fig.tight_layout()\n",
    "sns.despine(trim=True, offset=10)\n",
    "\n",
    "if save_fig:\n",
    "    fig.savefig(''.join([basefile, 'regression-sticky-dam.png']), dpi=400)    "
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "45e503b1",
   "metadata": {},
   "outputs": [],
   "source": [
    "ssp5_eis_r2_d = get_r2_in_time(ssp5_eis, ssp5_exp_dam)\n",
    "ssp5_ra_r2_d = get_r2_in_time(ssp5_ra, ssp5_exp_dam)\n",
    "ssp5_prtp_r2_d = get_r2_in_time(ssp5_prtp, ssp5_exp_dam)\n",
    "ssp5_tech_r2_d = get_tech_r2_in_time_alllin(ssp5_tottech, ssp5_exp_dam)\n",
    "\n",
    "ssp4_eis_r2_d = get_r2_in_time(ssp4_eis, ssp4_exp_dam)\n",
    "ssp4_ra_r2_d = get_r2_in_time(ssp4_ra, ssp4_exp_dam)\n",
    "ssp4_prtp_r2_d = get_r2_in_time(ssp4_prtp, ssp4_exp_dam)\n",
    "ssp4_tech_r2_d = get_tech_r2_in_time_alllin(ssp4_tottech, ssp4_exp_dam)\n",
    "\n",
    "ssp3_eis_r2_d = get_r2_in_time(ssp3_eis, ssp3_exp_dam)\n",
    "ssp3_ra_r2_d = get_r2_in_time(ssp3_ra, ssp3_exp_dam)\n",
    "ssp3_prtp_r2_d = get_r2_in_time(ssp3_prtp, ssp3_exp_dam)\n",
    "ssp3_tech_r2_d = get_tech_r2_in_time_alllin(ssp3_tottech, ssp3_exp_dam)\n",
    "\n",
    "ssp2_eis_r2_d = get_r2_in_time(ssp2_eis, ssp2_exp_dam)\n",
    "ssp2_ra_r2_d = get_r2_in_time(ssp2_ra, ssp2_exp_dam)\n",
    "ssp2_prtp_r2_d = get_r2_in_time(ssp2_prtp, ssp2_exp_dam)\n",
    "ssp2_tech_r2_d = get_tech_r2_in_time_alllin(ssp2_tottech, ssp2_exp_dam)\n",
    "\n",
    "ssp1_eis_r2_d = get_r2_in_time(ssp1_eis, ssp1_exp_dam)\n",
    "ssp1_ra_r2_d = get_r2_in_time(ssp1_ra, ssp1_exp_dam)\n",
    "ssp1_prtp_r2_d = get_r2_in_time(ssp1_prtp, ssp1_exp_dam)\n",
    "ssp1_tech_r2_d = get_tech_r2_in_time_alllin(ssp1_tottech, ssp1_exp_dam)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "bf76595d",
   "metadata": {},
   "outputs": [],
   "source": [
    "fig, ax = plt.subplots(1,5, figsize=(20,20), sharey=True)\n",
    "\n",
    "ax[0].set_ylim((0,1))\n",
    "\n",
    "alpha = 0.8\n",
    "lw = 3\n",
    "l1cind = 0\n",
    "l2cind = 1\n",
    "l3cind = 2\n",
    "l4cind = 3\n",
    "l5cind = 4\n",
    "\n",
    "################################\n",
    "# SSP5\n",
    "################################\n",
    "ax[0].plot(node_times[1:], ssp5_ra_r2_d[1:], linestyle='dashed', linewidth=lw, \n",
    "        color=color_list[l1cind])\n",
    "ax[0].fill_between(node_times[1:], np.zeros_like(node_times[1:]), ssp5_ra_r2_d[1:], alpha=alpha, \n",
    "                color=color_list[l1cind], label=\"RA\")\n",
    "\n",
    "ax[0].plot(node_times[1:], ssp5_eis_r2_d[1:] + ssp5_ra_r2_d[1:], linestyle='dashed', linewidth=lw, \n",
    "        color=color_list[l2cind])\n",
    "ax[0].fill_between(node_times[1:], ssp5_ra_r2_d[1:], ssp5_ra_r2_d[1:] + ssp5_eis_r2_d[1:], alpha=alpha, \n",
    "                color=color_list[l2cind], label=\"EIS\")\n",
    "\n",
    "ax[0].plot(node_times[1:], ssp5_eis_r2_d[1:] + ssp5_ra_r2_d[1:] + ssp5_prtp_r2_d[1:], linestyle='dashed', linewidth=lw, \n",
    "        color=color_list[l3cind])\n",
    "ax[0].fill_between(node_times[1:], ssp5_eis_r2_d[1:] + ssp5_ra_r2_d[1:], ssp5_ra_r2_d[1:] + ssp5_eis_r2_d[1:] + ssp5_prtp_r2_d[1:], alpha=alpha, \n",
    "                color=color_list[l3cind], label=\"PRTP\")\n",
    "\n",
    "ax[0].plot(node_times[1:], ssp5_eis_r2_d[1:] + ssp5_ra_r2_d[1:] + ssp5_prtp_r2_d[1:] + ssp5_tech_r2_d[1:], linestyle='dashed', linewidth=lw, \n",
    "        color=color_list[l4cind])\n",
    "ax[0].fill_between(node_times[1:], ssp5_eis_r2_d[1:] + ssp5_ra_r2_d[1:] + ssp5_prtp_r2_d[1:], ssp5_ra_r2_d[1:] + ssp5_eis_r2_d[1:] + ssp5_prtp_r2_d[1:] + ssp5_tech_r2_d[1:], alpha=alpha, \n",
    "                color=color_list[l4cind], label=\"Tech. Growth\")\n",
    "\n",
    "\n",
    "################################\n",
    "# SSP4\n",
    "################################\n",
    "ax[1].plot(node_times[1:], ssp4_ra_r2_d[1:], linestyle='dashed', linewidth=lw, \n",
    "        color=color_list[l1cind])\n",
    "ax[1].fill_between(node_times[1:], np.zeros_like(node_times[1:]), ssp4_ra_r2_d[1:], alpha=alpha, \n",
    "                color=color_list[l1cind], label=\"RA\")\n",
    "\n",
    "ax[1].plot(node_times[1:], ssp4_eis_r2_d[1:] + ssp4_ra_r2_d[1:], linestyle='dashed', linewidth=lw, \n",
    "        color=color_list[l2cind])\n",
    "ax[1].fill_between(node_times[1:], ssp4_ra_r2_d[1:], ssp5_ra_r2_d[1:] + ssp4_eis_r2_d[1:], alpha=alpha, \n",
    "                color=color_list[l2cind], label=\"EIS\")\n",
    "\n",
    "ax[1].plot(node_times[1:], ssp4_eis_r2_d[1:] + ssp4_ra_r2_d[1:] + ssp4_prtp_r2_d[1:], linestyle='dashed', linewidth=lw, \n",
    "        color=color_list[l3cind])\n",
    "ax[1].fill_between(node_times[1:], ssp4_eis_r2_d[1:] + ssp4_ra_r2_d[1:], ssp4_ra_r2_d[1:] + ssp4_eis_r2_d[1:] + ssp4_prtp_r2_d[1:], alpha=alpha, \n",
    "                color=color_list[l3cind], label=\"PRTP\")\n",
    "\n",
    "ax[1].plot(node_times[1:], ssp4_eis_r2_d[1:] + ssp4_ra_r2_d[1:] + ssp4_prtp_r2_d[1:] + ssp4_tech_r2_d[1:], linestyle='dashed', linewidth=lw, \n",
    "        color=color_list[l4cind])\n",
    "ax[1].fill_between(node_times[1:], ssp4_eis_r2_d[1:] + ssp4_ra_r2_d[1:] + ssp4_prtp_r2_d[1:], ssp4_ra_r2_d[1:] + ssp4_eis_r2_d[1:] + ssp4_prtp_r2_d[1:] + ssp4_tech_r2_d[1:], alpha=alpha, \n",
    "                color=color_list[l4cind], label=\"Tech. Growth\")\n",
    "\n",
    "################################\n",
    "# SSP3\n",
    "################################\n",
    "ax[2].plot(node_times[1:], ssp3_ra_r2_d[1:], linestyle='dashed', linewidth=lw, \n",
    "        color=color_list[l1cind])\n",
    "ax[2].fill_between(node_times[1:], np.zeros_like(node_times[1:]), ssp3_ra_r2_d[1:], alpha=alpha, \n",
    "                color=color_list[l1cind], label=\"RA\")\n",
    "\n",
    "ax[2].plot(node_times[1:], ssp3_eis_r2_d[1:] + ssp3_ra_r2_d[1:], linestyle='dashed', linewidth=lw, \n",
    "        color=color_list[l2cind])\n",
    "ax[2].fill_between(node_times[1:], ssp3_ra_r2_d[1:], ssp3_ra_r2_d[1:] + ssp3_eis_r2_d[1:], alpha=alpha, \n",
    "                color=color_list[l2cind], label=\"EIS\")\n",
    "\n",
    "ax[2].plot(node_times[1:], ssp3_eis_r2_d[1:] + ssp3_ra_r2_d[1:] + ssp3_prtp_r2_d[1:], linestyle='dashed', linewidth=lw, \n",
    "        color=color_list[l3cind])\n",
    "ax[2].fill_between(node_times[1:], ssp3_eis_r2_d[1:] + ssp3_ra_r2_d[1:], ssp3_ra_r2_d[1:] + ssp3_eis_r2_d[1:] + ssp3_prtp_r2_d[1:], alpha=alpha, \n",
    "                color=color_list[l3cind], label=\"PRTP\")\n",
    "\n",
    "ax[2].plot(node_times[1:], ssp3_eis_r2_d[1:] + ssp3_ra_r2_d[1:] + ssp3_prtp_r2_d[1:] + ssp3_tech_r2_d[1:], linestyle='dashed', linewidth=lw, \n",
    "        color=color_list[l4cind])\n",
    "ax[2].fill_between(node_times[1:], ssp3_eis_r2_d[1:] + ssp3_ra_r2_d[1:] + ssp3_prtp_r2_d[1:], ssp3_ra_r2_d[1:] + ssp3_eis_r2_d[1:] + ssp3_prtp_r2_d[1:] + ssp3_tech_r2_d[1:], alpha=alpha, \n",
    "                color=color_list[l4cind], label=\"Tech. Growth\")\n",
    "\n",
    "################################\n",
    "# SSP2\n",
    "################################\n",
    "ax[3].plot(node_times[1:], ssp2_ra_r2_d[1:], linestyle='dashed', linewidth=lw, \n",
    "        color=color_list[l1cind])\n",
    "ax[3].fill_between(node_times[1:], np.zeros_like(node_times[1:]), ssp2_ra_r2_d[1:], alpha=alpha, \n",
    "                color=color_list[l1cind], label=\"RA\")\n",
    "\n",
    "ax[3].plot(node_times[1:], ssp2_eis_r2_d[1:] + ssp2_ra_r2_d[1:], linestyle='dashed', linewidth=lw, \n",
    "        color=color_list[l2cind])\n",
    "ax[3].fill_between(node_times[1:], ssp2_ra_r2_d[1:], ssp2_ra_r2_d[1:] + ssp2_eis_r2_d[1:], alpha=alpha, \n",
    "                color=color_list[l2cind], label=\"EIS\")\n",
    "\n",
    "ax[3].plot(node_times[1:], ssp2_eis_r2_d[1:] + ssp2_ra_r2_d[1:] + ssp2_prtp_r2_d[1:], linestyle='dashed', linewidth=lw, \n",
    "        color=color_list[l3cind])\n",
    "ax[3].fill_between(node_times[1:], ssp2_eis_r2_d[1:] + ssp2_ra_r2_d[1:], ssp2_ra_r2_d[1:] + ssp2_eis_r2_d[1:] + ssp2_prtp_r2_d[1:], alpha=alpha, \n",
    "                color=color_list[l3cind], label=\"PRTP\")\n",
    "\n",
    "ax[3].plot(node_times[1:], ssp2_eis_r2_d[1:] + ssp2_ra_r2_d[1:], ssp2_prtp_r2_d[1:] + ssp2_tech_r2_d[1:], linestyle='dashed', linewidth=lw, \n",
    "        color=color_list[l4cind])\n",
    "ax[3].fill_between(node_times[1:], ssp2_eis_r2_d[1:] + ssp2_ra_r2_d[1:] + ssp2_prtp_r2_d[1:], ssp2_ra_r2_d[1:] + ssp2_eis_r2_d[1:] + ssp2_prtp_r2_d[1:] + ssp2_tech_r2_d[1:], alpha=alpha, \n",
    "                color=color_list[l4cind], label=\"Tech. Growth\")\n",
    "\n",
    "\n",
    "################################\n",
    "# SSP1\n",
    "################################\n",
    "ax[4].plot(node_times[1:], ssp1_ra_r2_d[1:], linestyle='dashed', linewidth=lw, \n",
    "        color=color_list[l1cind])\n",
    "ax[4].fill_between(node_times[1:], np.zeros_like(node_times[1:]), ssp5_ra_r2_d[1:], alpha=alpha, \n",
    "                color=color_list[l1cind], label=\"RA\")\n",
    "\n",
    "ax[4].plot(node_times[1:], ssp1_eis_r2_d[1:] + ssp1_ra_r2_d[1:], linestyle='dashed', linewidth=lw, \n",
    "        color=color_list[l2cind])\n",
    "ax[4].fill_between(node_times[1:], ssp1_ra_r2_d[1:], ssp1_ra_r2_d[1:] + ssp1_eis_r2_d[1:], alpha=alpha, \n",
    "                color=color_list[l2cind], label=\"EIS\")\n",
    "\n",
    "ax[4].plot(node_times[1:], ssp1_eis_r2_d[1:] + ssp1_ra_r2_d[1:] + ssp1_prtp_r2_d[1:], linestyle='dashed', linewidth=lw, \n",
    "        color=color_list[l3cind])\n",
    "ax[4].fill_between(node_times[1:], ssp1_eis_r2_d[1:] + ssp1_ra_r2_d[1:], ssp1_ra_r2_d[1:] + ssp1_eis_r2_d[1:] + ssp1_prtp_r2_d[1:], alpha=alpha, \n",
    "                color=color_list[l3cind], label=\"PRTP\")\n",
    "\n",
    "ax[4].plot(node_times[1:], ssp1_eis_r2_d[1:] + ssp1_ra_r2_d[1:] + ssp1_prtp_r2_d[1:] + ssp1_tech_r2_d[1:], linestyle='dashed', linewidth=lw, \n",
    "        color=color_list[l4cind])\n",
    "ax[4].fill_between(node_times[1:], ssp1_eis_r2_d[1:] + ssp1_ra_r2_d[1:] + ssp1_prtp_r2_d[1:], ssp1_ra_r2_d[1:] + ssp1_eis_r2_d[1:] + ssp1_prtp_r2_d[1:] + ssp1_tech_r2_d[1:], alpha=alpha, \n",
    "                color=color_list[l4cind], label=\"Tech. Growth\")\n",
    "\n",
    "ax[2].legend(loc='upper center', bbox_to_anchor=(0.5, -0.25), fancybox=True, shadow=True, ncol=5)\n",
    "\n",
    "ax[0].set_ylabel(\"r$^{2}$\")\n",
    "ax[0].set_title(\"SSP5\")\n",
    "ax[1].set_title(\"SSP4\")\n",
    "ax[2].set_title(\"SSP3\")\n",
    "ax[3].set_title(\"SSP2\")\n",
    "ax[4].set_title(\"SSP1\")\n",
    "\n",
    "for i in range(5):\n",
    "    ax[i].set_xlim((node_times[1], node_times[-1]))\n",
    "    ax[i].set_aspect(1.0/ax[i].get_data_ratio(), adjustable='box')\n",
    "    ax[i].set_xlabel(\"Year\")\n",
    "\n",
    "#fig.tight_layout()"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "afca5d59",
   "metadata": {},
   "outputs": [],
   "source": [
    "fig, ax = plt.subplots(1,5, figsize=(20,20), sharey=True)\n",
    "\n",
    "ax[0].set_ylim((0,1))\n",
    "\n",
    "alpha = 0.8\n",
    "lw = 3\n",
    "l1cind = 0\n",
    "l2cind = 1\n",
    "l3cind = 2\n",
    "l4cind = 3\n",
    "l5cind = 4\n",
    "\n",
    "################################\n",
    "# SSP5\n",
    "################################\n",
    "tot_ssp5 = ssp5_ra_r2_d[1:] + ssp5_eis_r2_d[1:] + ssp5_prtp_r2_d[1:] + ssp5_tech_r2_d[1:]\n",
    "ax[0].plot(node_times[1:], ssp5_ra_r2_d[1:] * tot_ssp5**(-1), linestyle='dashed', linewidth=lw, \n",
    "        color=color_list[l1cind])\n",
    "ax[0].fill_between(node_times[1:], np.zeros_like(node_times[1:]), ssp5_ra_r2_d[1:] * tot_ssp5**(-1), alpha=alpha, \n",
    "                color=color_list[l1cind], label=\"RA\")\n",
    "\n",
    "ax[0].plot(node_times[1:], (ssp5_eis_r2_d[1:] + ssp5_ra_r2_d[1:])  * tot_ssp5**(-1), linestyle='dashed', linewidth=lw, \n",
    "        color=color_list[l2cind])\n",
    "ax[0].fill_between(node_times[1:], ssp5_ra_r2_co[1:]  * tot_ssp5**(-1), \n",
    "                   (ssp5_ra_r2_d[1:] + ssp5_eis_r2_d[1:]) * tot_ssp5**(-1), alpha=alpha, \n",
    "                color=color_list[l2cind], label=\"EIS\")\n",
    "\n",
    "ax[0].plot(node_times[1:], (ssp5_eis_r2_d[1:] + ssp5_ra_r2_d[1:] + ssp5_prtp_r2_d[1:])  * tot_ssp5**(-1), linestyle='dashed', linewidth=lw, \n",
    "        color=color_list[l3cind])\n",
    "ax[0].fill_between(node_times[1:], (ssp5_eis_r2_d[1:] + ssp5_ra_r2_d[1:]) * tot_ssp5**(-1), \n",
    "                  (ssp5_ra_r2_d[1:] + ssp5_eis_r2_d[1:] + ssp5_prtp_r2_d[1:]) * tot_ssp5**(-1), alpha=alpha, \n",
    "                color=color_list[l3cind], label=\"PRTP\")\n",
    "\n",
    "ax[0].plot(node_times[1:], (ssp5_eis_r2_d[1:] + ssp5_ra_r2_d[1:] + ssp5_prtp_r2_d[1:] + ssp5_tech_r2_d[1:]) * tot_ssp5**(-1), \n",
    "           linestyle='dashed', linewidth=lw, color=color_list[l4cind])\n",
    "ax[0].fill_between(node_times[1:], (ssp5_eis_r2_d[1:] + ssp5_ra_r2_d[1:] + ssp5_prtp_r2_d[1:]) * tot_ssp5**(-1), \n",
    "                   (ssp5_ra_r2_d[1:] + ssp5_eis_r2_d[1:] + ssp5_prtp_r2_d[1:] + ssp5_tech_r2_d[1:]) * tot_ssp5**(-1), alpha=alpha, \n",
    "                color=color_list[l4cind], label=\"Tech. Growth\")\n",
    "\n",
    "\n",
    "################################\n",
    "# SSP4\n",
    "################################\n",
    "tot_ssp4 = ssp4_ra_r2_d[1:] + ssp4_eis_r2_d[1:] + ssp4_prtp_r2_d[1:] + ssp4_tech_r2_d[1:]\n",
    "ax[1].plot(node_times[1:], ssp4_ra_r2_d[1:] * tot_ssp4**(-1), linestyle='dashed', linewidth=lw, \n",
    "        color=color_list[l1cind])\n",
    "ax[1].fill_between(node_times[1:], np.zeros_like(node_times[1:]), ssp4_ra_r2_d[1:] * tot_ssp4**(-1), alpha=alpha, \n",
    "                color=color_list[l1cind], label=\"RA\")\n",
    "\n",
    "ax[1].plot(node_times[1:], (ssp4_eis_r2_d[1:] + ssp4_ra_r2_d[1:]) * tot_ssp4**(-1), linestyle='dashed', linewidth=lw, \n",
    "        color=color_list[l2cind])\n",
    "ax[1].fill_between(node_times[1:], ssp4_ra_r2_d[1:] * tot_ssp4**(-1), \n",
    "                   (ssp4_ra_r2_d[1:] + ssp4_eis_r2_d[1:]) * tot_ssp4**(-1), alpha=alpha, \n",
    "                color=color_list[l2cind], label=\"EIS\")\n",
    "\n",
    "ax[1].plot(node_times[1:], (ssp4_eis_r2_d[1:] + ssp4_ra_r2_d[1:] + ssp4_prtp_r2_d[1:]) * tot_ssp4**(-1), linestyle='dashed', linewidth=lw, \n",
    "        color=color_list[l3cind])\n",
    "ax[1].fill_between(node_times[1:], (ssp4_eis_r2_d[1:] + ssp4_ra_r2_d[1:]) * tot_ssp4**(-1), \n",
    "                  (ssp4_ra_r2_d[1:] + ssp4_eis_r2_d[1:] + ssp4_prtp_r2_d[1:]) * tot_ssp4**(-1), alpha=alpha, \n",
    "                color=color_list[l3cind], label=\"PRTP\")\n",
    "\n",
    "ax[1].plot(node_times[1:], (ssp4_eis_r2_d[1:] + ssp4_ra_r2_d[1:] + ssp4_prtp_r2_d[1:] + ssp4_tech_r2_d[1:]) * tot_ssp4**(-1), \n",
    "           linestyle='dashed', linewidth=lw, color=color_list[l4cind])\n",
    "ax[1].fill_between(node_times[1:], (ssp4_eis_r2_d[1:] + ssp4_ra_r2_d[1:] + ssp4_prtp_r2_d[1:]) * tot_ssp4**(-1), \n",
    "                   (ssp4_ra_r2_d[1:] + ssp4_eis_r2_d[1:] + ssp4_prtp_r2_d[1:] + ssp4_tech_r2_d[1:]) * tot_ssp4**(-1), alpha=alpha, \n",
    "                color=color_list[l4cind], label=\"Tech. Growth\")\n",
    "\n",
    "################################\n",
    "# SSP3\n",
    "################################\n",
    "ssp3_tot = ssp3_ra_r2_d[1:] + ssp3_eis_r2_d[1:] + ssp3_prtp_r2_d[1:] + ssp3_tech_r2_d[1:]\n",
    "ax[2].plot(node_times[1:], ssp3_ra_r2_d[1:] * ssp3_tot**(-1), linestyle='dashed', linewidth=lw, \n",
    "        color=color_list[l1cind])\n",
    "ax[2].fill_between(node_times[1:], np.zeros_like(node_times[1:]), ssp3_ra_r2_d[1:] * ssp3_tot**(-1), alpha=alpha, \n",
    "                color=color_list[l1cind], label=\"RA\")\n",
    "\n",
    "ax[2].plot(node_times[1:], (ssp3_eis_r2_d[1:] + ssp3_ra_r2_d[1:]) * ssp3_tot**(-1), linestyle='dashed', linewidth=lw, \n",
    "        color=color_list[l2cind])\n",
    "ax[2].fill_between(node_times[1:], ssp3_ra_r2_d[1:] * ssp3_tot**(-1), \n",
    "                   (ssp3_ra_r2_d[1:] + ssp3_eis_r2_d[1:]) * ssp3_tot**(-1), alpha=alpha, \n",
    "                color=color_list[l2cind], label=\"EIS\")\n",
    "\n",
    "ax[2].plot(node_times[1:], (ssp3_eis_r2_d[1:] + ssp3_ra_r2_d[1:] + ssp3_prtp_r2_d[1:]) * ssp3_tot**(-1), linestyle='dashed', linewidth=lw, \n",
    "        color=color_list[l3cind])\n",
    "ax[2].fill_between(node_times[1:], (ssp3_eis_r2_d[1:] + ssp3_ra_r2_d[1:]) * ssp3_tot**(-1), \n",
    "                   (ssp3_ra_r2_d[1:] + ssp3_eis_r2_d[1:] + ssp3_prtp_r2_d[1:]) * ssp3_tot**(-1), alpha=alpha, \n",
    "                color=color_list[l3cind], label=\"PRTP\")\n",
    "\n",
    "ax[2].plot(node_times[1:], (ssp3_eis_r2_d[1:] + ssp3_ra_r2_d[1:] + ssp3_prtp_r2_d[1:] + ssp3_tech_r2_d[1:]) * ssp3_tot**(-1), linestyle='dashed', linewidth=lw, \n",
    "        color=color_list[l4cind])\n",
    "ax[2].fill_between(node_times[1:], (ssp3_eis_r2_d[1:] + ssp3_ra_r2_d[1:] + ssp3_prtp_r2_d[1:]) * ssp3_tot**(-1), \n",
    "                   (ssp3_ra_r2_d[1:] + ssp3_eis_r2_d[1:] + ssp3_prtp_r2_d[1:] + ssp3_tech_r2_d[1:]) * ssp3_tot**(-1), alpha=alpha, \n",
    "                color=color_list[l4cind], label=\"Tech. Growth\")\n",
    "\n",
    "################################\n",
    "# SSP2\n",
    "################################\n",
    "ssp2_tot = ssp2_ra_r2_d[1:] + ssp2_eis_r2_d[1:] + ssp2_prtp_r2_d[1:] + ssp2_tech_r2_d[1:]\n",
    "ax[3].plot(node_times[1:], ssp2_ra_r2_d[1:] * ssp2_tot**(-1), linestyle='dashed', linewidth=lw, \n",
    "        color=color_list[l1cind])\n",
    "ax[3].fill_between(node_times[1:], np.zeros_like(node_times[1:]), ssp2_ra_r2_d[1:]* ssp2_tot**(-1), alpha=alpha, \n",
    "                color=color_list[l1cind], label=\"RA\")\n",
    "\n",
    "ax[3].plot(node_times[1:], (ssp2_eis_r2_d[1:] + ssp2_ra_r2_d[1:])* ssp2_tot**(-1), linestyle='dashed', linewidth=lw, \n",
    "        color=color_list[l2cind])\n",
    "ax[3].fill_between(node_times[1:], ssp2_ra_r2_d[1:] * ssp2_tot**(-1), \n",
    "                   (ssp2_ra_r2_d[1:] + ssp2_eis_r2_d[1:]) * ssp2_tot**(-1), alpha=alpha, \n",
    "                color=color_list[l2cind], label=\"EIS\")\n",
    "\n",
    "ax[3].plot(node_times[1:], (ssp2_eis_r2_d[1:] + ssp2_ra_r2_d[1:] + ssp2_prtp_r2_d[1:]) * ssp2_tot**(-1), linestyle='dashed', linewidth=lw, \n",
    "        color=color_list[l3cind])\n",
    "ax[3].fill_between(node_times[1:], (ssp2_eis_r2_d[1:] + ssp2_ra_r2_d[1:]) * ssp2_tot**(-1), \n",
    "                   (ssp2_ra_r2_d[1:] + ssp2_eis_r2_d[1:] + ssp2_prtp_r2_d[1:]) * ssp2_tot**(-1), alpha=alpha, \n",
    "                color=color_list[l3cind], label=\"PRTP\")\n",
    "\n",
    "ax[3].plot(node_times[1:], (ssp2_eis_r2_d[1:] + ssp2_ra_r2_d[1:] + ssp2_prtp_r2_d[1:] + ssp2_tech_r2_d[1:]) * ssp2_tot**(-1), linestyle='dashed', linewidth=lw, \n",
    "        color=color_list[l4cind])\n",
    "ax[3].fill_between(node_times[1:], (ssp2_eis_r2_d[1:] + ssp2_ra_r2_d[1:] + ssp2_prtp_r2_d[1:]) * ssp2_tot**(-1), \n",
    "                   (ssp2_ra_r2_d[1:] + ssp2_eis_r2_d[1:] + ssp2_prtp_r2_d[1:] + ssp2_tech_r2_d[1:]) * ssp2_tot**(-1), alpha=alpha, \n",
    "                color=color_list[l4cind], label=\"Tech. Growth\")\n",
    "\n",
    "\n",
    "################################\n",
    "# SSP1\n",
    "################################\n",
    "ssp1_tot = ssp1_ra_r2_d[1:] + ssp1_eis_r2_d[1:] + ssp1_prtp_r2_d[1:] + ssp1_tech_r2_d[1:]\n",
    "ax[4].plot(node_times[1:], ssp1_ra_r2_d[1:] * ssp1_tot**(-1), linestyle='dashed', linewidth=lw, \n",
    "        color=color_list[l1cind])\n",
    "ax[4].fill_between(node_times[1:], np.zeros_like(node_times[1:]), ssp5_ra_r2_d[1:] * ssp1_tot**(-1), alpha=alpha, \n",
    "                color=color_list[l1cind], label=\"RA\")\n",
    "\n",
    "ax[4].plot(node_times[1:], (ssp1_eis_r2_d[1:] + ssp1_ra_r2_d[1:]) * ssp1_tot**(-1), linestyle='dashed', linewidth=lw, \n",
    "        color=color_list[l2cind])\n",
    "ax[4].fill_between(node_times[1:], ssp1_ra_r2_d[1:] * ssp1_tot**(-1), \n",
    "                   (ssp1_ra_r2_d[1:] + ssp1_eis_r2_d[1:]) * ssp1_tot**(-1), alpha=alpha, \n",
    "                    color=color_list[l2cind], label=\"EIS\")\n",
    "\n",
    "ax[4].plot(node_times[1:], (ssp1_eis_r2_d[1:] + ssp1_ra_r2_d[1:] + ssp1_prtp_r2_d[1:]) * ssp1_tot**(-1), linestyle='dashed', linewidth=lw, \n",
    "        color=color_list[l3cind])\n",
    "ax[4].fill_between(node_times[1:], (ssp1_eis_r2_d[1:] + ssp1_ra_r2_d[1:]) * ssp1_tot**(-1), \n",
    "                   (ssp1_ra_r2_d[1:] + ssp1_eis_r2_d[1:] + ssp1_prtp_r2_d[1:]) * ssp1_tot**(-1), alpha=alpha, \n",
    "                    color=color_list[l3cind], label=\"PRTP\")\n",
    "\n",
    "ax[4].plot(node_times[1:], (ssp1_eis_r2_d[1:] + ssp1_ra_r2_d[1:] + ssp1_prtp_r2_d[1:] + ssp1_tech_r2_d[1:]) * ssp1_tot**(-1), linestyle='dashed', linewidth=lw, \n",
    "        color=color_list[l4cind])\n",
    "ax[4].fill_between(node_times[1:], (ssp1_eis_r2_d[1:] + ssp1_ra_r2_d[1:] + ssp1_prtp_r2_d[1:]) * ssp1_tot**(-1), \n",
    "                   (ssp1_ra_r2_d[1:] + ssp1_eis_r2_d[1:] + ssp1_prtp_r2_d[1:] + ssp1_tech_r2_d[1:]) * ssp1_tot**(-1), alpha=alpha, \n",
    "                color=color_list[l4cind], label=\"Tech. Growth\")\n",
    "\n",
    "ax[2].legend(loc='upper center', bbox_to_anchor=(0.5, -0.25), fancybox=True, shadow=True, ncol=5, fontsize=20)\n",
    "\n",
    "ax[0].set_ylabel(\"$r^{2} / r_{tot}^{2}$\")\n",
    "ax[0].set_title(\"SSP5\")\n",
    "ax[1].set_title(\"SSP4\")\n",
    "ax[2].set_title(\"SSP3\")\n",
    "ax[3].set_title(\"SSP2\")\n",
    "ax[4].set_title(\"SSP1\")\n",
    "\n",
    "panel_label = [\"A\", \"B\", \"C\", \"D\", \"E\"]\n",
    "for i in range(5):\n",
    "    ax[i].set_xlim((node_times[1], node_times[-1]))\n",
    "    ax[i].set_aspect(1.0/ax[i].get_data_ratio(), adjustable='box')\n",
    "    ax[i].set_xlabel(\"Year\")\n",
    "    ax[i].text(2168, 0.88, panel_label[i], fontsize=22, color='white', fontweight='bold')\n",
    "    ax[i].set_xticks([2030,2100,2200])\n",
    "    ax[i].set_xticklabels(['2030', '2100', '2200'])\n",
    "    ax[i].tick_params(axis='both', labelsize=14)\n",
    "\n",
    "sns.despine(trim=True, offset=10)\n",
    "\n",
    "if save_fig:\n",
    "    fig.savefig(''.join([basefile, 'rsquared-dam.png']), dpi=400)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "83952115",
   "metadata": {},
   "outputs": [],
   "source": [
    "import matplotlib.transforms as mtransforms\n",
    "\n",
    "cost_row = ['A', 'B', 'C', 'D', 'E']\n",
    "t_row = ['F', 'G', 'H', 'I', 'J']\n",
    "conc_row = ['K', 'L', 'M', 'N', 'O']\n",
    "dam_row = ['P', 'Q', 'R', 'S', 'T']\n",
    "\n",
    "fig, ax = plt.subplot_mosaic([cost_row, t_row, conc_row, dam_row], sharex=False,\n",
    "                             gridspec_kw={'height_ratios': [1, 1, 1, 1], 'width_ratios': [1, 1, 1, 1, 1]},\n",
    "                            figsize=(25,19))\n",
    "\n",
    "alpha = 0.8\n",
    "lw = 3\n",
    "l1cind = 0\n",
    "l2cind = 1\n",
    "l3cind = 2\n",
    "l4cind = 3\n",
    "l5cind = 4\n",
    "\n",
    "################################\n",
    "# SSP5\n",
    "################################\n",
    "tot_ssp5 = ssp5_ra_r2 + ssp5_eis_r2 + ssp5_prtp_r2 + ssp5_tech_r2\n",
    "ax['A'].plot(node_times, ssp5_ra_r2 * tot_ssp5**(-1), linestyle='dashed', linewidth=lw, \n",
    "        color=color_list[l1cind])\n",
    "ax['A'].fill_between(node_times, np.zeros_like(node_times), ssp5_ra_r2 * tot_ssp5**(-1), alpha=alpha, \n",
    "                color=color_list[l1cind], label=\"RA\")\n",
    "\n",
    "ax['A'].plot(node_times, (ssp5_eis_r2 + ssp5_ra_r2)  * tot_ssp5**(-1), linestyle='dashed', linewidth=lw, \n",
    "        color=color_list[l2cind])\n",
    "ax['A'].fill_between(node_times, ssp5_ra_r2  * tot_ssp5**(-1), \n",
    "                   (ssp5_ra_r2 + ssp5_eis_r2) * tot_ssp5**(-1), alpha=alpha, \n",
    "                color=color_list[l2cind], label=\"EIS\")\n",
    "\n",
    "ax['A'].plot(node_times, (ssp5_eis_r2 + ssp5_ra_r2 + ssp5_prtp_r2)  * tot_ssp5**(-1), linestyle='dashed', linewidth=lw, \n",
    "        color=color_list[l3cind])\n",
    "ax['A'].fill_between(node_times, (ssp5_eis_r2 + ssp5_ra_r2) * tot_ssp5**(-1), \n",
    "                  (ssp5_ra_r2 + ssp5_eis_r2 + ssp5_prtp_r2) * tot_ssp5**(-1), alpha=alpha, \n",
    "                color=color_list[l3cind], label=\"PRTP\")\n",
    "\n",
    "ax['A'].plot(node_times, (ssp5_eis_r2 + ssp5_ra_r2 + ssp5_prtp_r2 + ssp5_tech_r2) * tot_ssp5**(-1), \n",
    "           linestyle='dashed', linewidth=lw, color=color_list[l4cind])\n",
    "ax['A'].fill_between(node_times, (ssp5_eis_r2 + ssp5_ra_r2 + ssp5_prtp_r2) * tot_ssp5**(-1), \n",
    "                   (ssp5_ra_r2 + ssp5_eis_r2 + ssp5_prtp_r2 + ssp5_tech_r2) * tot_ssp5**(-1), alpha=alpha, \n",
    "                color=color_list[l4cind], label=\"Tech. Growth\")\n",
    "\n",
    "\n",
    "################################\n",
    "# SSP4\n",
    "################################\n",
    "tot_ssp4 = ssp4_ra_r2 + ssp4_eis_r2 + ssp4_prtp_r2 + ssp4_tech_r2\n",
    "ax['B'].plot(node_times, ssp4_ra_r2 * tot_ssp4**(-1), linestyle='dashed', linewidth=lw, \n",
    "        color=color_list[l1cind])\n",
    "ax['B'].fill_between(node_times, np.zeros_like(node_times), ssp4_ra_r2 * tot_ssp4**(-1), alpha=alpha, \n",
    "                color=color_list[l1cind], label=\"RA\")\n",
    "\n",
    "ax['B'].plot(node_times, (ssp4_eis_r2 + ssp4_ra_r2) * tot_ssp4**(-1), linestyle='dashed', linewidth=lw, \n",
    "        color=color_list[l2cind])\n",
    "ax['B'].fill_between(node_times, ssp4_ra_r2 * tot_ssp4**(-1), \n",
    "                   (ssp4_ra_r2 + ssp4_eis_r2) * tot_ssp4**(-1), alpha=alpha, \n",
    "                color=color_list[l2cind], label=\"EIS\")\n",
    "\n",
    "ax['B'].plot(node_times, (ssp4_eis_r2 + ssp4_ra_r2 + ssp4_prtp_r2) * tot_ssp4**(-1), linestyle='dashed', linewidth=lw, \n",
    "        color=color_list[l3cind])\n",
    "ax['B'].fill_between(node_times, (ssp4_eis_r2 + ssp4_ra_r2) * tot_ssp4**(-1), \n",
    "                  (ssp4_ra_r2 + ssp4_eis_r2 + ssp4_prtp_r2) * tot_ssp4**(-1), alpha=alpha, \n",
    "                color=color_list[l3cind], label=\"PRTP\")\n",
    "\n",
    "ax['B'].plot(node_times, (ssp4_eis_r2 + ssp4_ra_r2 + ssp4_prtp_r2 + ssp4_tech_r2) * tot_ssp4**(-1), \n",
    "           linestyle='dashed', linewidth=lw, color=color_list[l4cind])\n",
    "ax['B'].fill_between(node_times, (ssp4_eis_r2 + ssp4_ra_r2 + ssp4_prtp_r2) * tot_ssp4**(-1), \n",
    "                   (ssp4_ra_r2 + ssp4_eis_r2 + ssp4_prtp_r2 + ssp4_tech_r2) * tot_ssp4**(-1), alpha=alpha, \n",
    "                color=color_list[l4cind], label=\"Tech. Growth\")\n",
    "\n",
    "################################\n",
    "# SSP3\n",
    "################################\n",
    "ssp3_tot = ssp3_ra_r2 + ssp3_eis_r2 + ssp3_prtp_r2 + ssp3_tech_r2\n",
    "ax['C'].plot(node_times, ssp3_ra_r2 * ssp3_tot**(-1), linestyle='dashed', linewidth=lw, \n",
    "        color=color_list[l1cind])\n",
    "ax['C'].fill_between(node_times, np.zeros_like(node_times), ssp3_ra_r2 * ssp3_tot**(-1), alpha=alpha, \n",
    "                color=color_list[l1cind], label=\"RA\")\n",
    "\n",
    "ax['C'].plot(node_times, (ssp3_eis_r2 + ssp3_ra_r2) * ssp3_tot**(-1), linestyle='dashed', linewidth=lw, \n",
    "        color=color_list[l2cind])\n",
    "ax['C'].fill_between(node_times, ssp3_ra_r2 * ssp3_tot**(-1), \n",
    "                   (ssp3_ra_r2 + ssp3_eis_r2) * ssp3_tot**(-1), alpha=alpha, \n",
    "                color=color_list[l2cind], label=\"EIS\")\n",
    "\n",
    "ax['C'].plot(node_times, (ssp3_eis_r2 + ssp3_ra_r2 + ssp3_prtp_r2) * ssp3_tot**(-1), linestyle='dashed', linewidth=lw, \n",
    "        color=color_list[l3cind])\n",
    "ax['C'].fill_between(node_times, (ssp3_eis_r2 + ssp3_ra_r2) * ssp3_tot**(-1), \n",
    "                   (ssp3_ra_r2 + ssp3_eis_r2 + ssp3_prtp_r2) * ssp3_tot**(-1), alpha=alpha, \n",
    "                color=color_list[l3cind], label=\"PRTP\")\n",
    "\n",
    "ax['C'].plot(node_times, (ssp3_eis_r2 + ssp3_ra_r2 + ssp3_prtp_r2 + ssp3_tech_r2) * ssp3_tot**(-1), linestyle='dashed', linewidth=lw, \n",
    "        color=color_list[l4cind])\n",
    "ax['C'].fill_between(node_times, (ssp3_eis_r2 + ssp3_ra_r2 + ssp3_prtp_r2) * ssp3_tot**(-1), \n",
    "                   (ssp3_ra_r2 + ssp3_eis_r2 + ssp3_prtp_r2 + ssp3_tech_r2) * ssp3_tot**(-1), alpha=alpha, \n",
    "                color=color_list[l4cind], label=\"Tech. Growth\")\n",
    "\n",
    "################################\n",
    "# SSP2\n",
    "################################\n",
    "ssp2_tot = ssp2_ra_r2 + ssp2_eis_r2 + ssp2_prtp_r2 + ssp2_tech_r2\n",
    "ax['D'].plot(node_times, ssp2_ra_r2 * ssp2_tot**(-1), linestyle='dashed', linewidth=lw, \n",
    "        color=color_list[l1cind])\n",
    "ax['D'].fill_between(node_times, np.zeros_like(node_times), ssp2_ra_r2* ssp2_tot**(-1), alpha=alpha, \n",
    "                color=color_list[l1cind], label=\"RA\")\n",
    "\n",
    "ax['D'].plot(node_times, (ssp2_eis_r2 + ssp2_ra_r2)* ssp2_tot**(-1), linestyle='dashed', linewidth=lw, \n",
    "        color=color_list[l2cind])\n",
    "ax['D'].fill_between(node_times, ssp2_ra_r2 * ssp2_tot**(-1), \n",
    "                   (ssp2_ra_r2 + ssp2_eis_r2) * ssp2_tot**(-1), alpha=alpha, \n",
    "                color=color_list[l2cind], label=\"EIS\")\n",
    "\n",
    "ax['D'].plot(node_times, (ssp2_eis_r2 + ssp2_ra_r2 + ssp2_prtp_r2) * ssp2_tot**(-1), linestyle='dashed', linewidth=lw, \n",
    "        color=color_list[l3cind])\n",
    "ax['D'].fill_between(node_times, (ssp2_eis_r2 + ssp2_ra_r2) * ssp2_tot**(-1), \n",
    "                   (ssp2_ra_r2 + ssp2_eis_r2 + ssp2_prtp_r2) * ssp2_tot**(-1), alpha=alpha, \n",
    "                color=color_list[l3cind], label=\"PRTP\")\n",
    "\n",
    "ax['D'].plot(node_times, (ssp2_eis_r2 + ssp2_ra_r2 + ssp2_prtp_r2 + ssp2_tech_r2) * ssp2_tot**(-1), linestyle='dashed', linewidth=lw, \n",
    "        color=color_list[l4cind])\n",
    "ax['D'].fill_between(node_times, (ssp2_eis_r2 + ssp2_ra_r2 + ssp2_prtp_r2) * ssp2_tot**(-1), \n",
    "                   (ssp2_ra_r2 + ssp2_eis_r2 + ssp2_prtp_r2 + ssp2_tech_r2) * ssp2_tot**(-1), alpha=alpha, \n",
    "                color=color_list[l4cind], label=\"Tech. Growth\")\n",
    "\n",
    "\n",
    "################################\n",
    "# SSP1\n",
    "################################\n",
    "ssp1_tot = ssp1_ra_r2 + ssp1_eis_r2 + ssp1_prtp_r2 + ssp1_tech_r2\n",
    "ax['E'].plot(node_times, ssp1_ra_r2 * ssp1_tot**(-1), linestyle='dashed', linewidth=lw, \n",
    "        color=color_list[l1cind])\n",
    "ax['E'].fill_between(node_times, np.zeros_like(node_times), ssp5_ra_r2 * ssp1_tot**(-1), alpha=alpha, \n",
    "                color=color_list[l1cind], label=\"RA\")\n",
    "\n",
    "ax['E'].plot(node_times, (ssp1_eis_r2 + ssp1_ra_r2) * ssp1_tot**(-1), linestyle='dashed', linewidth=lw, \n",
    "        color=color_list[l2cind])\n",
    "ax['E'].fill_between(node_times, ssp1_ra_r2 * ssp1_tot**(-1), \n",
    "                   (ssp1_ra_r2 + ssp1_eis_r2) * ssp1_tot**(-1), alpha=alpha, \n",
    "                    color=color_list[l2cind], label=\"EIS\")\n",
    "\n",
    "ax['E'].plot(node_times, (ssp1_eis_r2 + ssp1_ra_r2 + ssp1_prtp_r2) * ssp1_tot**(-1), linestyle='dashed', linewidth=lw, \n",
    "        color=color_list[l3cind])\n",
    "ax['E'].fill_between(node_times, (ssp1_eis_r2 + ssp1_ra_r2) * ssp1_tot**(-1), \n",
    "                   (ssp1_ra_r2 + ssp1_eis_r2 + ssp1_prtp_r2) * ssp1_tot**(-1), alpha=alpha, \n",
    "                    color=color_list[l3cind], label=\"PRTP\")\n",
    "\n",
    "ax['E'].plot(node_times, (ssp1_eis_r2 + ssp1_ra_r2 + ssp1_prtp_r2 + ssp1_tech_r2) * ssp1_tot**(-1), linestyle='dashed', linewidth=lw, \n",
    "        color=color_list[l4cind])\n",
    "ax['E'].fill_between(node_times, (ssp1_eis_r2 + ssp1_ra_r2 + ssp1_prtp_r2) * ssp1_tot**(-1), \n",
    "                   (ssp1_ra_r2 + ssp1_eis_r2 + ssp1_prtp_r2 + ssp1_tech_r2) * ssp1_tot**(-1), alpha=alpha, \n",
    "                color=color_list[l4cind], label=\"Tech. Growth\")\n",
    "\n",
    "\n",
    "################################\n",
    "# SSP5\n",
    "################################\n",
    "tot_ssp5 = ssp5_ra_r2_t[1:] + ssp5_eis_r2_t[1:] + ssp5_prtp_r2_t[1:] + ssp5_tech_r2_t[1:]\n",
    "ax['F'].plot(node_times[1:], ssp5_ra_r2_t[1:] * tot_ssp5**(-1), linestyle='dashed', linewidth=lw, \n",
    "        color=color_list[l1cind])\n",
    "ax['F'].fill_between(node_times[1:], np.zeros_like(node_times[1:]), ssp5_ra_r2_t[1:] * tot_ssp5**(-1), alpha=alpha, \n",
    "                color=color_list[l1cind], label=\"RA\")\n",
    "\n",
    "ax['F'].plot(node_times[1:], (ssp5_eis_r2_t[1:] + ssp5_ra_r2_t[1:])  * tot_ssp5**(-1), linestyle='dashed', linewidth=lw, \n",
    "        color=color_list[l2cind])\n",
    "ax['F'].fill_between(node_times[1:], ssp5_ra_r2_t[1:]  * tot_ssp5**(-1), \n",
    "                   (ssp5_ra_r2_t[1:] + ssp5_eis_r2_t[1:]) * tot_ssp5**(-1), alpha=alpha, \n",
    "                color=color_list[l2cind], label=\"EIS\")\n",
    "\n",
    "ax['F'].plot(node_times[1:], (ssp5_eis_r2_t[1:] + ssp5_ra_r2_t[1:] + ssp5_prtp_r2_t[1:])  * tot_ssp5**(-1), linestyle='dashed', linewidth=lw, \n",
    "        color=color_list[l3cind])\n",
    "ax['F'].fill_between(node_times[1:], (ssp5_eis_r2_t[1:] + ssp5_ra_r2_t[1:]) * tot_ssp5**(-1), \n",
    "                  (ssp5_ra_r2_t[1:] + ssp5_eis_r2_t[1:] + ssp5_prtp_r2_t[1:]) * tot_ssp5**(-1), alpha=alpha, \n",
    "                color=color_list[l3cind], label=\"PRTP\")\n",
    "\n",
    "ax['F'].plot(node_times[1:], (ssp5_eis_r2_t[1:] + ssp5_ra_r2_t[1:] + ssp5_prtp_r2_t[1:] + ssp5_tech_r2_t[1:]) * tot_ssp5**(-1), \n",
    "           linestyle='dashed', linewidth=lw, color=color_list[l4cind])\n",
    "ax['F'].fill_between(node_times[1:], (ssp5_eis_r2_t[1:] + ssp5_ra_r2_t[1:] + ssp5_prtp_r2_t[1:]) * tot_ssp5**(-1), \n",
    "                   (ssp5_ra_r2_t[1:] + ssp5_eis_r2_t[1:] + ssp5_prtp_r2_t[1:] + ssp5_tech_r2_t[1:]) * tot_ssp5**(-1), alpha=alpha, \n",
    "                color=color_list[l4cind], label=\"Tech. Growth\")\n",
    "\n",
    "\n",
    "################################\n",
    "# SSP4\n",
    "################################\n",
    "tot_ssp4 = ssp4_ra_r2_t[1:] + ssp4_eis_r2_t[1:] + ssp4_prtp_r2_t[1:] + ssp4_tech_r2_t[1:]\n",
    "ax['G'].plot(node_times[1:], ssp4_ra_r2_t[1:] * tot_ssp4**(-1), linestyle='dashed', linewidth=lw, \n",
    "        color=color_list[l1cind])\n",
    "ax['G'].fill_between(node_times[1:], np.zeros_like(node_times[1:]), ssp4_ra_r2_t[1:] * tot_ssp4**(-1), alpha=alpha, \n",
    "                color=color_list[l1cind], label=\"RA\")\n",
    "\n",
    "ax['G'].plot(node_times[1:], (ssp4_eis_r2_t[1:] + ssp4_ra_r2_t[1:]) * tot_ssp4**(-1), linestyle='dashed', linewidth=lw, \n",
    "        color=color_list[l2cind])\n",
    "ax['G'].fill_between(node_times[1:], ssp4_ra_r2_t[1:] * tot_ssp4**(-1), \n",
    "                   (ssp4_ra_r2_t[1:] + ssp4_eis_r2_t[1:]) * tot_ssp4**(-1), alpha=alpha, \n",
    "                color=color_list[l2cind], label=\"EIS\")\n",
    "\n",
    "ax['G'].plot(node_times[1:], (ssp4_eis_r2_t[1:] + ssp4_ra_r2_t[1:] + ssp4_prtp_r2_t[1:]) * tot_ssp4**(-1), linestyle='dashed', linewidth=lw, \n",
    "        color=color_list[l3cind])\n",
    "ax['G'].fill_between(node_times[1:], (ssp4_eis_r2_t[1:] + ssp4_ra_r2_t[1:]) * tot_ssp4**(-1), \n",
    "                  (ssp4_ra_r2_t[1:] + ssp4_eis_r2_t[1:] + ssp4_prtp_r2_t[1:]) * tot_ssp4**(-1), alpha=alpha, \n",
    "                color=color_list[l3cind], label=\"PRTP\")\n",
    "\n",
    "ax['G'].plot(node_times[1:], (ssp4_eis_r2_t[1:] + ssp4_ra_r2_t[1:] + ssp4_prtp_r2_t[1:] + ssp4_tech_r2_t[1:]) * tot_ssp4**(-1), \n",
    "           linestyle='dashed', linewidth=lw, color=color_list[l4cind])\n",
    "ax['G'].fill_between(node_times[1:], (ssp4_eis_r2_t[1:] + ssp4_ra_r2_t[1:] + ssp4_prtp_r2_t[1:]) * tot_ssp4**(-1), \n",
    "                   (ssp4_ra_r2_t[1:] + ssp4_eis_r2_t[1:] + ssp4_prtp_r2_t[1:] + ssp4_tech_r2_t[1:]) * tot_ssp4**(-1), alpha=alpha, \n",
    "                color=color_list[l4cind], label=\"Tech. Growth\")\n",
    "\n",
    "################################\n",
    "# SSP3\n",
    "################################\n",
    "ssp3_tot = ssp3_ra_r2_t[1:] + ssp3_eis_r2_t[1:] + ssp3_prtp_r2_t[1:] + ssp3_tech_r2_t[1:]\n",
    "ax['H'].plot(node_times[1:], ssp3_ra_r2_t[1:] * ssp3_tot**(-1), linestyle='dashed', linewidth=lw, \n",
    "        color=color_list[l1cind])\n",
    "ax['H'].fill_between(node_times[1:], np.zeros_like(node_times[1:]), ssp3_ra_r2_t[1:] * ssp3_tot**(-1), alpha=alpha, \n",
    "                color=color_list[l1cind], label=\"RA\")\n",
    "\n",
    "ax['H'].plot(node_times[1:], (ssp3_eis_r2_t[1:] + ssp3_ra_r2_t[1:]) * ssp3_tot**(-1), linestyle='dashed', linewidth=lw, \n",
    "        color=color_list[l2cind])\n",
    "ax['H'].fill_between(node_times[1:], ssp3_ra_r2_t[1:] * ssp3_tot**(-1), \n",
    "                   (ssp3_ra_r2_t[1:] + ssp3_eis_r2_t[1:]) * ssp3_tot**(-1), alpha=alpha, \n",
    "                color=color_list[l2cind], label=\"EIS\")\n",
    "\n",
    "ax['H'].plot(node_times[1:], (ssp3_eis_r2_t[1:] + ssp3_ra_r2_t[1:] + ssp3_prtp_r2_t[1:]) * ssp3_tot**(-1), linestyle='dashed', linewidth=lw, \n",
    "        color=color_list[l3cind])\n",
    "ax['H'].fill_between(node_times[1:], (ssp3_eis_r2_t[1:] + ssp3_ra_r2_t[1:]) * ssp3_tot**(-1), \n",
    "                   (ssp3_ra_r2_t[1:] + ssp3_eis_r2_t[1:] + ssp3_prtp_r2_t[1:]) * ssp3_tot**(-1), alpha=alpha, \n",
    "                color=color_list[l3cind], label=\"PRTP\")\n",
    "\n",
    "ax['H'].plot(node_times[1:], (ssp3_eis_r2_t[1:] + ssp3_ra_r2_t[1:] + ssp3_prtp_r2_t[1:] + ssp3_tech_r2_t[1:]) * ssp3_tot**(-1), linestyle='dashed', linewidth=lw, \n",
    "        color=color_list[l4cind])\n",
    "ax['H'].fill_between(node_times[1:], (ssp3_eis_r2_t[1:] + ssp3_ra_r2_t[1:] + ssp3_prtp_r2_t[1:]) * ssp3_tot**(-1), \n",
    "                   (ssp3_ra_r2_t[1:] + ssp3_eis_r2_t[1:] + ssp3_prtp_r2_t[1:] + ssp3_tech_r2_t[1:]) * ssp3_tot**(-1), alpha=alpha, \n",
    "                color=color_list[l4cind], label=\"Tech. Growth\")\n",
    "\n",
    "################################\n",
    "# SSP2\n",
    "################################\n",
    "ssp2_tot = ssp2_ra_r2_t[1:] + ssp2_eis_r2_t[1:] + ssp2_prtp_r2_t[1:] + ssp2_tech_r2_t[1:]\n",
    "ax['I'].plot(node_times[1:], ssp2_ra_r2_t[1:] * ssp2_tot**(-1), linestyle='dashed', linewidth=lw, \n",
    "        color=color_list[l1cind])\n",
    "ax['I'].fill_between(node_times[1:], np.zeros_like(node_times[1:]), ssp2_ra_r2_t[1:]* ssp2_tot**(-1), alpha=alpha, \n",
    "                color=color_list[l1cind], label=\"RA\")\n",
    "\n",
    "ax['I'].plot(node_times[1:], (ssp2_eis_r2_t[1:] + ssp2_ra_r2_t[1:])* ssp2_tot**(-1), linestyle='dashed', linewidth=lw, \n",
    "        color=color_list[l2cind])\n",
    "ax['I'].fill_between(node_times[1:], ssp2_ra_r2_t[1:] * ssp2_tot**(-1), \n",
    "                   (ssp2_ra_r2_t[1:] + ssp2_eis_r2_t[1:]) * ssp2_tot**(-1), alpha=alpha, \n",
    "                color=color_list[l2cind], label=\"EIS\")\n",
    "\n",
    "ax['I'].plot(node_times[1:], (ssp2_eis_r2_t[1:] + ssp2_ra_r2_t[1:] + ssp2_prtp_r2_t[1:]) * ssp2_tot**(-1), linestyle='dashed', linewidth=lw, \n",
    "        color=color_list[l3cind])\n",
    "ax['I'].fill_between(node_times[1:], (ssp2_eis_r2_t[1:] + ssp2_ra_r2_t[1:]) * ssp2_tot**(-1), \n",
    "                   (ssp2_ra_r2_t[1:] + ssp2_eis_r2_t[1:] + ssp2_prtp_r2_t[1:]) * ssp2_tot**(-1), alpha=alpha, \n",
    "                color=color_list[l3cind], label=\"PRTP\")\n",
    "\n",
    "ax['I'].plot(node_times[1:], (ssp2_eis_r2_t[1:] + ssp2_ra_r2_t[1:] + ssp2_prtp_r2_t[1:] + ssp2_tech_r2_t[1:]) * ssp2_tot**(-1), linestyle='dashed', linewidth=lw, \n",
    "        color=color_list[l4cind])\n",
    "ax['I'].fill_between(node_times[1:], (ssp2_eis_r2_t[1:] + ssp2_ra_r2_t[1:] + ssp2_prtp_r2_t[1:]) * ssp2_tot**(-1), \n",
    "                   (ssp2_ra_r2_t[1:] + ssp2_eis_r2_t[1:] + ssp2_prtp_r2_t[1:] + ssp2_tech_r2_t[1:]) * ssp2_tot**(-1), alpha=alpha, \n",
    "                color=color_list[l4cind], label=\"Tech. Growth\")\n",
    "\n",
    "\n",
    "################################\n",
    "# SSP1\n",
    "################################\n",
    "ssp1_tot = ssp1_ra_r2_t[1:] + ssp1_eis_r2_t[1:] + ssp1_prtp_r2_t[1:] + ssp1_tech_r2_t[1:]\n",
    "ax['J'].plot(node_times[1:], ssp1_ra_r2_t[1:] * ssp1_tot**(-1), linestyle='dashed', linewidth=lw, \n",
    "        color=color_list[l1cind])\n",
    "ax['J'].fill_between(node_times[1:], np.zeros_like(node_times[1:]), ssp5_ra_r2_t[1:] * ssp1_tot**(-1), alpha=alpha, \n",
    "                color=color_list[l1cind], label=\"RA\")\n",
    "\n",
    "ax['J'].plot(node_times[1:], (ssp1_eis_r2_t[1:] + ssp1_ra_r2_t[1:]) * ssp1_tot**(-1), linestyle='dashed', linewidth=lw, \n",
    "        color=color_list[l2cind])\n",
    "ax['J'].fill_between(node_times[1:], ssp1_ra_r2_t[1:] * ssp1_tot**(-1), \n",
    "                   (ssp1_ra_r2_t[1:] + ssp1_eis_r2_t[1:]) * ssp1_tot**(-1), alpha=alpha, \n",
    "                    color=color_list[l2cind], label=\"EIS\")\n",
    "\n",
    "ax['J'].plot(node_times[1:], (ssp1_eis_r2_t[1:] + ssp1_ra_r2_t[1:] + ssp1_prtp_r2_t[1:]) * ssp1_tot**(-1), linestyle='dashed', linewidth=lw, \n",
    "        color=color_list[l3cind])\n",
    "ax['J'].fill_between(node_times[1:], (ssp1_eis_r2_t[1:] + ssp1_ra_r2_t[1:]) * ssp1_tot**(-1), \n",
    "                   (ssp1_ra_r2_t[1:] + ssp1_eis_r2_t[1:] + ssp1_prtp_r2_t[1:]) * ssp1_tot**(-1), alpha=alpha, \n",
    "                    color=color_list[l3cind], label=\"PRTP\")\n",
    "\n",
    "ax['J'].plot(node_times[1:], (ssp1_eis_r2_t[1:] + ssp1_ra_r2_t[1:] + ssp1_prtp_r2_t[1:] + ssp1_tech_r2_t[1:]) * ssp1_tot**(-1), linestyle='dashed', linewidth=lw, \n",
    "        color=color_list[l4cind])\n",
    "ax['J'].fill_between(node_times[1:], (ssp1_eis_r2_t[1:] + ssp1_ra_r2_t[1:] + ssp1_prtp_r2_t[1:]) * ssp1_tot**(-1), \n",
    "                   (ssp1_ra_r2_t[1:] + ssp1_eis_r2_t[1:] + ssp1_prtp_r2_t[1:] + ssp1_tech_r2_t[1:]) * ssp1_tot**(-1), alpha=alpha, \n",
    "                color=color_list[l4cind], label=\"Tech. Growth\")\n",
    "\n",
    "################################\n",
    "# SSP5\n",
    "################################\n",
    "tot_ssp5 = ssp5_ra_r2_co[1:] + ssp5_eis_r2_co[1:] + ssp5_prtp_r2_co[1:] + ssp5_tech_r2_co[1:]\n",
    "ax['K'].plot(node_times[1:], ssp5_ra_r2_co[1:] * tot_ssp5**(-1), linestyle='dashed', linewidth=lw, \n",
    "        color=color_list[l1cind])\n",
    "ax['K'].fill_between(node_times[1:], np.zeros_like(node_times[1:]), ssp5_ra_r2_co[1:] * tot_ssp5**(-1), alpha=alpha, \n",
    "                color=color_list[l1cind], label=\"RA\")\n",
    "\n",
    "ax['K'].plot(node_times[1:], (ssp5_eis_r2_co[1:] + ssp5_ra_r2_co[1:])  * tot_ssp5**(-1), linestyle='dashed', linewidth=lw, \n",
    "        color=color_list[l2cind])\n",
    "ax['K'].fill_between(node_times[1:], ssp5_ra_r2_co[1:]  * tot_ssp5**(-1), \n",
    "                   (ssp5_ra_r2_co[1:] + ssp5_eis_r2_co[1:]) * tot_ssp5**(-1), alpha=alpha, \n",
    "                color=color_list[l2cind], label=\"EIS\")\n",
    "\n",
    "ax['K'].plot(node_times[1:], (ssp5_eis_r2_co[1:] + ssp5_ra_r2_co[1:] + ssp5_prtp_r2_co[1:])  * tot_ssp5**(-1), linestyle='dashed', linewidth=lw, \n",
    "        color=color_list[l3cind])\n",
    "ax['K'].fill_between(node_times[1:], (ssp5_eis_r2_co[1:] + ssp5_ra_r2_co[1:]) * tot_ssp5**(-1), \n",
    "                  (ssp5_ra_r2_co[1:] + ssp5_eis_r2_co[1:] + ssp5_prtp_r2_co[1:]) * tot_ssp5**(-1), alpha=alpha, \n",
    "                color=color_list[l3cind], label=\"PRTP\")\n",
    "\n",
    "ax['K'].plot(node_times[1:], (ssp5_eis_r2_co[1:] + ssp5_ra_r2_co[1:] + ssp5_prtp_r2_co[1:] + ssp5_tech_r2_co[1:]) * tot_ssp5**(-1), \n",
    "           linestyle='dashed', linewidth=lw, color=color_list[l4cind])\n",
    "ax['K'].fill_between(node_times[1:], (ssp5_eis_r2_co[1:] + ssp5_ra_r2_co[1:] + ssp5_prtp_r2_co[1:]) * tot_ssp5**(-1), \n",
    "                   (ssp5_ra_r2_co[1:] + ssp5_eis_r2_co[1:] + ssp5_prtp_r2_co[1:] + ssp5_tech_r2_co[1:]) * tot_ssp5**(-1), alpha=alpha, \n",
    "                color=color_list[l4cind], label=\"Tech. Growth\")\n",
    "\n",
    "\n",
    "################################\n",
    "# SSP4\n",
    "################################\n",
    "tot_ssp4 = ssp4_ra_r2_co[1:] + ssp4_eis_r2_co[1:] + ssp4_prtp_r2_co[1:] + ssp4_tech_r2_co[1:]\n",
    "ax['L'].plot(node_times[1:], ssp4_ra_r2_co[1:] * tot_ssp4**(-1), linestyle='dashed', linewidth=lw, \n",
    "        color=color_list[l1cind])\n",
    "ax['L'].fill_between(node_times[1:], np.zeros_like(node_times[1:]), ssp4_ra_r2_co[1:] * tot_ssp4**(-1), alpha=alpha, \n",
    "                color=color_list[l1cind], label=\"RA\")\n",
    "\n",
    "ax['L'].plot(node_times[1:], (ssp4_eis_r2_co[1:] + ssp4_ra_r2_co[1:]) * tot_ssp4**(-1), linestyle='dashed', linewidth=lw, \n",
    "        color=color_list[l2cind])\n",
    "ax['L'].fill_between(node_times[1:], ssp4_ra_r2_co[1:] * tot_ssp4**(-1), \n",
    "                   (ssp4_ra_r2_co[1:] + ssp4_eis_r2_co[1:]) * tot_ssp4**(-1), alpha=alpha, \n",
    "                color=color_list[l2cind], label=\"EIS\")\n",
    "\n",
    "ax['L'].plot(node_times[1:], (ssp4_eis_r2_co[1:] + ssp4_ra_r2_co[1:] + ssp4_prtp_r2_co[1:]) * tot_ssp4**(-1), linestyle='dashed', linewidth=lw, \n",
    "        color=color_list[l3cind])\n",
    "ax['L'].fill_between(node_times[1:], (ssp4_eis_r2_co[1:] + ssp4_ra_r2_co[1:]) * tot_ssp4**(-1), \n",
    "                  (ssp4_ra_r2_co[1:] + ssp4_eis_r2_co[1:] + ssp4_prtp_r2_co[1:]) * tot_ssp4**(-1), alpha=alpha, \n",
    "                color=color_list[l3cind], label=\"PRTP\")\n",
    "\n",
    "ax['L'].plot(node_times[1:], (ssp4_eis_r2_co[1:] + ssp4_ra_r2_co[1:] + ssp4_prtp_r2_co[1:] + ssp4_tech_r2_co[1:]) * tot_ssp4**(-1), \n",
    "           linestyle='dashed', linewidth=lw, color=color_list[l4cind])\n",
    "ax['L'].fill_between(node_times[1:], (ssp4_eis_r2_co[1:] + ssp4_ra_r2_co[1:] + ssp4_prtp_r2_co[1:]) * tot_ssp4**(-1), \n",
    "                   (ssp4_ra_r2_co[1:] + ssp4_eis_r2_co[1:] + ssp4_prtp_r2_co[1:] + ssp4_tech_r2_co[1:]) * tot_ssp4**(-1), alpha=alpha, \n",
    "                color=color_list[l4cind], label=\"Tech. Growth\")\n",
    "\n",
    "################################\n",
    "# SSP3\n",
    "################################\n",
    "ssp3_tot = ssp3_ra_r2_co[1:] + ssp3_eis_r2_co[1:] + ssp3_prtp_r2_co[1:] + ssp3_tech_r2_co[1:]\n",
    "ax['M'].plot(node_times[1:], ssp3_ra_r2_co[1:] * ssp3_tot**(-1), linestyle='dashed', linewidth=lw, \n",
    "        color=color_list[l1cind])\n",
    "ax['M'].fill_between(node_times[1:], np.zeros_like(node_times[1:]), ssp3_ra_r2_co[1:] * ssp3_tot**(-1), alpha=alpha, \n",
    "                color=color_list[l1cind], label=\"RA\")\n",
    "\n",
    "ax['M'].plot(node_times[1:], (ssp3_eis_r2_co[1:] + ssp3_ra_r2_co[1:]) * ssp3_tot**(-1), linestyle='dashed', linewidth=lw, \n",
    "        color=color_list[l2cind])\n",
    "ax['M'].fill_between(node_times[1:], ssp3_ra_r2_co[1:] * ssp3_tot**(-1), \n",
    "                   (ssp3_ra_r2_co[1:] + ssp3_eis_r2_co[1:]) * ssp3_tot**(-1), alpha=alpha, \n",
    "                color=color_list[l2cind], label=\"EIS\")\n",
    "\n",
    "ax['M'].plot(node_times[1:], (ssp3_eis_r2_co[1:] + ssp3_ra_r2_co[1:] + ssp3_prtp_r2_co[1:]) * ssp3_tot**(-1), linestyle='dashed', linewidth=lw, \n",
    "        color=color_list[l3cind])\n",
    "ax['M'].fill_between(node_times[1:], (ssp3_eis_r2_co[1:] + ssp3_ra_r2_co[1:]) * ssp3_tot**(-1), \n",
    "                   (ssp3_ra_r2_co[1:] + ssp3_eis_r2_co[1:] + ssp3_prtp_r2_co[1:]) * ssp3_tot**(-1), alpha=alpha, \n",
    "                color=color_list[l3cind], label=\"PRTP\")\n",
    "\n",
    "ax['M'].plot(node_times[1:], (ssp3_eis_r2_co[1:] + ssp3_ra_r2_co[1:] + ssp3_prtp_r2_co[1:] + ssp3_tech_r2_co[1:]) * ssp3_tot**(-1), linestyle='dashed', linewidth=lw, \n",
    "        color=color_list[l4cind])\n",
    "ax['M'].fill_between(node_times[1:], (ssp3_eis_r2_co[1:] + ssp3_ra_r2_co[1:] + ssp3_prtp_r2_co[1:]) * ssp3_tot**(-1), \n",
    "                   (ssp3_ra_r2_co[1:] + ssp3_eis_r2_co[1:] + ssp3_prtp_r2_co[1:] + ssp3_tech_r2_co[1:]) * ssp3_tot**(-1), alpha=alpha, \n",
    "                color=color_list[l4cind], label=\"Tech. Growth\")\n",
    "\n",
    "################################\n",
    "# SSP2\n",
    "################################\n",
    "ssp2_tot = ssp2_ra_r2_co[1:] + ssp2_eis_r2_co[1:] + ssp2_prtp_r2_co[1:] + ssp2_tech_r2_co[1:]\n",
    "ax['N'].plot(node_times[1:], ssp2_ra_r2_co[1:] * ssp2_tot**(-1), linestyle='dashed', linewidth=lw, \n",
    "        color=color_list[l1cind])\n",
    "ax['N'].fill_between(node_times[1:], np.zeros_like(node_times[1:]), ssp2_ra_r2_co[1:]* ssp2_tot**(-1), alpha=alpha, \n",
    "                color=color_list[l1cind], label=\"RA\")\n",
    "\n",
    "ax['N'].plot(node_times[1:], (ssp2_eis_r2_co[1:] + ssp2_ra_r2_co[1:])* ssp2_tot**(-1), linestyle='dashed', linewidth=lw, \n",
    "        color=color_list[l2cind])\n",
    "ax['N'].fill_between(node_times[1:], ssp2_ra_r2_co[1:] * ssp2_tot**(-1), \n",
    "                   (ssp2_ra_r2_co[1:] + ssp2_eis_r2_co[1:]) * ssp2_tot**(-1), alpha=alpha, \n",
    "                color=color_list[l2cind], label=\"EIS\")\n",
    "\n",
    "ax['N'].plot(node_times[1:], (ssp2_eis_r2_co[1:] + ssp2_ra_r2_co[1:] + ssp2_prtp_r2_co[1:]) * ssp2_tot**(-1), linestyle='dashed', linewidth=lw, \n",
    "        color=color_list[l3cind])\n",
    "ax['N'].fill_between(node_times[1:], (ssp2_eis_r2_co[1:] + ssp2_ra_r2_co[1:]) * ssp2_tot**(-1), \n",
    "                   (ssp2_ra_r2_co[1:] + ssp2_eis_r2_co[1:] + ssp2_prtp_r2_co[1:]) * ssp2_tot**(-1), alpha=alpha, \n",
    "                color=color_list[l3cind], label=\"PRTP\")\n",
    "\n",
    "ax['N'].plot(node_times[1:], (ssp2_eis_r2_co[1:] + ssp2_ra_r2_co[1:] + ssp2_prtp_r2_co[1:] + ssp2_tech_r2_co[1:]) * ssp2_tot**(-1), linestyle='dashed', linewidth=lw, \n",
    "        color=color_list[l4cind])\n",
    "ax['N'].fill_between(node_times[1:], (ssp2_eis_r2_co[1:] + ssp2_ra_r2_co[1:] + ssp2_prtp_r2_co[1:]) * ssp2_tot**(-1), \n",
    "                   (ssp2_ra_r2_co[1:] + ssp2_eis_r2_co[1:] + ssp2_prtp_r2_co[1:] + ssp2_tech_r2_co[1:]) * ssp2_tot**(-1), alpha=alpha, \n",
    "                color=color_list[l4cind], label=\"Tech. Growth\")\n",
    "\n",
    "\n",
    "################################\n",
    "# SSP1\n",
    "################################\n",
    "ssp1_tot = ssp1_ra_r2_co[1:] + ssp1_eis_r2_co[1:] + ssp1_prtp_r2_co[1:] + ssp1_tech_r2_co[1:]\n",
    "ax['O'].plot(node_times[1:], ssp1_ra_r2_co[1:] * ssp1_tot**(-1), linestyle='dashed', linewidth=lw, \n",
    "        color=color_list[l1cind])\n",
    "ax['O'].fill_between(node_times[1:], np.zeros_like(node_times[1:]), ssp5_ra_r2_co[1:] * ssp1_tot**(-1), alpha=alpha, \n",
    "                color=color_list[l1cind], label=\"RA\")\n",
    "\n",
    "ax['O'].plot(node_times[1:], (ssp1_eis_r2_co[1:] + ssp1_ra_r2_co[1:]) * ssp1_tot**(-1), linestyle='dashed', linewidth=lw, \n",
    "        color=color_list[l2cind])\n",
    "ax['O'].fill_between(node_times[1:], ssp1_ra_r2_co[1:] * ssp1_tot**(-1), \n",
    "                   (ssp1_ra_r2_co[1:] + ssp1_eis_r2_co[1:]) * ssp1_tot**(-1), alpha=alpha, \n",
    "                    color=color_list[l2cind], label=\"EIS\")\n",
    "\n",
    "ax['O'].plot(node_times[1:], (ssp1_eis_r2_co[1:] + ssp1_ra_r2_co[1:] + ssp1_prtp_r2_co[1:]) * ssp1_tot**(-1), linestyle='dashed', linewidth=lw, \n",
    "        color=color_list[l3cind])\n",
    "ax['O'].fill_between(node_times[1:], (ssp1_eis_r2_co[1:] + ssp1_ra_r2_co[1:]) * ssp1_tot**(-1), \n",
    "                   (ssp1_ra_r2_co[1:] + ssp1_eis_r2_co[1:] + ssp1_prtp_r2_co[1:]) * ssp1_tot**(-1), alpha=alpha, \n",
    "                    color=color_list[l3cind], label=\"PRTP\")\n",
    "\n",
    "ax['O'].plot(node_times[1:], (ssp1_eis_r2_co[1:] + ssp1_ra_r2_co[1:] + ssp1_prtp_r2_co[1:] + ssp1_tech_r2_co[1:]) * ssp1_tot**(-1), linestyle='dashed', linewidth=lw, \n",
    "        color=color_list[l4cind])\n",
    "ax['O'].fill_between(node_times[1:], (ssp1_eis_r2_co[1:] + ssp1_ra_r2_co[1:] + ssp1_prtp_r2_co[1:]) * ssp1_tot**(-1), \n",
    "                   (ssp1_ra_r2_co[1:] + ssp1_eis_r2_co[1:] + ssp1_prtp_r2_co[1:] + ssp1_tech_r2_co[1:]) * ssp1_tot**(-1), alpha=alpha, \n",
    "                color=color_list[l4cind], label=\"Tech. Growth\")\n",
    "\n",
    "################################\n",
    "# SSP5\n",
    "################################\n",
    "tot_ssp5 = ssp5_ra_r2_d[1:] + ssp5_eis_r2_d[1:] + ssp5_prtp_r2_d[1:] + ssp5_tech_r2_d[1:]\n",
    "ax['P'].plot(node_times[1:], ssp5_ra_r2_d[1:] * tot_ssp5**(-1), linestyle='dashed', linewidth=lw, \n",
    "        color=color_list[l1cind])\n",
    "ax['P'].fill_between(node_times[1:], np.zeros_like(node_times[1:]), ssp5_ra_r2_d[1:] * tot_ssp5**(-1), alpha=alpha, \n",
    "                color=color_list[l1cind], label=\"RA\")\n",
    "\n",
    "ax['P'].plot(node_times[1:], (ssp5_eis_r2_d[1:] + ssp5_ra_r2_d[1:])  * tot_ssp5**(-1), linestyle='dashed', linewidth=lw, \n",
    "        color=color_list[l2cind])\n",
    "ax['P'].fill_between(node_times[1:], ssp5_ra_r2_co[1:]  * tot_ssp5**(-1), \n",
    "                   (ssp5_ra_r2_d[1:] + ssp5_eis_r2_d[1:]) * tot_ssp5**(-1), alpha=alpha, \n",
    "                color=color_list[l2cind], label=\"EIS\")\n",
    "\n",
    "ax['P'].plot(node_times[1:], (ssp5_eis_r2_d[1:] + ssp5_ra_r2_d[1:] + ssp5_prtp_r2_d[1:])  * tot_ssp5**(-1), linestyle='dashed', linewidth=lw, \n",
    "        color=color_list[l3cind])\n",
    "ax['P'].fill_between(node_times[1:], (ssp5_eis_r2_d[1:] + ssp5_ra_r2_d[1:]) * tot_ssp5**(-1), \n",
    "                  (ssp5_ra_r2_d[1:] + ssp5_eis_r2_d[1:] + ssp5_prtp_r2_d[1:]) * tot_ssp5**(-1), alpha=alpha, \n",
    "                color=color_list[l3cind], label=\"PRTP\")\n",
    "\n",
    "ax['P'].plot(node_times[1:], (ssp5_eis_r2_d[1:] + ssp5_ra_r2_d[1:] + ssp5_prtp_r2_d[1:] + ssp5_tech_r2_d[1:]) * tot_ssp5**(-1), \n",
    "           linestyle='dashed', linewidth=lw, color=color_list[l4cind])\n",
    "ax['P'].fill_between(node_times[1:], (ssp5_eis_r2_d[1:] + ssp5_ra_r2_d[1:] + ssp5_prtp_r2_d[1:]) * tot_ssp5**(-1), \n",
    "                   (ssp5_ra_r2_d[1:] + ssp5_eis_r2_d[1:] + ssp5_prtp_r2_d[1:] + ssp5_tech_r2_d[1:]) * tot_ssp5**(-1), alpha=alpha, \n",
    "                color=color_list[l4cind], label=\"Tech. Growth\")\n",
    "\n",
    "\n",
    "################################\n",
    "# SSP4\n",
    "################################\n",
    "tot_ssp4 = ssp4_ra_r2_d[1:] + ssp4_eis_r2_d[1:] + ssp4_prtp_r2_d[1:] + ssp4_tech_r2_d[1:]\n",
    "ax['Q'].plot(node_times[1:], ssp4_ra_r2_d[1:] * tot_ssp4**(-1), linestyle='dashed', linewidth=lw, \n",
    "        color=color_list[l1cind])\n",
    "ax['Q'].fill_between(node_times[1:], np.zeros_like(node_times[1:]), ssp4_ra_r2_d[1:] * tot_ssp4**(-1), alpha=alpha, \n",
    "                color=color_list[l1cind], label=\"RA\")\n",
    "\n",
    "ax['Q'].plot(node_times[1:], (ssp4_eis_r2_d[1:] + ssp4_ra_r2_d[1:]) * tot_ssp4**(-1), linestyle='dashed', linewidth=lw, \n",
    "        color=color_list[l2cind])\n",
    "ax['Q'].fill_between(node_times[1:], ssp4_ra_r2_d[1:] * tot_ssp4**(-1), \n",
    "                   (ssp4_ra_r2_d[1:] + ssp4_eis_r2_d[1:]) * tot_ssp4**(-1), alpha=alpha, \n",
    "                color=color_list[l2cind], label=\"EIS\")\n",
    "\n",
    "ax['Q'].plot(node_times[1:], (ssp4_eis_r2_d[1:] + ssp4_ra_r2_d[1:] + ssp4_prtp_r2_d[1:]) * tot_ssp4**(-1), linestyle='dashed', linewidth=lw, \n",
    "        color=color_list[l3cind])\n",
    "ax['Q'].fill_between(node_times[1:], (ssp4_eis_r2_d[1:] + ssp4_ra_r2_d[1:]) * tot_ssp4**(-1), \n",
    "                  (ssp4_ra_r2_d[1:] + ssp4_eis_r2_d[1:] + ssp4_prtp_r2_d[1:]) * tot_ssp4**(-1), alpha=alpha, \n",
    "                color=color_list[l3cind], label=\"PRTP\")\n",
    "\n",
    "ax['Q'].plot(node_times[1:], (ssp4_eis_r2_d[1:] + ssp4_ra_r2_d[1:] + ssp4_prtp_r2_d[1:] + ssp4_tech_r2_d[1:]) * tot_ssp4**(-1), \n",
    "           linestyle='dashed', linewidth=lw, color=color_list[l4cind])\n",
    "ax['Q'].fill_between(node_times[1:], (ssp4_eis_r2_d[1:] + ssp4_ra_r2_d[1:] + ssp4_prtp_r2_d[1:]) * tot_ssp4**(-1), \n",
    "                   (ssp4_ra_r2_d[1:] + ssp4_eis_r2_d[1:] + ssp4_prtp_r2_d[1:] + ssp4_tech_r2_d[1:]) * tot_ssp4**(-1), alpha=alpha, \n",
    "                color=color_list[l4cind], label=\"Tech. Growth\")\n",
    "\n",
    "################################\n",
    "# SSP3\n",
    "################################\n",
    "ssp3_tot = ssp3_ra_r2_d[1:] + ssp3_eis_r2_d[1:] + ssp3_prtp_r2_d[1:] + ssp3_tech_r2_d[1:]\n",
    "ax['R'].plot(node_times[1:], ssp3_ra_r2_d[1:] * ssp3_tot**(-1), linestyle='dashed', linewidth=lw, \n",
    "        color=color_list[l1cind])\n",
    "ax['R'].fill_between(node_times[1:], np.zeros_like(node_times[1:]), ssp3_ra_r2_d[1:] * ssp3_tot**(-1), alpha=alpha, \n",
    "                color=color_list[l1cind], label=\"RA\")\n",
    "\n",
    "ax['R'].plot(node_times[1:], (ssp3_eis_r2_d[1:] + ssp3_ra_r2_d[1:]) * ssp3_tot**(-1), linestyle='dashed', linewidth=lw, \n",
    "        color=color_list[l2cind])\n",
    "ax['R'].fill_between(node_times[1:], ssp3_ra_r2_d[1:] * ssp3_tot**(-1), \n",
    "                   (ssp3_ra_r2_d[1:] + ssp3_eis_r2_d[1:]) * ssp3_tot**(-1), alpha=alpha, \n",
    "                color=color_list[l2cind], label=\"EIS\")\n",
    "\n",
    "ax['R'].plot(node_times[1:], (ssp3_eis_r2_d[1:] + ssp3_ra_r2_d[1:] + ssp3_prtp_r2_d[1:]) * ssp3_tot**(-1), linestyle='dashed', linewidth=lw, \n",
    "        color=color_list[l3cind])\n",
    "ax['R'].fill_between(node_times[1:], (ssp3_eis_r2_d[1:] + ssp3_ra_r2_d[1:]) * ssp3_tot**(-1), \n",
    "                   (ssp3_ra_r2_d[1:] + ssp3_eis_r2_d[1:] + ssp3_prtp_r2_d[1:]) * ssp3_tot**(-1), alpha=alpha, \n",
    "                color=color_list[l3cind], label=\"PRTP\")\n",
    "\n",
    "ax['R'].plot(node_times[1:], (ssp3_eis_r2_d[1:] + ssp3_ra_r2_d[1:] + ssp3_prtp_r2_d[1:] + ssp3_tech_r2_d[1:]) * ssp3_tot**(-1), linestyle='dashed', linewidth=lw, \n",
    "        color=color_list[l4cind])\n",
    "ax['R'].fill_between(node_times[1:], (ssp3_eis_r2_d[1:] + ssp3_ra_r2_d[1:] + ssp3_prtp_r2_d[1:]) * ssp3_tot**(-1), \n",
    "                   (ssp3_ra_r2_d[1:] + ssp3_eis_r2_d[1:] + ssp3_prtp_r2_d[1:] + ssp3_tech_r2_d[1:]) * ssp3_tot**(-1), alpha=alpha, \n",
    "                color=color_list[l4cind], label=\"Tech. Growth\")\n",
    "\n",
    "################################\n",
    "# SSP2\n",
    "################################\n",
    "ssp2_tot = ssp2_ra_r2_d[1:] + ssp2_eis_r2_d[1:] + ssp2_prtp_r2_d[1:] + ssp2_tech_r2_d[1:]\n",
    "ax['S'].plot(node_times[1:], ssp2_ra_r2_d[1:] * ssp2_tot**(-1), linestyle='dashed', linewidth=lw, \n",
    "        color=color_list[l1cind])\n",
    "ax['S'].fill_between(node_times[1:], np.zeros_like(node_times[1:]), ssp2_ra_r2_d[1:]* ssp2_tot**(-1), alpha=alpha, \n",
    "                color=color_list[l1cind], label=\"RA\")\n",
    "\n",
    "ax['S'].plot(node_times[1:], (ssp2_eis_r2_d[1:] + ssp2_ra_r2_d[1:])* ssp2_tot**(-1), linestyle='dashed', linewidth=lw, \n",
    "        color=color_list[l2cind])\n",
    "ax['S'].fill_between(node_times[1:], ssp2_ra_r2_d[1:] * ssp2_tot**(-1), \n",
    "                   (ssp2_ra_r2_d[1:] + ssp2_eis_r2_d[1:]) * ssp2_tot**(-1), alpha=alpha, \n",
    "                color=color_list[l2cind], label=\"EIS\")\n",
    "\n",
    "ax['S'].plot(node_times[1:], (ssp2_eis_r2_d[1:] + ssp2_ra_r2_d[1:] + ssp2_prtp_r2_d[1:]) * ssp2_tot**(-1), linestyle='dashed', linewidth=lw, \n",
    "        color=color_list[l3cind])\n",
    "ax['S'].fill_between(node_times[1:], (ssp2_eis_r2_d[1:] + ssp2_ra_r2_d[1:]) * ssp2_tot**(-1), \n",
    "                   (ssp2_ra_r2_d[1:] + ssp2_eis_r2_d[1:] + ssp2_prtp_r2_d[1:]) * ssp2_tot**(-1), alpha=alpha, \n",
    "                color=color_list[l3cind], label=\"PRTP\")\n",
    "\n",
    "ax['S'].plot(node_times[1:], (ssp2_eis_r2_d[1:] + ssp2_ra_r2_d[1:] + ssp2_prtp_r2_d[1:] + ssp2_tech_r2_d[1:]) * ssp2_tot**(-1), linestyle='dashed', linewidth=lw, \n",
    "        color=color_list[l4cind])\n",
    "ax['S'].fill_between(node_times[1:], (ssp2_eis_r2_d[1:] + ssp2_ra_r2_d[1:] + ssp2_prtp_r2_d[1:]) * ssp2_tot**(-1), \n",
    "                   (ssp2_ra_r2_d[1:] + ssp2_eis_r2_d[1:] + ssp2_prtp_r2_d[1:] + ssp2_tech_r2_d[1:]) * ssp2_tot**(-1), alpha=alpha, \n",
    "                color=color_list[l4cind], label=\"Tech. Growth\")\n",
    "\n",
    "\n",
    "################################\n",
    "# SSP1\n",
    "################################\n",
    "ssp1_tot = ssp1_ra_r2_d[1:] + ssp1_eis_r2_d[1:] + ssp1_prtp_r2_d[1:] + ssp1_tech_r2_d[1:]\n",
    "ax['T'].plot(node_times[1:], ssp1_ra_r2_d[1:] * ssp1_tot**(-1), linestyle='dashed', linewidth=lw, \n",
    "        color=color_list[l1cind])\n",
    "ax['T'].fill_between(node_times[1:], np.zeros_like(node_times[1:]), ssp5_ra_r2_d[1:] * ssp1_tot**(-1), alpha=alpha, \n",
    "                color=color_list[l1cind], label=\"Epstein-Zin Risk Aversion, $\\psi$\")\n",
    "\n",
    "ax['T'].plot(node_times[1:], (ssp1_eis_r2_d[1:] + ssp1_ra_r2_d[1:]) * ssp1_tot**(-1), linestyle='dashed', linewidth=lw, \n",
    "        color=color_list[l2cind])\n",
    "ax['T'].fill_between(node_times[1:], ssp1_ra_r2_d[1:] * ssp1_tot**(-1), \n",
    "                   (ssp1_ra_r2_d[1:] + ssp1_eis_r2_d[1:]) * ssp1_tot**(-1), alpha=alpha, \n",
    "                    color=color_list[l2cind], label=\"Elasticity of Intertemporal Substitution, $\\sigma$\")\n",
    "\n",
    "ax['T'].plot(node_times[1:], (ssp1_eis_r2_d[1:] + ssp1_ra_r2_d[1:] + ssp1_prtp_r2_d[1:]) * ssp1_tot**(-1), linestyle='dashed', linewidth=lw, \n",
    "        color=color_list[l3cind])\n",
    "ax['T'].fill_between(node_times[1:], (ssp1_eis_r2_d[1:] + ssp1_ra_r2_d[1:]) * ssp1_tot**(-1), \n",
    "                   (ssp1_ra_r2_d[1:] + ssp1_eis_r2_d[1:] + ssp1_prtp_r2_d[1:]) * ssp1_tot**(-1), alpha=alpha, \n",
    "                    color=color_list[l3cind], label=\"Pure rate of time preference, $\\rho$\")\n",
    "\n",
    "ax['T'].plot(node_times[1:], (ssp1_eis_r2_d[1:] + ssp1_ra_r2_d[1:] + ssp1_prtp_r2_d[1:] + ssp1_tech_r2_d[1:]) * ssp1_tot**(-1), linestyle='dashed', linewidth=lw, \n",
    "        color=color_list[l4cind])\n",
    "ax['T'].fill_between(node_times[1:], (ssp1_eis_r2_d[1:] + ssp1_ra_r2_d[1:] + ssp1_prtp_r2_d[1:]) * ssp1_tot**(-1), \n",
    "                   (ssp1_ra_r2_d[1:] + ssp1_eis_r2_d[1:] + ssp1_prtp_r2_d[1:] + ssp1_tech_r2_d[1:]) * ssp1_tot**(-1), alpha=alpha, \n",
    "                color=color_list[l4cind], label=\"Technology Growth, $\\varphi = \\varphi_{0} + \\varphi_{1} X_{t}\")\n",
    "\n",
    "ax['R'].legend(loc='upper center', bbox_to_anchor=(0.5, -0.15), fancybox=True, shadow=True, ncol=5, fontsize=20)\n",
    "\n",
    "\n",
    "for i in cost_row:\n",
    "    ax[i].set_xlim((node_times[0], node_times[-1]))\n",
    "    ax[i].set_xticks([2020, 2100, 2200])\n",
    "    ax[i].set_xticklabels(['2020', '', ''])\n",
    "    ax[i].set_aspect(1.0/ax[i].get_data_ratio(), adjustable='box')\n",
    "    ax[i].set_ylim((0,1))\n",
    "    \n",
    "for i in t_row:\n",
    "    ax[i].set_xlim((node_times[1], node_times[-1]))\n",
    "    ax[i].set_xticks([2030, 2100, 2200])\n",
    "    ax[i].set_xticklabels([])\n",
    "    ax[i].set_aspect(1.0/ax[i].get_data_ratio(), adjustable='box')\n",
    "    ax[i].set_ylim((0,1))\n",
    "    \n",
    "for i in conc_row:\n",
    "    ax[i].set_xlim((node_times[1], node_times[-1]))\n",
    "    ax[i].set_xticks([2030, 2100, 2200])\n",
    "    ax[i].set_xticklabels([])\n",
    "    ax[i].set_aspect(1.0/ax[i].get_data_ratio(), adjustable='box')\n",
    "    ax[i].set_ylim((0,1))\n",
    "    \n",
    "for i in dam_row:\n",
    "    ax[i].set_xlim((node_times[1], node_times[-1]))\n",
    "    ax[i].set_xticks([2030, 2100, 2200])\n",
    "    ax[i].set_xticklabels(['2030', '2100', '2200'])\n",
    "    ax[i].set_aspect(1.0/ax[i].get_data_ratio(), adjustable='box')\n",
    "    \n",
    "for i in cost_row[1:] + t_row[1:] + conc_row[1:] + dam_row[1:]:\n",
    "    ax[i].set_yticklabels([])\n",
    "\n",
    "sns.despine(trim=True, offset=10)\n",
    "\n",
    "ax['A'].set_ylabel(\"$r^{2} / r_{tot}^{2}$\")\n",
    "ax['F'].set_ylabel(\"$r^{2} / r_{tot}^{2}$\")\n",
    "ax['K'].set_ylabel(\"$r^{2} / r_{tot}^{2}$\")\n",
    "ax['P'].set_ylabel(\"$r^{2} / r_{tot}^{2}$\")\n",
    "\n",
    "\n",
    "ax['A'].set_title(\"Cost              SSP5$-$8.5\")\n",
    "ax['B'].set_title(\"SSP4$-6.0$\")\n",
    "ax['C'].set_title(\"SSP3$-$7.0\")\n",
    "ax['D'].set_title(\"SSP2$-$4.5\")\n",
    "ax['E'].set_title(\"SSP1$-$1.9\")\n",
    "ax['F'].set_title(\"Temperature\")\n",
    "ax['K'].set_title(\"CO$_2$ concentrations\")\n",
    "ax['P'].set_title(\"Damages\")\n",
    "\n",
    "# label panels\n",
    "for label, ax in ax.items():\n",
    "    # label physical distance in and down:\n",
    "    trans = mtransforms.ScaledTranslation(2.9, -0.13, fig.dpi_scale_trans)\n",
    "    ax.text(0.0, 1.0, label, color='white', transform=ax.transAxes + trans, fontsize=22, fontweight='bold',\n",
    "            verticalalignment='top', bbox=dict(facecolor='none', edgecolor='none', pad=1))\n",
    "    ax.set_ylim((0,1))\n",
    "\n",
    "fig.subplots_adjust(wspace=0.2, hspace=0.3)\n",
    "\n",
    "if save_fig:\n",
    "    fig.savefig(''.join([basefile, 'rsquared-all.png']), dpi=400)"
   ]
  },
  {
   "cell_type": "markdown",
   "id": "monetary-species",
   "metadata": {},
   "source": [
    "# Old stuff\n",
    "\n",
    "Where we treated endogeneous and exogeneous as their own parameters, and didn't take into account agent behavior in the calculation of the technological growth rate."
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "abstract-continent",
   "metadata": {},
   "outputs": [],
   "source": [
    "fig, ax = plt.subplots(5, 6, figsize=(30, 20))\n",
    "\n",
    "ax[0,0].set_title(\"2020\")\n",
    "ax[0,1].set_title(\"2030\")\n",
    "ax[0,2].set_title(\"2060\")\n",
    "ax[0,3].set_title(\"2100\")\n",
    "ax[0,4].set_title(\"2150\")\n",
    "ax[0,5].set_title(\"2200\")\n",
    "\n",
    "###########################\n",
    "## RISK AVERSION ROW\n",
    "###########################\n",
    "make_panel_w_regression(ax, 0, 0, ssp5_ra, ssp4_ra, ssp3_ra, ssp2_ra, ssp1_ra, ssp5_c2020, ssp4_c2020, ssp3_c2020, ssp2_c2020, ssp1_c2020, \n",
    "           \"Risk Aversion\", \"Cost\\n(2020 $USD)\")\n",
    "\n",
    "make_panel_w_regression(ax, 0, 1, ssp5_ra, ssp4_ra, ssp3_ra, ssp2_ra, ssp1_ra, ssp5_c2030, ssp4_c2030, ssp3_c2030, ssp2_c2030, ssp1_c2030, \n",
    "           \"Risk Aversion\", None)\n",
    "\n",
    "make_panel_w_regression(ax, 0, 2, ssp5_ra, ssp4_ra, ssp3_ra, ssp2_ra, ssp1_ra, ssp5_c2060, ssp4_c2060, ssp3_c2060, ssp2_c2060, ssp1_c2060, \n",
    "           \"Risk Aversion\", None)\n",
    "\n",
    "make_panel_w_regression(ax, 0, 3, ssp5_ra, ssp4_ra, ssp3_ra, ssp2_ra, ssp1_ra, ssp5_c2100, ssp4_c2100, ssp3_c2100, ssp2_c2100, ssp1_c2100, \n",
    "           \"Risk Aversion\", None)\n",
    "\n",
    "make_panel_w_regression(ax, 0, 4, ssp5_ra, ssp4_ra, ssp3_ra, ssp2_ra, ssp1_ra, ssp5_c2150, ssp4_c2150, ssp3_c2150, ssp2_c2150, ssp1_c2150, \n",
    "           \"Risk Aversion\", None)\n",
    "\n",
    "make_panel_w_regression(ax, 0, 5, ssp5_ra, ssp4_ra, ssp3_ra, ssp2_ra, ssp1_ra, ssp5_c2200, ssp4_c2200, ssp3_c2200, ssp2_c2200, ssp1_c2200, \n",
    "           \"Risk Aversion\", None)\n",
    "\n",
    "###########################\n",
    "# EIS ROW\n",
    "###########################\n",
    "make_panel_w_regression(ax, 1, 0, ssp5_eis, ssp4_eis, ssp3_eis, ssp2_eis, ssp1_eis, ssp5_c2020, ssp4_c2020, ssp3_c2020, ssp2_c2020, ssp1_c2020, \n",
    "           \"EIS\", \"Cost\\n(2020 $USD)\")\n",
    "\n",
    "make_panel_w_regression(ax, 1, 1, ssp5_eis, ssp4_eis, ssp3_eis, ssp2_eis, ssp1_eis, ssp5_c2030, ssp4_c2030, ssp3_c2030, ssp2_c2030, ssp1_c2030, \n",
    "           \"EIS\", None)\n",
    "\n",
    "make_panel_w_regression(ax, 1, 2, ssp5_eis, ssp4_eis, ssp3_eis, ssp2_eis, ssp1_eis, ssp5_c2060, ssp4_c2060, ssp3_c2060, ssp2_c2060, ssp1_c2060, \n",
    "           \"EIS\", None)\n",
    "\n",
    "make_panel_w_regression(ax, 1, 3, ssp5_eis, ssp4_eis, ssp3_eis, ssp2_eis, ssp1_eis, ssp5_c2100, ssp4_c2100, ssp3_c2100, ssp2_c2100, ssp1_c2100, \n",
    "           \"EIS\", None)\n",
    "\n",
    "make_panel_w_regression(ax, 1, 4, ssp5_eis, ssp4_eis, ssp3_eis, ssp2_eis, ssp1_eis, ssp5_c2150, ssp4_c2150, ssp3_c2150, ssp2_c2150, ssp1_c2150, \n",
    "           \"EIS\", None)\n",
    "\n",
    "make_panel_w_regression(ax, 1, 5, ssp5_eis, ssp4_eis, ssp3_eis, ssp2_eis, ssp1_eis, ssp5_c2200, ssp4_c2200, ssp3_c2200, ssp2_c2200, ssp1_c2200, \n",
    "           \"EIS\", None)\n",
    "\n",
    "###########################\n",
    "# PRTP ROW\n",
    "###########################\n",
    "make_panel_w_regression(ax, 2, 0, ssp5_prtp * 100, ssp4_prtp * 100, ssp3_prtp * 100, ssp2_prtp * 100, ssp1_prtp * 100, ssp5_c2020, ssp4_c2020, ssp3_c2020, ssp2_c2020, ssp1_c2020, \n",
    "           r\"$\\delta$ (%)\", \"Cost\\n(2020 $USD)\")\n",
    "\n",
    "make_panel_w_regression(ax, 2, 1, ssp5_prtp * 100, ssp4_prtp * 100, ssp3_prtp * 100, ssp2_prtp * 100, ssp1_prtp * 100, ssp5_c2030, ssp4_c2030, ssp3_c2030, ssp2_c2030, ssp1_c2030, \n",
    "           r\"$\\delta$ (%)\", None)\n",
    "\n",
    "make_panel_w_regression(ax, 2, 2, ssp5_prtp * 100, ssp4_prtp * 100, ssp3_prtp * 100, ssp2_prtp * 100, ssp1_prtp * 100, ssp5_c2060, ssp4_c2060, ssp3_c2060, ssp2_c2060, ssp1_c2060, \n",
    "           r\"$\\delta$ (%)\", None)\n",
    "\n",
    "make_panel_w_regression(ax, 2, 3, ssp5_prtp * 100, ssp4_prtp * 100, ssp3_prtp * 100, ssp2_prtp * 100, ssp1_prtp * 100, ssp5_c2100, ssp4_c2100, ssp3_c2100, ssp2_c2100, ssp1_c2100, \n",
    "           r\"$\\delta$ (%)\", None)\n",
    "\n",
    "make_panel_w_regression(ax, 2, 4, ssp5_prtp * 100, ssp4_prtp * 100, ssp3_prtp * 100, ssp2_prtp * 100, ssp1_prtp * 100, ssp5_c2150, ssp4_c2150, ssp3_c2150, ssp2_c2150, ssp1_c2150, \n",
    "           r\"$\\delta$ (%)\", None)\n",
    "\n",
    "make_panel_w_regression(ax, 2, 5, ssp5_prtp * 100, ssp4_prtp * 100, ssp3_prtp * 100, ssp2_prtp * 100, ssp1_prtp * 100, ssp5_c2200, ssp4_c2200, ssp3_c2200, ssp2_c2200, ssp1_c2200, \n",
    "           r\"$\\delta$ (%)\", None)\n",
    "\n",
    "###########################\n",
    "# EX TECH GROWTH ROW\n",
    "###########################\n",
    "make_panel_w_regression(ax, 3, 0, ssp5_ex, ssp4_ex, ssp3_ex, ssp2_ex, ssp1_ex, ssp5_c2020, ssp4_c2020, ssp3_c2020, ssp2_c2020, ssp1_c2020, \n",
    "           \"Ex. Tech Growth (%)\", \"Cost\\n(2020 $USD)\")\n",
    "\n",
    "make_panel_w_regression(ax, 3, 1, ssp5_ex, ssp4_ex, ssp3_ex, ssp2_ex, ssp1_ex, ssp5_c2030, ssp4_c2030, ssp3_c2030, ssp2_c2030, ssp1_c2030, \n",
    "           \"Ex. Tech Growth (%)\", None)\n",
    "\n",
    "make_panel_w_regression(ax, 3, 2, ssp5_ex, ssp4_ex, ssp3_ex, ssp2_ex, ssp1_ex, ssp5_c2060, ssp4_c2060, ssp3_c2060, ssp2_c2060, ssp1_c2060, \n",
    "           \"Ex. Tech Growth (%)\", None)\n",
    "\n",
    "make_panel_w_regression(ax, 3, 3, ssp5_ex, ssp4_ex, ssp3_ex, ssp2_ex, ssp1_ex, ssp5_c2100, ssp4_c2100, ssp3_c2100, ssp2_c2100, ssp1_c2100, \n",
    "           \"Ex. Tech Growth (%)\", None)\n",
    "\n",
    "make_panel_w_regression(ax, 3, 4, ssp5_ex, ssp4_ex, ssp3_ex, ssp2_ex, ssp1_ex, ssp5_c2150, ssp4_c2150, ssp3_c2150, ssp2_c2150, ssp1_c2150, \n",
    "           \"Ex. Tech Growth (%)\", None)\n",
    "\n",
    "make_panel_w_regression(ax, 3, 5, ssp5_ex, ssp4_ex, ssp3_ex, ssp2_ex, ssp1_ex, ssp5_c2200, ssp4_c2200, ssp3_c2200, ssp2_c2200, ssp1_c2200, \n",
    "           \"Ex. Tech Growth (%)\", None)\n",
    "\n",
    "###########################\n",
    "# END TECH GROWTH ROW\n",
    "###########################\n",
    "make_panel_w_regression(ax, 4, 0, ssp5_end, ssp4_end, ssp3_end, ssp2_end, ssp1_end, ssp5_c2020, ssp4_c2020, ssp3_c2020, ssp2_c2020, ssp1_c2020, \n",
    "           \"End. Tech Growth (%)\", \"Cost\\n(2020 $USD)\")\n",
    "\n",
    "make_panel_w_regression(ax, 4, 1, ssp5_end, ssp4_end, ssp3_end, ssp2_end, ssp1_end, ssp5_c2030, ssp4_c2030, ssp3_c2030, ssp2_c2030, ssp1_c2030, \n",
    "           \"End. Tech Growth (%)\", None)\n",
    "\n",
    "make_panel_w_regression(ax, 4, 2, ssp5_end, ssp4_end, ssp3_end, ssp2_end, ssp1_end, ssp5_c2060, ssp4_c2060, ssp3_c2060, ssp2_c2060, ssp1_c2060, \n",
    "           \"End. Tech Growth (%)\", None)\n",
    "\n",
    "make_panel_w_regression(ax, 4, 3, ssp5_end, ssp4_end, ssp3_end, ssp2_end, ssp1_end, ssp5_c2100, ssp4_c2100, ssp3_c2100, ssp2_c2100, ssp1_c2100, \n",
    "           \"End. Tech Growth (%)\", None)\n",
    "\n",
    "make_panel_w_regression(ax, 4, 4, ssp5_end, ssp4_end, ssp3_end, ssp2_end, ssp1_end, ssp5_c2150, ssp4_c2150, ssp3_c2150, ssp2_c2150, ssp1_c2150, \n",
    "           \"End. Tech Growth (%)\", None)\n",
    "\n",
    "make_panel_w_regression(ax, 4, 5, ssp5_end, ssp4_end, ssp3_end, ssp2_end, ssp1_end, ssp5_c2200, ssp4_c2200, ssp3_c2200, ssp2_c2200, ssp1_c2200, \n",
    "           \"End. Tech Growth (%)\", None)\n",
    "\n",
    "fig.tight_layout()"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "therapeutic-atlanta",
   "metadata": {},
   "outputs": [],
   "source": [
    "ssp5_ex_r2 = get_r2_in_time(ssp5_ex, ssp5_paths)\n",
    "ssp5_end_r2 = get_r2_in_time(ssp5_end, ssp5_paths)\n",
    "\n",
    "ssp4_ex_r2 = get_r2_in_time(ssp4_ex, ssp4_paths)\n",
    "ssp4_end_r2 = get_r2_in_time(ssp4_end, ssp4_paths)\n",
    "\n",
    "ssp3_ex_r2 = get_r2_in_time(ssp3_ex, ssp3_paths)\n",
    "ssp3_end_r2 = get_r2_in_time(ssp3_end, ssp3_paths)\n",
    "\n",
    "ssp2_ex_r2 = get_r2_in_time(ssp2_ex, ssp2_paths)\n",
    "ssp2_end_r2 = get_r2_in_time(ssp2_end, ssp2_paths)\n",
    "\n",
    "ssp1_ex_r2 = get_r2_in_time(ssp1_ex, ssp1_paths)\n",
    "ssp1_end_r2 = get_r2_in_time(ssp1_end, ssp1_paths)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "moved-equality",
   "metadata": {},
   "outputs": [],
   "source": [
    "fig, ax = plt.subplots(1,5, figsize=(20,20), sharey=True)\n",
    "\n",
    "ax[0].set_ylim((0,1))\n",
    "\n",
    "alpha = 0.8\n",
    "lw = 3\n",
    "l1cind = 0\n",
    "l2cind = 1\n",
    "l3cind = 2\n",
    "l4cind = 3\n",
    "l5cind = 4\n",
    "\n",
    "################################\n",
    "# SSP5\n",
    "################################\n",
    "ax[0].plot(node_times, ssp5_ra_r2, linestyle='dashed', linewidth=lw, \n",
    "        color=color_list[l1cind])\n",
    "ax[0].fill_between(node_times, np.zeros_like(node_times), ssp5_ra_r2, alpha=alpha, \n",
    "                color=color_list[l1cind], label=\"RA\")\n",
    "\n",
    "ax[0].plot(node_times, ssp5_eis_r2 + ssp5_ra_r2, linestyle='dashed', linewidth=lw, \n",
    "        color=color_list[l2cind])\n",
    "ax[0].fill_between(node_times, ssp5_ra_r2, ssp5_ra_r2 + ssp5_eis_r2, alpha=alpha, \n",
    "                color=color_list[l2cind], label=\"EIS\")\n",
    "\n",
    "ax[0].plot(node_times, ssp5_eis_r2 + ssp5_ra_r2 + ssp5_prtp_r2, linestyle='dashed', linewidth=lw, \n",
    "        color=color_list[l3cind])\n",
    "ax[0].fill_between(node_times, ssp5_eis_r2 + ssp5_ra_r2, ssp5_ra_r2 + ssp5_eis_r2 + ssp5_prtp_r2, alpha=alpha, \n",
    "                color=color_list[l3cind], label=\"PRTP\")\n",
    "\n",
    "ax[0].plot(node_times, ssp5_eis_r2 + ssp5_ra_r2 + ssp5_prtp_r2 + ssp5_ex_r2, linestyle='dashed', linewidth=lw, \n",
    "        color=color_list[l4cind])\n",
    "ax[0].fill_between(node_times, ssp5_eis_r2 + ssp5_ra_r2 + ssp5_prtp_r2, ssp5_ra_r2 + ssp5_eis_r2 + ssp5_prtp_r2 + ssp5_ex_r2, alpha=alpha, \n",
    "                color=color_list[l4cind], label=\"Ex. Tech. Growth\")\n",
    "\n",
    "ax[0].plot(node_times, ssp5_eis_r2 + ssp5_ra_r2 + ssp5_prtp_r2 + ssp5_ex_r2 + ssp5_end_r2, linestyle='dashed', linewidth=lw, \n",
    "        color=color_list[l5cind])\n",
    "ax[0].fill_between(node_times, ssp5_eis_r2 + ssp5_ra_r2 + ssp5_prtp_r2 + ssp5_ex_r2, ssp5_ra_r2 + ssp5_eis_r2 + ssp5_prtp_r2 + ssp5_ex_r2 + ssp5_end_r2, alpha=alpha, \n",
    "                color=color_list[l5cind], label=\"End. Tech. Growth\")\n",
    "\n",
    "################################\n",
    "# SSP4\n",
    "################################\n",
    "ax[1].plot(node_times, ssp4_ra_r2, linestyle='dashed', linewidth=lw, \n",
    "        color=color_list[l1cind])\n",
    "ax[1].fill_between(node_times, np.zeros_like(node_times), ssp4_ra_r2, alpha=alpha, \n",
    "                color=color_list[l1cind], label=\"RA\")\n",
    "\n",
    "ax[1].plot(node_times, ssp4_eis_r2 + ssp4_ra_r2, linestyle='dashed', linewidth=lw, \n",
    "        color=color_list[l2cind])\n",
    "ax[1].fill_between(node_times, ssp4_ra_r2, ssp5_ra_r2 + ssp4_eis_r2, alpha=alpha, \n",
    "                color=color_list[l2cind], label=\"EIS\")\n",
    "\n",
    "ax[1].plot(node_times, ssp4_eis_r2 + ssp4_ra_r2 + ssp4_prtp_r2, linestyle='dashed', linewidth=lw, \n",
    "        color=color_list[l3cind])\n",
    "ax[1].fill_between(node_times, ssp4_eis_r2 + ssp4_ra_r2, ssp4_ra_r2 + ssp4_eis_r2 + ssp4_prtp_r2, alpha=alpha, \n",
    "                color=color_list[l3cind], label=\"PRTP\")\n",
    "\n",
    "ax[1].plot(node_times, ssp4_eis_r2 + ssp4_ra_r2 + ssp4_prtp_r2 + ssp4_ex_r2, linestyle='dashed', linewidth=lw, \n",
    "        color=color_list[l4cind])\n",
    "ax[1].fill_between(node_times, ssp4_eis_r2 + ssp4_ra_r2 + ssp4_prtp_r2, ssp4_ra_r2 + ssp4_eis_r2 + ssp4_prtp_r2 + ssp4_ex_r2, alpha=alpha, \n",
    "                color=color_list[l4cind], label=\"Ex. Tech. Growth\")\n",
    "\n",
    "ax[1].plot(node_times, ssp4_eis_r2 + ssp4_ra_r2 + ssp4_prtp_r2 + ssp4_ex_r2 + ssp4_end_r2, linestyle='dashed', linewidth=lw, \n",
    "        color=color_list[l5cind])\n",
    "ax[1].fill_between(node_times, ssp4_eis_r2 + ssp4_ra_r2 + ssp4_prtp_r2 + ssp4_ex_r2, ssp4_ra_r2 + ssp4_eis_r2 + ssp4_prtp_r2 + ssp4_ex_r2 + ssp4_end_r2, alpha=alpha, \n",
    "                color=color_list[l5cind], label=\"End. Tech. Growth\")\n",
    "\n",
    "################################\n",
    "# SSP3\n",
    "################################\n",
    "ax[2].plot(node_times, ssp3_ra_r2, linestyle='dashed', linewidth=lw, \n",
    "        color=color_list[l1cind])\n",
    "ax[2].fill_between(node_times, np.zeros_like(node_times), ssp3_ra_r2, alpha=alpha, \n",
    "                color=color_list[l1cind], label=\"RA\")\n",
    "\n",
    "ax[2].plot(node_times, ssp3_eis_r2 + ssp3_ra_r2, linestyle='dashed', linewidth=lw, \n",
    "        color=color_list[l2cind])\n",
    "ax[2].fill_between(node_times, ssp3_ra_r2, ssp3_ra_r2 + ssp3_eis_r2, alpha=alpha, \n",
    "                color=color_list[l2cind], label=\"EIS\")\n",
    "\n",
    "ax[2].plot(node_times, ssp3_eis_r2 + ssp3_ra_r2 + ssp3_prtp_r2, linestyle='dashed', linewidth=lw, \n",
    "        color=color_list[l3cind])\n",
    "ax[2].fill_between(node_times, ssp3_eis_r2 + ssp3_ra_r2, ssp3_ra_r2 + ssp3_eis_r2 + ssp3_prtp_r2, alpha=alpha, \n",
    "                color=color_list[l3cind], label=\"PRTP\")\n",
    "\n",
    "ax[2].plot(node_times, ssp3_eis_r2 + ssp3_ra_r2 + ssp3_prtp_r2 + ssp3_ex_r2, linestyle='dashed', linewidth=lw, \n",
    "        color=color_list[l4cind])\n",
    "ax[2].fill_between(node_times, ssp3_eis_r2 + ssp3_ra_r2 + ssp3_prtp_r2, ssp3_ra_r2 + ssp3_eis_r2 + ssp3_prtp_r2 + ssp3_ex_r2, alpha=alpha, \n",
    "                color=color_list[l4cind], label=\"Ex. Tech. Growth\")\n",
    "\n",
    "ax[2].plot(node_times, ssp3_eis_r2 + ssp3_ra_r2 + ssp3_prtp_r2 + ssp3_ex_r2 + ssp3_end_r2, linestyle='dashed', linewidth=lw, \n",
    "        color=color_list[l5cind])\n",
    "ax[2].fill_between(node_times, ssp3_eis_r2 + ssp3_ra_r2 + ssp3_prtp_r2 + ssp3_ex_r2, ssp3_ra_r2 + ssp3_eis_r2 + ssp3_prtp_r2 + ssp3_ex_r2 + ssp3_end_r2, alpha=alpha, \n",
    "                color=color_list[l5cind], label=\"End. Tech. Growth\")\n",
    "\n",
    "################################\n",
    "# SSP2\n",
    "################################\n",
    "ax[3].plot(node_times, ssp2_ra_r2, linestyle='dashed', linewidth=lw, \n",
    "        color=color_list[l1cind])\n",
    "ax[3].fill_between(node_times, np.zeros_like(node_times), ssp2_ra_r2, alpha=alpha, \n",
    "                color=color_list[l1cind], label=\"RA\")\n",
    "\n",
    "ax[3].plot(node_times, ssp2_eis_r2 + ssp2_ra_r2, linestyle='dashed', linewidth=lw, \n",
    "        color=color_list[l2cind])\n",
    "ax[3].fill_between(node_times, ssp2_ra_r2, ssp2_ra_r2 + ssp2_eis_r2, alpha=alpha, \n",
    "                color=color_list[l2cind], label=\"EIS\")\n",
    "\n",
    "ax[3].plot(node_times, ssp2_eis_r2 + ssp2_ra_r2 + ssp2_prtp_r2, linestyle='dashed', linewidth=lw, \n",
    "        color=color_list[l3cind])\n",
    "ax[3].fill_between(node_times, ssp2_eis_r2 + ssp2_ra_r2, ssp2_ra_r2 + ssp2_eis_r2 + ssp2_prtp_r2, alpha=alpha, \n",
    "                color=color_list[l3cind], label=\"PRTP\")\n",
    "\n",
    "ax[3].plot(node_times, ssp2_eis_r2 + ssp2_ra_r2 + ssp2_prtp_r2 + ssp2_ex_r2, linestyle='dashed', linewidth=lw, \n",
    "        color=color_list[l4cind])\n",
    "ax[3].fill_between(node_times, ssp2_eis_r2 + ssp2_ra_r2 + ssp2_prtp_r2, ssp2_ra_r2 + ssp2_eis_r2 + ssp2_prtp_r2 + ssp2_ex_r2, alpha=alpha, \n",
    "                color=color_list[l4cind], label=\"Ex. Tech. Growth\")\n",
    "\n",
    "ax[3].plot(node_times, ssp2_eis_r2 + ssp2_ra_r2 + ssp2_prtp_r2 + ssp2_ex_r2 + ssp2_end_r2, linestyle='dashed', linewidth=lw, \n",
    "        color=color_list[l5cind])\n",
    "ax[3].fill_between(node_times, ssp2_eis_r2 + ssp2_ra_r2 + ssp2_prtp_r2 + ssp2_ex_r2, ssp2_ra_r2 + ssp2_eis_r2 + ssp2_prtp_r2 + ssp2_ex_r2 + ssp2_end_r2, alpha=alpha, \n",
    "                color=color_list[l5cind], label=\"End. Tech. Growth\")\n",
    "\n",
    "################################\n",
    "# SSP1\n",
    "################################\n",
    "ax[4].plot(node_times, ssp1_ra_r2, linestyle='dashed', linewidth=lw, \n",
    "        color=color_list[l1cind])\n",
    "ax[4].fill_between(node_times, np.zeros_like(node_times), ssp5_ra_r2, alpha=alpha, \n",
    "                color=color_list[l1cind], label=\"RA\")\n",
    "\n",
    "ax[4].plot(node_times, ssp1_eis_r2 + ssp1_ra_r2, linestyle='dashed', linewidth=lw, \n",
    "        color=color_list[l2cind])\n",
    "ax[4].fill_between(node_times, ssp1_ra_r2, ssp1_ra_r2 + ssp1_eis_r2, alpha=alpha, \n",
    "                color=color_list[l2cind], label=\"EIS\")\n",
    "\n",
    "ax[4].plot(node_times, ssp1_eis_r2 + ssp1_ra_r2 + ssp1_prtp_r2, linestyle='dashed', linewidth=lw, \n",
    "        color=color_list[l3cind])\n",
    "ax[4].fill_between(node_times, ssp1_eis_r2 + ssp1_ra_r2, ssp1_ra_r2 + ssp1_eis_r2 + ssp1_prtp_r2, alpha=alpha, \n",
    "                color=color_list[l3cind], label=\"PRTP\")\n",
    "\n",
    "ax[4].plot(node_times, ssp1_eis_r2 + ssp1_ra_r2 + ssp1_prtp_r2 + ssp1_ex_r2, linestyle='dashed', linewidth=lw, \n",
    "        color=color_list[l4cind])\n",
    "ax[4].fill_between(node_times, ssp1_eis_r2 + ssp1_ra_r2 + ssp1_prtp_r2, ssp1_ra_r2 + ssp1_eis_r2 + ssp1_prtp_r2 + ssp1_ex_r2, alpha=alpha, \n",
    "                color=color_list[l4cind], label=\"Ex. Tech. Growth\")\n",
    "\n",
    "ax[4].plot(node_times, ssp1_eis_r2 + ssp1_ra_r2 + ssp1_prtp_r2 + ssp1_ex_r2 + ssp1_end_r2, linestyle='dashed', linewidth=lw, \n",
    "        color=color_list[l5cind])\n",
    "ax[4].fill_between(node_times, ssp1_eis_r2 + ssp1_ra_r2 + ssp1_prtp_r2 + ssp1_ex_r2, ssp1_ra_r2 + ssp1_eis_r2 + ssp1_prtp_r2 + ssp1_ex_r2 + ssp1_end_r2, alpha=alpha, \n",
    "                color=color_list[l5cind], label=\"End. Tech. Growth\")\n",
    "\n",
    "ax[2].legend(loc='upper center', bbox_to_anchor=(0.5, -0.25), fancybox=True, shadow=True, ncol=5)\n",
    "\n",
    "ax[0].set_ylabel(\"r$^{2}$\")\n",
    "ax[0].set_title(\"SSP5\")\n",
    "ax[1].set_title(\"SSP4\")\n",
    "ax[2].set_title(\"SSP3\")\n",
    "ax[3].set_title(\"SSP2\")\n",
    "ax[4].set_title(\"SSP1\")\n",
    "\n",
    "for i in range(5):\n",
    "    ax[i].set_xlim((node_times[0], node_times[-1]))\n",
    "    ax[i].set_aspect(1.0/ax[i].get_data_ratio(), adjustable='box')\n",
    "    ax[i].set_xlabel(\"Year\")\n",
    "\n",
    "#fig.tight_layout()"
   ]
  }
 ],
 "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.9.12"
  }
 },
 "nbformat": 4,
 "nbformat_minor": 5
}