Cleanup colorbar, legend, colormap-slicing examples

Author: lukelbd

docs/colorbars_legends.ipynb

@@ -120,7 +120,7 @@
    "source": [
     "import proplot as plot\n",
     "import numpy as np\n",
-    "f, axs = plot.subplots(ncols=3, nrows=3, axwidth=1.2)\n",
+    "f, axs = plot.subplots(ncols=3, nrows=3, axwidth=1.4)\n",
     "state = np.random.RandomState(51423)\n",
     "m = axs.pcolormesh(\n",
     "    state.rand(20, 20), cmap='grays',\n",
@@ -145,7 +145,7 @@
     "import proplot as plot\n",
     "import numpy as np\n",
     "f, axs = plot.subplots(\n",
-    "    ncols=2, nrows=2, axwidth=1.3,\n",
+    "    ncols=2, nrows=2, axwidth=1.7,\n",
     "    share=0, wspace=0.3, order='F'\n",
     ")\n",
     "\n",
@@ -164,8 +164,10 @@
     "f.colorbar(m[0], label='colorbar label', loc='l')\n",
     "f.legend(hs, ncols=2, center=True, frame=False, loc='b', col=2)\n",
     "f.legend(hs, ncols=1, label='legend label', frame=False, loc='r')\n",
-    "axs.format(suptitle='Figure colorbars and legends demo',\n",
-    "           abc=True, abcloc='ul', abcstyle='A')\n",
+    "axs.format(\n",
+    "    suptitle='Figure colorbars and legends demo',\n",
+    "    abc=True, abcloc='ul', abcstyle='A'\n",
+    ")\n",
     "for ax, title in zip(axs, ['2D dataset #1', '2D dataset #2', 'Line set #1', 'Line set #2']):\n",
     "    ax.format(xlabel='xlabel', title=title)"
    ]
