BUG: Fixed DataFrame info() for duplicated columns, #11761

Author: yoongkang

doc/source/whatsnew/v0.18.0.txt

@@ -169,6 +169,7 @@ Bug Fixes
 
 - Bug in ``Timedelta.round`` with negative values (:issue:`11690`)
 - Bug in ``.loc`` against ``CategoricalIndex`` may result in normal ``Index`` (:issue:`11586`)
+- Bug in ``DataFrame.info`` when duplicated column names exist (:issue:`11761`)
 
 
 

pandas/core/frame.py

@@ -1655,7 +1655,7 @@ def _verbose_repr():
 
             dtypes = self.dtypes
             for i, col in enumerate(self.columns):
-                dtype = dtypes[col]
+                dtype = dtypes.iloc[i]
                 col = com.pprint_thing(col)
 
                 count = ""

pandas/tests/test_frame.py

@@ -7559,6 +7559,18 @@ def test_info_duplicate_columns(self):
                           columns=['a', 'a', 'b', 'b'])
         frame.info(buf=io)
 
+    def test_info_duplicate_columns_shows_correct_dtypes(self):
+        # GH11761
+        io = StringIO()
+
+        frame = DataFrame([[1, 2.0]],
+                          columns=['a', 'a'])
+        frame.info(buf=io)
+        io.seek(0)
+        lines = io.readlines()
+        self.assertEqual('a    1 non-null int64\n', lines[3])
+        self.assertEqual('a    1 non-null float64\n', lines[4])
+
     def test_info_shows_column_dtypes(self):
         dtypes = ['int64', 'float64', 'datetime64[ns]', 'timedelta64[ns]',
                   'complex128', 'object', 'bool']