BUG: allow itertuples to work with duplicate columns

Author: cpcloud

pandas/core/frame.py

@@ -818,7 +818,9 @@ def itertuples(self, index=True):
         arrays = []
         if index:
             arrays.append(self.index)
-        arrays.extend(self[k] for k in self.columns)
+
+        # use integer indexing because of possible duplicate column names
+        arrays.extend(self.iloc[:, k] for k in xrange(len(self.columns)))
         return izip(*arrays)
 
     iterkv = iteritems

pandas/tests/test_frame.py

@@ -3951,6 +3951,10 @@ def test_itertuples(self):
         for tup in df.itertuples(index=False):
             self.assert_(isinstance(tup[1], np.integer))
 
+        df = DataFrame(data={"a": [1, 2, 3], "b": [4, 5, 6]})
+        dfaa = df[['a', 'a']]
+        self.assertEqual(list(dfaa.itertuples()), [(0, 1, 1), (1, 2, 2), (2, 3, 3)])
+
     def test_len(self):
         self.assertEqual(len(self.frame), len(self.frame.index))