Work on heat maps.

Author: cwant

.gitignore

@@ -104,5 +104,5 @@ venv.bak/
 .mypy_cache/
 
 notebooks/my-first-scatter.html
-Untitled.ipynb
+Untitled*.ipynb
 gapminder_gdp_americas.csv

notebooks/03d-plotly-3dviz.ipynb

@@ -32,6 +32,35 @@
     "py.init_notebook_mode(connected=False)"
    ]
   },
+  {
+   "cell_type": "markdown",
+   "metadata": {},
+   "source": [
+    "People using Colab need this (initialization and download of supporting materials)"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "# Stuff for Colab ...\n",
+    "import sys\n",
+    "def enable_plotly_in_cell():\n",
+    "  import IPython\n",
+    "  from plotly.offline import init_notebook_mode\n",
+    "  display(IPython.core.display.HTML('''\n",
+    "        <script src=\"/static/components/requirejs/require.js\"></script>\n",
+    "  '''))\n",
+    "  init_notebook_mode(connected=False)\n",
+    "\n",
+    "if 'google.colab' in sys.modules:\n",
+    "    get_ipython().events.register('pre_run_cell', enable_plotly_in_cell)\n",
+    "    !mkdir -p data\n",
+    "    !wget -P data https://raw.githubusercontent.com/ualberta-rcg/python-plotting/master/notebooks/data/gapminder_gdp_europe.csv"
+   ]
+  },
   {
    "cell_type": "markdown",
    "metadata": {},
@@ -357,11 +386,14 @@
    "metadata": {},
    "outputs": [],
    "source": [
+    "# Some versions will prefer `z=df.to_numpy()` below\n",
+    "\n",
     "data = [\n",
     "    go.Heatmap(\n",
     "        x=years,\n",
     "        y=df.index,\n",
     "        z=df,\n",
+    "        colorscale='Jet'\n",
     "    )\n",
     "]\n",
     "layout = go.Layout(\n",
@@ -468,7 +500,8 @@
     "        y=df.index,\n",
     "        z=df,\n",
     "        hoverinfo='text',\n",
-    "        text=hovertext\n",
+    "        text=hovertext,\n",
+    "        colorscale='Jet'\n",
     "    )\n",
     "]\n",
     "layout = go.Layout(\n",
@@ -493,6 +526,7 @@
     "        x=years,\n",
     "        y=df_sorted.index,\n",
     "        z=df_sorted,\n",
+    "        colorscale='Jet'\n",
     "    )\n",
     "]\n",
     "layout = go.Layout(\n",