BUG: Fix passing Colormap instance to plot methods with mpl >= 3.6 (p…

…andas-dev#49377)

* Create failing test for passing Colormap inst

to `.plot.scatter`

* Use `mpl.colormaps.get_cmap`

when a Colormap instance might be passed

* Test hexbin too

* Add whatsnew

doc/source/whatsnew/v1.5.2.rst

@@ -14,6 +14,8 @@ including other versions of pandas.
 Fixed regressions
 ~~~~~~~~~~~~~~~~~
 - Fixed regression in :meth:`Series.replace` raising ``RecursionError`` with numeric dtype and when specifying ``value=None`` (:issue:`45725`)
+- Fixed regression in :meth:`DataFrame.plot` preventing :class:`~matplotlib.colors.Colormap` instance
+  from being passed using the ``colormap`` argument if Matplotlib 3.6+ is used (:issue:`49374`)
 -
 
 .. ---------------------------------------------------------------------------

pandas/io/formats/style.py

@@ -3610,7 +3610,7 @@ def _background_gradient(
             if cmap is None:
                 rgbas = mpl.colormaps[mpl.rcParams["image.cmap"]](norm(gmap))
             else:
-                rgbas = mpl.colormaps[cmap](norm(gmap))
+                rgbas = mpl.colormaps.get_cmap(cmap)(norm(gmap))
         else:
             rgbas = plt.cm.get_cmap(cmap)(norm(gmap))
 

pandas/plotting/_matplotlib/core.py

@@ -1230,7 +1230,7 @@ def _make_plot(self):
 
         if self.colormap is not None:
             if mpl_ge_3_6_0():
-                cmap = mpl.colormaps[self.colormap]
+                cmap = mpl.colormaps.get_cmap(self.colormap)
             else:
                 cmap = self.plt.cm.get_cmap(self.colormap)
         else:
@@ -1310,7 +1310,7 @@ def _make_plot(self) -> None:
         # pandas uses colormap, matplotlib uses cmap.
         cmap = self.colormap or "BuGn"
         if mpl_ge_3_6_0():
-            cmap = mpl.colormaps[cmap]
+            cmap = mpl.colormaps.get_cmap(cmap)
         else:
             cmap = self.plt.cm.get_cmap(cmap)
         cb = self.kwds.pop("colorbar", True)

pandas/tests/io/formats/style/test_matplotlib.py

@@ -288,3 +288,17 @@ def test_bar_color_raises(df):
     msg = "`color` and `cmap` cannot both be given"
     with pytest.raises(ValueError, match=msg):
         df.style.bar(color="something", cmap="something else").to_html()
+
+
+@pytest.mark.parametrize(
+    "plot_method",
+    ["scatter", "hexbin"],
+)
+def test_pass_colormap_instance(df, plot_method):
+    # https://github.com/pandas-dev/pandas/issues/49374
+    cmap = mpl.colors.ListedColormap([[1, 1, 1], [0, 0, 0]])
+    df["c"] = df.A + df.B
+    kwargs = dict(x="A", y="B", c="c", colormap=cmap)
+    if plot_method == "hexbin":
+        kwargs["C"] = kwargs.pop("c")
+    getattr(df.plot, plot_method)(**kwargs)