Accept range for list-requiring kwargs in pd.read_csv

pandas/io/parsers.py

@@ -943,6 +943,8 @@ def _clean_options(self, options, engine):
         if _is_index_col(index_col):
             if not isinstance(index_col, (list, tuple, np.ndarray)):
                 index_col = [index_col]
+        if PY3 and isinstance(index_col, range):
+            index_col = list(index_col)
         result['index_col'] = index_col
 
         names = list(names) if names is not None else names
@@ -1191,6 +1193,11 @@ def __init__(self, kwds):
 
         # validate header options for mi
         self.header = kwds.get('header')
+        if PY3:
+            if isinstance(self.header, range):
+                self.header = list(self.header)
+            if isinstance(self.index_col, range):
+                self.index_col = list(self.index_col)
         if isinstance(self.header, (list, tuple, np.ndarray)):
             if not all(map(is_integer, self.header)):
                 raise ValueError("header must be integer or list of integers")
@@ -1213,7 +1220,6 @@ def __init__(self, kwds):
                         is_integer(self.index_col)):
                     raise ValueError("index_col must only contain row numbers "
                                      "when specifying a multi-index header")
-
         # GH 16338
         elif self.header is not None and not is_integer(self.header):
             raise ValueError("header must be integer or list of integers")

pandas/tests/io/parser/index_col.py

@@ -10,7 +10,7 @@
 
 import pandas.util.testing as tm
 
-from pandas import DataFrame, Index, MultiIndex
+from pandas import DataFrame, Index, MultiIndex, date_range
 from pandas.compat import StringIO
 
 
@@ -141,3 +141,20 @@ def test_empty_with_index_col_false(self):
         result = self.read_csv(StringIO(data), index_col=False)
         expected = DataFrame([], columns=['x', 'y'])
         tm.assert_frame_equal(result, expected)
+
+    def test_range_index_col(self):
+        cols = MultiIndex.from_arrays([['A', 'B', 'C'], ['foo', 'bar', 'baz']])
+        index = date_range('2016-01-02', periods=3, freq='D')
+        data = [[1, 2, 3], [4, 5, 6], [7, 8, 9]]
+        df = DataFrame(data, index=index, columns=cols)
+
+        data = (",A,B,C\n"
+                ",foo,bar,baz\n"
+                "2016-01-02,1,2,3\n"
+                "2016-01-03,4,5,6\n"
+                "2016-01-04,7,8,9"
+                )
+        res = self.read_csv(StringIO(df.to_csv()),
+                            index_col=range(1),
+                            header=range(2))
+        tm.assert_frame_equal(df, res)