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

doc/source/whatsnew/v0.20.0.txt

@@ -177,6 +177,7 @@ Other enhancements
 - The ``usecols`` argument in ``pd.read_csv`` now accepts a callable function as a value  (:issue:`14154`)
 - The ``skiprows`` argument in ``pd.read_csv`` now accepts a callable function as a value  (:issue:`10882`)
 - ``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 (: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`)
 - ``pandas.io.json.json_normalize()`` gained the option ``errors='ignore'|'raise'``; the default is ``errors='raise'`` which is backward compatible. (:issue:`14583`)

pandas/tests/plotting/test_frame.py

@@ -141,6 +141,12 @@ def test_plot(self):
             result = ax.get_axes()  # deprecated
         self.assertIs(result, axes[0])
 
+    def test_mpl2_color_cycle_str(self):
+        colors = ['C' + str(x) for x in range(10)]
+        df = DataFrame(randn(10, 3), columns=['a', 'b', 'c'])
+        for c in colors:
+            _check_plot_works(df.plot, color=c)
+
     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 +1606,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'])
+        tm.close()
+
         # GH 10299
         custom_colors = ['#FF0000', '#0000FF', '#FFFF00', '#000000', '#FFFFFF']
         ax = df.plot(color=custom_colors)

pandas/tools/plotting.py

@@ -217,10 +217,13 @@ 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)))
+            if color[0] == 'C' and len(color) == 2:
+                colors = [colors]
+            else:
+                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: