BUG: Accept list-like color with single col in plot

Closes #3486
pandas-dev · May 11, 2017 · 7980460 · 7980460
1 parent fdc2185
commit 7980460
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 1 deletion.
diff --git a/doc/source/whatsnew/v0.20.2.txt b/doc/source/whatsnew/v0.20.2.txt
@@ -55,6 +55,8 @@ I/O
 Plotting
 ^^^^^^^^
 
+- Bug in ``DataFrame.plot`` with a single column and a list-like ``color`` (:issue:`3486`)
+
 
 
 

diff --git a/pandas/plotting/_core.py b/pandas/plotting/_core.py
@@ -180,7 +180,8 @@ def _validate_color_args(self):
             colors = self.kwds.pop('colors')
             self.kwds['color'] = colors
 
-        if ('color' in self.kwds and self.nseries == 1):
+        if ('color' in self.kwds and self.nseries == 1 and
+                not is_list_like(self.kwds['color'])):
             # support series.plot(color='green')
             self.kwds['color'] = [self.kwds['color']]
 

diff --git a/pandas/tests/plotting/test_frame.py b/pandas/tests/plotting/test_frame.py
@@ -153,6 +153,11 @@ def test_mpl2_color_cycle_str(self):
         else:
             pytest.skip("not supported in matplotlib < 2.0.0")
 
+    def test_color_single_series_list(self):
+        # GH 3486
+        df = DataFrame({"A": [1, 2, 3]})
+        _check_plot_works(df.plot, color=['red'])
+
     def test_color_empty_string(self):
         df = DataFrame(randn(10, 2))
         with pytest.raises(ValueError):