PopulationDemo.ipynb

{
 "cells": [
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "import os.path\n",
    "import bqplot\n",
    "import ui.charts\n",
    "import ui.color\n",
    "from IPython.display import display, HTML\n",
    "import ipywidgets\n",
    "import numpy as np\n",
    "import pandas as pd\n",
    "\n",
    "solutions = pd.read_csv(os.path.join('data', 'overview', 'solutions.csv'),\n",
    "                                    index_col=False, skipinitialspace=True, header=0,\n",
    "                                    skip_blank_lines=True, comment='#')\n",
    "soln_results = pd.read_csv(os.path.join('data', 'overview', 'populationdemo.csv'),\n",
    "                                    index_col=False, skipinitialspace=True, header=0,\n",
    "                                    skip_blank_lines=True, comment='#')\n",
    "solutions = solutions.merge(soln_results, on='Solution', how='left')\n",
    "\n",
    "soln_data = solutions['CO2eq-pop9B'].tolist()\n",
    "soln_labels = solutions['Solution'].tolist()\n",
    "soln_colors = solutions['Sector'].apply(ui.color.get_sector_color).tolist()\n",
    "soln_tt = bqplot.Tooltip(fields=['label', 'size'], formats=['', '0.1f'], labels=['Name', 'GTons-CO2'])\n",
    "soln_pie = bqplot.Pie(sizes=soln_data, display_labels='none', labels=soln_labels, sort=True, tooltip=soln_tt,\n",
    "                 display_values=True, values_format='.1f', radius=170, inner_radius=150,\n",
    "                 colors=soln_colors, stroke='Gainsboro')\n",
    "soln_pie.tooltip_style = {'opacity': 0.7, 'background-color': 'White', 'border': 'none', 'border-collapse': 'collapse'}\n",
    "\n",
    "sectors = solutions.pivot_table(index='Sector',aggfunc=sum)\n",
    "solutions['CO2eq'] = solutions['CO2eq-pop9B']\n",
    "solutions['SectorCO2eq'] = solutions.apply(lambda row: sectors.loc[row['Sector'], 'CO2eq-pop9B'], axis=1)\n",
    "sctr_data = sectors['CO2eq-pop9B'].tolist()\n",
    "sctr_labels = sectors.index.tolist()\n",
    "sctr_colors = [ui.color.get_sector_color(v) for v in sctr_labels]\n",
    "sctr_tt = bqplot.Tooltip(fields=['label', 'size'], formats=['', '0.1f'], labels=['Name', 'GTons-CO2eq'])\n",
    "sctr_pie = bqplot.Pie(sizes=sctr_data, display_labels='none', labels=sctr_labels, sort=True, tooltip=sctr_tt,\n",
    "                 display_values=True, values_format='.1f', radius=138, inner_radius=118,\n",
    "                 colors=sctr_colors, stroke='Gainsboro')\n",
    "sctr_pie.tooltip_style = {'opacity': 0.7, 'background-color': 'White', 'border': 'none', 'border-collapse': 'collapse'}\n",
    "solution_chart = bqplot.Figure(marks=[soln_pie, sctr_pie], animation_duration=1000)\n",
    "solution_chart.background_style = {'fill': 'White'}\n",
    "\n",
    "def get_solution_html():\n",
    "    return (solutions.sort_values(['CO2eq'], ascending=False).loc[:,['Solution', 'Sector', 'CO2eq']].fillna(0.0)\n",
    "        .style.format({'CO2eq':'{:.02f}'}).set_table_styles(ui.charts.dataframe_css_styles)\n",
    "        .hide_index().render())\n",
    "    \n",
    "solution_list = ipywidgets.HTML(get_solution_html())"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "pop = ipywidgets.FloatSlider(\n",
    "    value=10.0,\n",
    "    min=8.0,\n",
    "    max=10.0,\n",
    "    step=0.01,\n",
    "    description='Pop (B):',\n",
    "    disabled=False,\n",
    "    continuous_update=False,\n",
    "    orientation='horizontal',\n",
    "    readout=True,\n",
    "    readout_format='.2f',\n",
    ")\n",
    "\n",
    "cap = ipywidgets.FloatSlider(\n",
    "    value=0.0,\n",
    "    min=0.0,\n",
    "    max=5.0,\n",
    "    step=0.01,\n",
    "    description='Cap (US$T):',\n",
    "    disabled=False,\n",
    "    continuous_update=False,\n",
    "    orientation='horizontal',\n",
    "    readout=True,\n",
    "    readout_format='.2f',\n",
    ")\n",
    "\n",
    "def update_solutions(population, capital):\n",
    "    delta_pop = (solutions['CO2eq-pop9B'] - solutions['CO2eq-pop8B']) * ((population - 8.0) / 2.0)\n",
    "    delta_cap = (solutions['CO2eq-fc5T'] - solutions['CO2eq-fc0T']) * (capital / 5.0)\n",
    "    solutions['CO2eq'] = solutions['CO2eq-pop8B'] + delta_pop + delta_cap\n",
    "    soln_pie.sizes = solutions['CO2eq'].tolist()\n",
    "    sectors = solutions.pivot_table(index='Sector',aggfunc=sum)\n",
    "    solutions['SectorCO2eq'] = solutions.apply(lambda row: sectors.loc[row['Sector'], 'CO2eq'], axis=1)\n",
    "    sctr_pie.sizes = sectors['CO2eq'].tolist()\n",
    "    solution_list.value = get_solution_html()\n",
    "\n",
    "p = ipywidgets.interactive(update_solutions, population=pop, capital=cap)\n",
    "display(p)\n",
    "\n",
    "h = ipywidgets.HBox(children=[solution_list, solution_chart])\n",
    "h.layout.justify_content = 'space-between'\n",
    "h"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": []
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": []
  }
 ],
 "metadata": {
  "kernelspec": {
   "display_name": "Python 3",
   "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.6.8"
  }
 },
 "nbformat": 4,
 "nbformat_minor": 2
}