BUG: icol duplicate columns with integer sequence failure. close pandas-dev#2228

Author: wesm

pandas/core/frame.py

@@ -1726,7 +1726,11 @@ def icol(self, i):
         else:
             label = self.columns[i]
             if isinstance(label, Index):
-                return self.reindex(columns=label)
+                if self.columns.inferred_type == 'integer':
+                    # XXX re: #2228
+                    return self.reindex(columns=label)
+                else:
+                    return self.ix[:, i]
 
             values = self._data.iget(i)
             return Series.from_array(values, index=self.index, name=label)

pandas/tests/test_frame.py

@@ -1323,6 +1323,10 @@ def test_irow_icol_duplicates(self):
         xp = df.T.ix[0]
         assert_series_equal(rs, xp)
 
+        rs = df.icol([0])
+        xp = df.ix[:, [0]]
+        assert_frame_equal(rs, xp)
+
     def test_iget_value(self):
         for i, row in enumerate(self.frame.index):
             for j, col in enumerate(self.frame.columns):