VIS: Allow 'C0'-like plotting for plotting colors #15516

doc/source/whatsnew/v0.20.0.txt

@@ -342,6 +342,7 @@ Other Enhancements
 - The ``skiprows`` argument in ``pd.read_csv()`` now accepts a callable function as a value  (:issue:`10882`)
 - The ``nrows`` and ``chunksize`` arguments in ``pd.read_csv()`` are supported if both are passed (:issue:`6774`, :issue:`15755`)
 - ``pd.DataFrame.plot`` now prints a title above each subplot if ``suplots=True`` and ``title`` is a list of strings (:issue:`14753`)
+- ``pd.DataFrame.plot`` can pass `matplotlib 2.0 default color cycle as a single string as color parameter <http://matplotlib.org/2.0.0/users/colors.html#cn-color-selection>`__. (:issue:`15516`)
 - ``pd.Series.interpolate`` now supports timedelta as an index type with ``method='time'`` (:issue:`6424`)
 - ``Timedelta.isoformat`` method added for formatting Timedeltas as an `ISO 8601 duration`_. See the :ref:`Timedelta docs <timedeltas.isoformat>` (:issue:`15136`)
 - ``.select_dtypes()`` now allows the string 'datetimetz' to generically select datetimes with tz (:issue:`14910`)

pandas/tests/plotting/test_frame.py

@@ -141,6 +141,21 @@ def test_plot(self):
             result = ax.get_axes()  # deprecated
         self.assertIs(result, axes[0])
 
+    # GH 15516
+    def test_mpl2_color_cycle_str(self):
+        # test CN mpl 2.0 color cycle
+        import matplotlib.pyplot as plt
+        # mpl 1.5 color cycle is 'bgrcmyk' and
+        # tests up to 'C6' for compatibility
+        colors = ['C' + str(x) for x in range(7)]
+        hex_color = [c['color']
+                     for c in list(plt.rcParams['axes.prop_cycle'])]
+        df = DataFrame(randn(10, 3), columns=['a', 'b', 'c'])
+        for c in colors:
+            _check_plot_works(df.plot, color=hex_color[int(c[1])])
+        with tm.assertRaises(ValueError):
+            df.plot(color='')
+
     def test_color_and_style_arguments(self):
         df = DataFrame({'x': [1, 2], 'y': [3, 4]})
         # passing both 'color' and 'style' arguments should be allowed
@@ -1600,6 +1615,10 @@ def test_line_colors(self):
         self._check_colors(ax.get_lines(), linecolors=['red'] * 5)
         tm.close()
 
+        ax = df.plot(color='C0')
+        self._check_colors(ax.get_lines(), linecolors=['C0'] * 5)
+        tm.close()
+
         # GH 10299
         custom_colors = ['#FF0000', '#0000FF', '#FFFF00', '#000000', '#FFFFFF']
         ax = df.plot(color=custom_colors)

pandas/tools/plotting.py

@@ -225,10 +225,18 @@ def _maybe_valid_colors(colors):
         # check whether each character can be convertable to colors
         maybe_color_cycle = _maybe_valid_colors(list(colors))
         if maybe_single_color and maybe_color_cycle and len(colors) > 1:
-            msg = ("'{0}' can be parsed as both single color and "
-                   "color cycle. Specify each color using a list "
-                   "like ['{0}'] or {1}")
-            raise ValueError(msg.format(colors, list(colors)))
+            # Special case for single str 'CN' match and convert to hex
+            # for supporting matplotlib < 2.0.0
+            if re.match(r'C[0-9]', colors) and len(colors) == 2:
+                hex_color = [c['color']
+                             for c in list(plt.rcParams['axes.prop_cycle'])]
+                colors = [hex_color[int(colors[1])]]
+            else:
+                # non-default cycle may be parsed as single or many colors
+                msg = ("'{0}' can be parsed as both single color and "
+                       "color cycle. Specify each color using a list "
+                       "like ['{0}'] or {1}")
+                raise ValueError(msg.format(colors, list(colors)))
         elif maybe_single_color:
             colors = [colors]
         else:
@@ -237,7 +245,10 @@ def _maybe_valid_colors(colors):
             pass
 
     if len(colors) != num_colors:
-        multiple = num_colors // len(colors) - 1
+        try:
+            multiple = num_colors // len(colors) - 1
+        except ZeroDivisionError:
+            raise ValueError("Invalid color argument: ''")
         mod = num_colors % len(colors)
 
         colors += multiple * colors