Merge pull request #7982 from shoyer/fix-to_latex-multicolumn

Fix DataFrame.to_latex() midrule positioning with MultiIndex columns

Author: jreback

doc/source/v0.15.0.txt

@@ -600,7 +600,7 @@ Bug Fixes
 
 - ``Holiday`` bug in Holiday with no offset or observance (:issue:`7987`)
 
-
+- Bug in ``DataFrame.to_latex`` formatting when columns or index is a ``MultiIndex`` (:issue:`7982`).
 
 
 

pandas/core/format.py

@@ -532,7 +532,7 @@ def write(buf, frame, column_format, strcols, longtable=False):
                 buf.write('\\begin{longtable}{%s}\n' % column_format)
                 buf.write('\\toprule\n')
 
-            nlevels = frame.index.nlevels
+            nlevels = frame.columns.nlevels
             for i, row in enumerate(zip(*strcols)):
                 if i == nlevels:
                     buf.write('\\midrule\n')  # End of header

pandas/tests/test_format.py

@@ -2083,6 +2083,31 @@ def test_to_latex(self):
 """
         self.assertEqual(withoutindex_result, withoutindex_expected)
 
+    def test_to_latex_multiindex(self):
+        df = DataFrame({('x', 'y'): ['a']})
+        result = df.to_latex()
+        expected = r"""\begin{tabular}{ll}
+\toprule
+{} &  x \\
+{} &  y \\
+\midrule
+0 &  a \\
+\bottomrule
+\end{tabular}
+"""
+        self.assertEqual(result, expected)
+
+        result = df.T.to_latex()
+        expected = r"""\begin{tabular}{ll}
+\toprule
+{} &  0 \\
+\midrule
+x y &  a \\
+\bottomrule
+\end{tabular}
+"""
+        self.assertEqual(result, expected)
+
     def test_to_latex_escape(self):
         a = 'a'
         b = 'b'