set index_label to False when header is False

Author: charlesdong1991

pandas/io/formats/csvs.py

@@ -51,6 +51,8 @@ def __init__(self, obj, path_or_buf=None, sep=",", na_rep='',
         self.header = header
         self.index = index
         self.index_label = index_label
+        if not (self.header or self.index_label):
+            self.index_label = False
         self.mode = mode
         if encoding is None:
             encoding = 'ascii' if compat.PY2 else 'utf-8'

pandas/tests/frame/test_to_csv.py

@@ -119,7 +119,7 @@ def test_to_csv_from_csv3(self):
             df2 = DataFrame(np.random.randn(3, 1))
 
             df1.to_csv(path)
-            df2.to_csv(path, mode='a', header=False, index_label=False)
+            df2.to_csv(path, mode='a', header=False)
 
             xp = pd.concat([df1, df2])
             rs = pd.read_csv(path, index_col=0)
@@ -765,7 +765,7 @@ def test_to_csv_wide_frame_formatting(self):
         # Issue #8621
         df = DataFrame(np.random.randn(1, 100010), columns=None, index=None)
         with ensure_clean() as filename:
-            df.to_csv(filename, header=False, index=False, index_label=False)
+            df.to_csv(filename, header=False, index=False)
             rs = read_csv(filename, header=None)
             assert_frame_equal(rs, df)
 

pandas/tests/io/formats/test_to_csv.py

@@ -30,7 +30,7 @@ def test_to_csv_with_single_column(self):
 1.0
 """
         with tm.ensure_clean('test.csv') as path:
-            df1.to_csv(path, header=None, index=None, index_label=False)
+            df1.to_csv(path, header=None, index=None)
             with open(path, 'r') as f:
                 assert f.read() == expected1
 
@@ -40,7 +40,7 @@ def test_to_csv_with_single_column(self):
 ""
 """
         with tm.ensure_clean('test.csv') as path:
-            df2.to_csv(path, header=None, index=None, index_label=False)
+            df2.to_csv(path, header=None, index=None)
             with open(path, 'r') as f:
                 assert f.read() == expected2
 
@@ -490,7 +490,7 @@ def test_to_csv_write_to_open_file(self):
         with tm.ensure_clean('test.txt') as path:
             with open(path, 'w') as f:
                 f.write('manual header\n')
-                df.to_csv(f, header=None, index=None, index_label=False)
+                df.to_csv(f, header=None, index=None)
             with open(path, 'r') as f:
                 assert f.read() == expected
 
@@ -507,7 +507,7 @@ def test_to_csv_write_to_open_file_with_newline_py3(self):
         with tm.ensure_clean('test.txt') as path:
             with open(path, 'w', newline='') as f:
                 f.write('manual header\n')
-                df.to_csv(f, header=None, index=None, index_label=False)
+                df.to_csv(f, header=None, index=None)
 
             with open(path, 'rb') as f:
                 assert f.read() == bytes(expected, 'utf-8')
@@ -525,7 +525,7 @@ def test_to_csv_write_to_open_file_with_newline_py2(self):
         with tm.ensure_clean('test.txt') as path:
             with open(path, 'wb') as f:
                 f.write('manual header\n')
-                df.to_csv(f, header=None, index=None, index_label=False)
+                df.to_csv(f, header=None, index=None)
 
             with open(path, 'rb') as f:
                 assert f.read() == expected

pandas/tests/series/test_io.py

@@ -49,11 +49,11 @@ def test_to_csv_deprecation(self, arg, datetime_series):
         # see gh-19715
         with ensure_clean() as path:
             if arg == "path":
-                kwargs = dict(path=path, header=False, index_label=False)
+                kwargs = dict(path=path, header=False)
             elif arg == "header":
-                kwargs = dict(path_or_buf=path, index_label=False)
+                kwargs = dict(path_or_buf=path)
             else:  # Both discrepancies match.
-                kwargs = dict(path=path, index_label=False)
+                kwargs = dict(path=path)
 
             with tm.assert_produces_warning(FutureWarning):
                 datetime_series.to_csv(**kwargs)
@@ -65,7 +65,7 @@ def test_to_csv_deprecation(self, arg, datetime_series):
     def test_from_csv(self, datetime_series, string_series):
 
         with ensure_clean() as path:
-            datetime_series.to_csv(path, header=False, index_label=False)
+            datetime_series.to_csv(path, header=False)
             ts = self.read_csv(path)
             assert_series_equal(datetime_series, ts, check_names=False)
 
@@ -82,7 +82,7 @@ def test_from_csv(self, datetime_series, string_series):
             ts_h = self.read_csv(path, header=0)
             assert ts_h.name == "ts"
 
-            string_series.to_csv(path, header=False, index_label=False)
+            string_series.to_csv(path, header=False)
             series = self.read_csv(path)
             assert_series_equal(string_series, series, check_names=False)
 
@@ -109,22 +109,21 @@ def test_to_csv(self, datetime_series):
         import io
 
         with ensure_clean() as path:
-            datetime_series.to_csv(path, header=False, index_label=False)
+            datetime_series.to_csv(path, header=False)
 
             with io.open(path, newline=None) as f:
                 lines = f.readlines()
             assert (lines[1] != '\n')
 
-            datetime_series.to_csv(path, index=False, header=False,
-                                   index_label=False)
+            datetime_series.to_csv(path, index=False, header=False)
             arr = np.loadtxt(path)
             assert_almost_equal(arr, datetime_series.values)
 
     def test_to_csv_unicode_index(self):
         buf = StringIO()
         s = Series([u("\u05d0"), "d2"], index=[u("\u05d0"), u("\u05d1")])
 
-        s.to_csv(buf, encoding="UTF-8", header=False, index_label=False)
+        s.to_csv(buf, encoding="UTF-8", header=False)
         buf.seek(0)
 
         s2 = self.read_csv(buf, index_col=0, encoding="UTF-8")
@@ -134,8 +133,7 @@ def test_to_csv_float_format(self):
 
         with ensure_clean() as filename:
             ser = Series([0.123456, 0.234567, 0.567567])
-            ser.to_csv(filename, float_format="%.2f", header=False,
-                       index_label=False)
+            ser.to_csv(filename, float_format="%.2f", header=False)
 
             rs = self.read_csv(filename)
             xp = Series([0.12, 0.23, 0.57])
@@ -154,7 +152,7 @@ def test_to_csv_path_is_none(self):
         # Series.to_csv() was returning None, inconsistent with
         # DataFrame.to_csv() which returned string
         s = Series([1, 2, 3])
-        csv_str = s.to_csv(path_or_buf=None, header=False, index_label=False)
+        csv_str = s.to_csv(path_or_buf=None, header=False)
         assert isinstance(csv_str, str)
 
     @pytest.mark.parametrize('s,encoding', [