fixes notebook resolution examples (#206)

Author: vuillaut

ctaplot/plots/plots.py

@@ -279,13 +279,13 @@ def plot_impact_point_heatmap(reco_x, reco_y, ax=None, outfile=None, **kwargs):
     outfile: string
         path to the output file. If None, the figure is not saved.
     """
-
+    from matplotlib import colormaps
     ax = plt.gca() if ax is None else ax
 
     unit = reco_x.unit
 
     kwargs.setdefault('norm', LogNorm())
-    kwargs.setdefault('cmap', plt.cm.get_cmap('PuBu'))
+    kwargs.setdefault('cmap', colormaps['PuBu'])
     kwargs.setdefault('bins', 50)
     h = ax.hist2d(reco_x.to_value(unit), reco_y.to_value(unit), **kwargs)
     cb = plt.colorbar(h[3], ax=ax)

examples/notebooks/resolution_examples.ipynb

@@ -95,8 +95,21 @@
    "outputs": [],
    "source": [
     "fig, axes = plt.subplots(1, 2, figsize=(20,5))\n",
-    "ctaplot.plot_theta2(reco_alt, reco_az, simu_alt, simu_az, bins=40, ax=axes[0])\n",
-    "ctaplot.plot_angular_resolution_per_energy(reco_alt, reco_az, simu_alt, simu_az, simu_energy, ax=axes[1]) \n",
+    "ctaplot.plot_theta2(simu_alt, reco_alt, simu_az, reco_az, bins=40, ax=axes[0])\n",
+    "ctaplot.plot_angular_resolution_per_energy(simu_alt, reco_alt, simu_az, reco_az, simu_energy, ax=axes[1]) \n",
+    "ctaplot.plot_angular_resolution_cta_requirement('south', ax=axes[1], color='black')\n",
+    "axes[1].legend();\n",
+    "plt.show()"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "reco_az, , simu_az, bins=40, ax=axes[0])\n",
+    "ctaplot.plot_angular_resolution_per_energy(simu_alt, reco_alt, simu_az, reco_az, simu_energy, ax=axes[1]) \n",
     "ctaplot.plot_angular_resolution_cta_requirement('south', ax=axes[1], color='black')\n",
     "axes[1].legend();\n",
     "plt.show()"
@@ -127,7 +140,7 @@
     "                    bias_correction=True,\n",
     "                    ax=axes[0])\n",
     "\n",
-    "ctaplot.plot_angular_resolution_per_energy(reco_alt, reco_az, simu_alt, simu_az, simu_energy,\n",
+    "ctaplot.plot_angular_resolution_per_energy(simu_alt, reco_alt, simu_az, reco_az, simu_energy,\n",
     "                                           bias_correction=True,\n",
     "                                           ax=axes[1])\n",
     "\n",