Bug#11637 fix to_csv incorrect output.

Added test case test_to_csv_with_mix_columns

Author: varunkumar-dev

doc/source/whatsnew/v0.17.1.txt

@@ -164,6 +164,7 @@ 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()`` incorrect output when a mix of integer and string column names passed as columns parameter (:issue:`11637`)
 
 
 

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,
@@ -6987,6 +6988,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.