@@ -196,29 +198,41 @@
    "source": [
     "import proplot as plot\n",
     "import numpy as np\n",
-    "f, axs = plot.subplots(share=0, ncols=2)\n",
+    "f, axs = plot.subplots(share=0, ncols=2, axwidth=2)\n",
     "\n",
     "# Colorbars from lines\n",
     "ax = axs[0]\n",
     "state = np.random.RandomState(51423)\n",
     "data = 1 + (state.rand(12, 10) - 0.45).cumsum(axis=0)\n",
     "cycle = plot.Cycle('algae')\n",
-    "hs = ax.plot(data, lw=4, cycle=cycle, colorbar='lr', colorbar_kw={\n",
-    "             'length': '8em', 'label': 'from lines'})\n",
-    "ax.colorbar(hs, loc='t', values=2*np.linspace(0.5, 9.5, 10),\n",
-    "            label='from lines', length=0.7, ticks=2)\n",
+    "hs = ax.plot(\n",
+    "    data, lw=4, cycle=cycle, colorbar='lr',\n",
+    "    colorbar_kw={'length': '8em', 'label': 'from lines'}\n",
+    ")\n",
+    "ax.colorbar(\n",
+    "    hs, loc='t', values=2*np.linspace(0.5, 9.5, 10),\n",
+    "    label='from lines', length=0.7, ticks=2\n",
+    ")\n",
     "\n",
     "# Colorbars from a mappable\n",
     "ax = axs[1]\n",
-    "m = ax.contourf(data.T, extend='both', cmap='algae',\n",
-    "                levels=plot.arange(0, 3, 0.5))\n",
-    "f.colorbar(m, length=1, loc='r', label='inside ticks',\n",
-    "           tickloc='left')\n",
-    "ax.colorbar(m, loc='ul', length=1, tickminor=True, drawedges=True, extendrect=True,\n",
-    "            label='inset colorbar', labelcolor='gray7', labelweight='bold',\n",
-    "            linewidth=1, edgecolor='gray7', ticklabelcolor='gray7', alpha=0.5)\n",
-    "axs.format(suptitle='Colorbar formatting demo',\n",
-    "           xlabel='xlabel', ylabel='ylabel', top=False)"
+    "m = ax.contourf(\n",
+    "    data.T, extend='both', cmap='algae',\n",
+    "    levels=plot.arange(0, 3, 0.5)\n",
+    ")\n",
+    "f.colorbar(\n",
+    "    m, length=1, loc='r', label='inside ticks',\n",
+    "    tickloc='left'\n",
+    ")\n",
+    "ax.colorbar(\n",
+    "    m, loc='ul', length=1, tickminor=True, drawedges=True,\n",
+    "    label='inset colorbar', labelcolor='gray7', labelweight='bold',\n",
+    "    linewidth=1, edgecolor='gray7', ticklabelcolor='gray7', alpha=0.5\n",
+    ")\n",
+    "axs.format(\n",
+    "    suptitle='Colorbar formatting demo',\n",
+    "    xlabel='xlabel', ylabel='ylabel', top=False\n",
+    ")"
    ]
   },
   {
@@ -249,24 +263,30 @@
     "import numpy as np\n",
     "plot.rc.cycle = 'contrast'\n",
     "labels = ['a', 'bb', 'ccc', 'dddd', 'eeeee']\n",
-    "f, axs = plot.subplots(ncols=2, span=False, share=1)\n",
+    "f, axs = plot.subplots(ncols=2, span=False, share=1, axwidth=2)\n",
     "hs1, hs2 = [], []\n",
     "\n",
     "# On-the-fly legends\n",
     "state = np.random.RandomState(51423)\n",
     "for i, label in enumerate(labels):\n",
-    "    data = (state.rand(20)-0.45).cumsum(axis=0)\n",
-    "    h1 = axs[0].plot(data, lw=4, label=label, legend='ul',\n",
-    "                     legend_kw={'order': 'F', 'title': 'column major'})  # add to legend in upper left\n",
+    "    data = (state.rand(20) - 0.45).cumsum(axis=0)\n",
+    "    h1 = axs[0].plot(\n",
+    "        data, lw=4, label=label, legend='ul',\n",
+    "        legend_kw={'order': 'F', 'title': 'column major'}\n",
+    "    )\n",
     "    hs1.extend(h1)\n",
-    "    h2 = axs[1].plot(data, lw=4, label=label, legend='r', cycle='floral',\n",
-    "                     legend_kw={'ncols': 1, 'frame': False, 'title': 'no frame'})  # add to legend in right panel\n",
+    "    h2 = axs[1].plot(\n",
+    "        data, lw=4, label=label, legend='r', cycle='floral',\n",
+    "        legend_kw={'ncols': 1, 'frame': False, 'title': 'no frame'}\n",
+    "    )\n",
     "    hs2.extend(h2)\n",
     "    \n",
     "# Outer legends\n",
     "ax = axs[0]\n",
-    "ax.legend(hs1, loc='b', ncols=3, linewidth=2, title='row major', order='C',\n",
-    "          edgecolor='gray4', facecolor='gray2')\n",
+    "ax.legend(\n",
+    "    hs1, loc='b', ncols=3, linewidth=2,title='row major', order='C',\n",
+    "    facecolor='gray2'\n",
+    ")\n",
     "ax = axs[1]\n",
     "ax.legend(hs2, loc='b', ncols=3, center=True, title='centered rows')\n",
     "axs.format(xlabel='xlabel', ylabel='ylabel', suptitle='Legend formatting demo')"

docs/colormaps.ipynb

@@ -267,8 +267,10 @@
    "source": [
     "import proplot as plot\n",
     "import numpy as np\n",
-    "f, axs = plot.subplots([[0, 1, 1, 2, 2, 3, 3, 0], [4, 4, 5, 5, 6, 6, 7, 7]],\n",
-    "                       axwidth=1.7, span=False)\n",
+    "f, axs = plot.subplots(\n",
+    "    [[1, 1, 2, 2, 3, 3], [0, 4, 4, 5, 5, 0], [0, 6, 6, 7, 7, 0]],\n",
+    "    axwidth=1.7, span=False\n",
+    ")\n",
     "state = np.random.RandomState(51423)\n",
     "data = state.rand(40, 40).cumsum(axis=0) - 12\n",
     "\n",
@@ -281,8 +283,10 @@
     "        title, cmap_kw = f'left={coord}', {'left': coord}\n",
     "    else:\n",
     "        title, cmap_kw = f'right={coord}', {'right': coord}\n",
-    "    ax.pcolormesh(data, cmap=cmap, cmap_kw=cmap_kw,\n",
-    "                  colorbar='b', colorbar_kw={'locator': 'null'})\n",
+    "    ax.pcolormesh(\n",
+    "        data, cmap=cmap, cmap_kw=cmap_kw,\n",
+    "        colorbar='b', colorbar_kw={'locator': 'null'}\n",
+    "    )\n",
     "    ax.format(xlabel='x axis', ylabel='y axis', title=title)\n",
     "    \n",
     "# Cutting central colors\n",