j1-template 2022.5.0.rc1 → 2022.5.0.rc4

data/lib/starter_web/pages/public/_jupyter/notebooks/martin_renou_bqplot/Applications/Wealth Of Nations/Choropleth.ipynb

@@ -0,0 +1,275 @@
+{
+ "cells": [
+  {
+   "cell_type": "markdown",
+   "metadata": {},
+   "source": [
+    "## Wealth Of Nations Dashboard\n",
+    "1. Use the dropdown to select the data type\n",
+    "2. Use the slider to show the heatmap for a specific year\n",
+    "3. Use the animation button to animate the choropleth for all the years\n",
+    "4. Clicking on any country toggles the display of timeseries for that country. (Clicking on oceans un-selects all the countries)"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "import numpy as np\n",
+    "import pandas as pd\n",
+    "\n",
+    "import ipywidgets as widgets\n",
+    "import bqplot as bq\n",
+    "import bqplot.pyplot as plt"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "initial_year = 1800"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {
+    "scrolled": false
+   },
+   "outputs": [],
+   "source": [
+    "data = pd.read_json(\"nations.json\")\n",
+    "country_codes = pd.read_csv(\"world_map_codes.csv\")\n",
+    "country_code_map = dict(zip(country_codes[\"ISON3\"], country_codes[\"Name\"]))"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "country_data = (\n",
+    "    data.merge(country_codes, left_on=\"name\", right_on=\"Name\")[\n",
+    "        [\"name\", \"ISON3\", \"region\", \"income\", \"lifeExpectancy\", \"population\"]\n",
+    "    ]\n",
+    "    .rename(columns={\"ISON3\": \"id\"})\n",
+    "    .set_index(\"id\")\n",
+    ")"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "def get_time_series(col):\n",
+    "    time_series_df = country_data[col].apply(dict).apply(pd.Series)\n",
+    "    time_series_df.index.name = None\n",
+    "    time_series_df.columns = time_series_df.columns.astype(\"int\")\n",
+    "\n",
+    "    return time_series_df"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "income_time_series = get_time_series(\"income\")\n",
+    "life_exp_time_series = get_time_series(\"lifeExpectancy\")\n",
+    "pop_time_series = get_time_series(\"population\")"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {
+    "scrolled": true
+   },
+   "outputs": [],
+   "source": [
+    "income_min, income_max = income_time_series.min().min(), income_time_series.max().max()\n",
+    "life_exp_min, life_exp_max = (\n",
+    "    life_exp_time_series.min().min(),\n",
+    "    life_exp_time_series.max().max(),\n",
+    ")\n",
+    "pop_min, pop_max = pop_time_series.min().min(), pop_time_series.max().max()"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {
+    "scrolled": true
+   },
+   "outputs": [],
+   "source": [
+    "data_dict = {\n",
+    "    \"Income\": income_time_series,\n",
+    "    \"Population\": pop_time_series,\n",
+    "    \"Life Expectancy\": life_exp_time_series,\n",
+    "}\n",
+    "\n",
+    "min_max_dict = {\n",
+    "    \"Income\": (np.log(income_min), np.log(income_max)),\n",
+    "    \"Population\": (np.log(pop_min), np.log(pop_max)),\n",
+    "    \"Life Expectancy\": (life_exp_min, life_exp_max),\n",
+    "}\n",
+    "\n",
+    "color_scheme_dict = {\n",
+    "    \"Income\": \"Greens\",\n",
+    "    \"Population\": \"Blues\",\n",
+    "    \"Life Expectancy\": \"Oranges\",\n",
+    "}\n",
+    "\n",
+    "data_type_dropdown = widgets.Dropdown(\n",
+    "    description=\"Data Type\",\n",
+    "    options=[\"Income\", \"Life Expectancy\", \"Population\"],\n",
+    "    value=\"Population\",\n",
+    ")\n",
+    "\n",
+    "year_slider = widgets.IntSlider(\n",
+    "    description=\"year\",\n",
+    "    min=1800,\n",
+    "    max=2008,\n",
+    "    continuous_update=False,\n",
+    "    layout=widgets.Layout(width=\"300px\"),\n",
+    ")\n",
+    "\n",
+    "play_button = widgets.Play(min=1800, max=2008, interval=10)\n",
+    "widgets.jslink((play_button, \"value\"), (year_slider, \"value\"))\n",
+    "\n",
+    "# world map\n",
+    "world_map_fig_title_tmpl = \"World {data_type} for year {year}\"\n",
+    "world_map_fig = plt.figure(\n",
+    "    layout=widgets.Layout(width=\"1000px\", height=\"500px\"),\n",
+    "    fig_margin=dict(top=45, bottom=10, left=20, right=20),\n",
+    ")\n",
+    "tooltip = bq.Tooltip(fields=[\"name\"], labels=[\"Country\"])\n",
+    "\n",
+    "plt.scales(scales={\"projection\": bq.Mercator(), \"color\": bq.ColorScale()})\n",
+    "\n",
+    "world_map = plt.geo(\n",
+    "    map_data=\"WorldMap\",\n",
+    "    tooltip=tooltip,\n",
+    "    interactions={\"click\": \"select\", \"hover\": \"tooltip\"},\n",
+    "    axes_options={\"color\": {\"visible\": False}},\n",
+    "    selected_styles={\"selected_fill\": \"yellow\"},\n",
+    "    colors={\"default_color\": \"Grey\"},\n",
+    ")\n",
+    "\n",
+    "# data time series fig\n",
+    "ts_fig_title_tmpl = \"{data_type} Time Series\"\n",
+    "time_series_fig = plt.figure(\n",
+    "    layout=widgets.Layout(width=\"1000px\", height=\"400px\"),\n",
+    "    fig_margin=dict(top=40, bottom=30, left=60, right=40),\n",
+    ")\n",
+    "country_time_series = plt.plot(\n",
+    "    list(pop_time_series.columns),\n",
+    "    [],\n",
+    "    axes_options={\"x\": {\"label\": \"Year\"}, \"y\": {\"tick_format\": \"~s\"}},\n",
+    "    display_legend=True,\n",
+    ")\n",
+    "\n",
+    "# link widgets\n",
+    "def on_data_type_update(*args):\n",
+    "    global data_time_series\n",
+    "    # reset color and selected attributes of world_map\n",
+    "    world_map.color = {}\n",
+    "    world_map.selected = []\n",
+    "\n",
+    "    data_type = data_type_dropdown.value\n",
+    "    data_time_series = data_dict[data_type]\n",
+    "\n",
+    "    scale_min, scale_max = min_max_dict[data_type]\n",
+    "    color_scale = world_map.scales[\"color\"]\n",
+    "    color_scale.scheme = color_scheme_dict[data_type]\n",
+    "    color_scale.min = scale_min\n",
+    "    color_scale.max = scale_max\n",
+    "    year_slider.value = initial_year\n",
+    "\n",
+    "\n",
+    "data_type_dropdown.observe(on_data_type_update, \"value\")\n",
+    "\n",
+    "\n",
+    "def update_color(*args):\n",
+    "    data_type = data_type_dropdown.value\n",
+    "    year = year_slider.value\n",
+    "\n",
+    "    if year in data_time_series.columns:\n",
+    "        if data_type in [\"Population\", \"Income\"]:\n",
+    "            color_data = np.log(data_time_series[year])\n",
+    "        else:\n",
+    "            color_data = data_time_series[year]\n",
+    "        world_map.color = dict(color_data)\n",
+    "    else:\n",
+    "        world_map.color = {}\n",
+    "\n",
+    "    world_map_fig.title = world_map_fig_title_tmpl.format(\n",
+    "        data_type=data_type, year=year\n",
+    "    )\n",
+    "\n",
+    "\n",
+    "year_slider.observe(update_color, \"value\")\n",
+    "\n",
+    "\n",
+    "def plot_time_series_for_country(*args):\n",
+    "    if world_map.selected is not None and len(world_map.selected) > 0:\n",
+    "        selected_countries = world_map.selected\n",
+    "        country_time_series.y = data_time_series.loc[selected_countries]\n",
+    "        country_time_series.labels = [\n",
+    "            country_code_map[c_id] for c_id in selected_countries\n",
+    "        ]\n",
+    "    else:\n",
+    "        country_time_series.labels = []\n",
+    "        country_time_series.y = []\n",
+    "    time_series_fig.title = ts_fig_title_tmpl.format(data_type=data_type_dropdown.value)\n",
+    "\n",
+    "\n",
+    "world_map.observe(plot_time_series_for_country, \"selected\")\n",
+    "\n",
+    "on_data_type_update(None)\n",
+    "\n",
+    "widgets.VBox(\n",
+    "    [\n",
+    "        world_map_fig,\n",
+    "        widgets.HBox(\n",
+    "            [\n",
+    "                time_series_fig,\n",
+    "                widgets.VBox([data_type_dropdown, year_slider, play_button]),\n",
+    "            ]\n",
+    "        ),\n",
+    "    ]\n",
+    ")"
+   ]
+  }
+ ],
+ "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.8.2"
+  }
+ },
+ "nbformat": 4,
+ "nbformat_minor": 1
+}