BUG: #11637 fix to_csv incorrect output.

Author: varunkumar-dev

doc/source/whatsnew/v0.17.1.txt

@@ -166,12 +166,8 @@ Bug Fixes
 - Bug in ``squeeze()`` with zero length arrays (:issue:`11230`, :issue:`8999`)
 - Bug in ``describe()`` dropping column names for hierarchical indexes (:issue:`11517`)
 - Bug in ``DataFrame.pct_change()`` not propagating ``axis`` keyword on ``.fillna`` method (:issue:`11150`)
-
-
-
+- Bug in ``.to_csv()`` when a mix of integer and string column names are passed as the ``columns`` parameter (:issue:`11637`)
 - Bug in indexing with a ``range``, (:issue:`11652`)
-
-
 - Bug in ``to_sql`` using unicode column names giving UnicodeEncodeError with (:issue:`11431`).
 - Fix regression in setting of ``xticks`` in ``plot`` (:issue:`11529`).
 - Bug in ``holiday.dates`` where observance rules could not be applied to holiday and doc enhancement (:issue:`11477`, :issue:`11533`)

pandas/core/format.py

@@ -1327,7 +1327,7 @@ def __init__(self, obj, path_or_buf=None, sep=",", na_rep='', float_format=None,
                                             date_format=date_format,
                                             quoting=self.quoting)
             else:
-                cols = np.asarray(list(cols))
+                cols = list(cols)
             self.obj = self.obj.loc[:, cols]
 
         # update columns to include possible multiplicity of dupes
@@ -1339,7 +1339,7 @@ def __init__(self, obj, path_or_buf=None, sep=",", na_rep='', float_format=None,
                                         date_format=date_format,
                                         quoting=self.quoting)
         else:
-            cols = np.asarray(list(cols))
+            cols = list(cols)
 
         # save it
         self.cols = cols

pandas/tests/test_frame.py

@@ -39,6 +39,7 @@
 from pandas.util.misc import is_little_endian
 
 from pandas.util.testing import (assert_almost_equal,
+                                 assert_equal,
                                  assert_numpy_array_equal,
                                  assert_series_equal,
                                  assert_frame_equal,
@@ -6989,6 +6990,15 @@ def test_to_csv_no_index(self):
             result = read_csv(path)
             assert_frame_equal(df,result)
 
+    def test_to_csv_with_mix_columns(self):
+        #GH11637, incorrect output when a mix of integer and string column
+        # names passed as columns parameter in to_csv
+
+        df = DataFrame({0: ['a', 'b', 'c'],
+                        1: ['aa', 'bb', 'cc']})
+        df['test'] = 'txt'
+        assert_equal(df.to_csv(), df.to_csv(columns=[0, 1, 'test']))
+
     def test_to_csv_headers(self):
         # GH6186, the presence or absence of `index` incorrectly
         # causes to_csv to have different header semantics.