fix to_latex for multiindex

Author: corbanvilla

doc/source/whatsnew/v2.3.0.rst

@@ -48,6 +48,8 @@ These are bug fixes that might have notable behavior changes.
 
 .. _whatsnew_230.notable_bug_fixes.notable_bug_fix1:
 
+- The `to_latex` is now fixed for hidden multiindex dataframes (:issue:`52218`)
+
 notable_bug_fix1
 ^^^^^^^^^^^^^^^^
 

pandas/io/formats/style_render.py

@@ -869,6 +869,8 @@ def _translate_latex(self, d: dict, clines: str | None) -> None:
         """
         index_levels = self.index.nlevels
         visible_index_level_n = index_levels - sum(self.hide_index_)
+        if visible_index_level_n == 0:
+            visible_index_level_n = 1
         d["head"] = [
             [
                 {**col, "cellstyle": self.ctx_columns[r, c - visible_index_level_n]}

pandas/tests/io/formats/test_to_latex.py

@@ -1405,3 +1405,85 @@ def test_to_latex_multiindex_multirow(self):
             """
         )
         assert result == expected
+
+    def test_to_latex_multiindex_format_single_index_hidden(self):
+        # GH 52218
+        df = pd.DataFrame({
+            "A": [1, 2],
+            "B": [4, 5],
+        })
+        result = (
+            df.style.hide(axis="index")
+            .map_index(lambda v: "textbf:--rwrap;", axis="columns")
+            .to_latex()
+        )
+        expected = _dedent(r"""
+            \begin{tabular}{rr}
+            \textbf{A} & \textbf{B} \\
+            1 & 4 \\
+            2 & 5 \\
+            \end{tabular}
+            """
+        )
+        assert result == expected
+
+    def test_to_latex_multiindex_format_triple_index_two_hidden(self):
+        # GH 52218
+        arrays = [
+            ["A", "A", "B", "B"],
+            ["one", "two", "one", "two"],
+            ["x", "x", "y", "y"],
+        ]
+        index = pd.MultiIndex.from_arrays(arrays, names=["Level 0", "Level 1", "Level 2"])
+        df = pd.DataFrame(
+            [[0, 0, 0], [0, 0, 0], [0, 0, 0], [0, 0, 0]], 
+            index=index, 
+            columns=["C1", "C2", "C3"]
+        )
+        result = (
+            df.style.hide(axis="index", level=[0, 1])
+            .map_index(lambda v: "textbf:--rwrap;", axis="columns")
+            .to_latex()
+        )
+        expected = _dedent(r"""
+            \begin{tabular}{lrrr}
+             & \textbf{C1} & \textbf{C2} & \textbf{C3} \\
+            Level 2 &  &  &  \\
+            x & 0 & 0 & 0 \\
+            x & 0 & 0 & 0 \\
+            y & 0 & 0 & 0 \\
+            y & 0 & 0 & 0 \\
+            \end{tabular}
+            """
+        )
+        assert result == expected
+
+    def test_to_latex_multiindex_format_triple_index_all_hidden(self):
+        # GH 52218
+        arrays = [
+            ["A", "A", "B", "B"],
+            ["one", "two", "one", "two"],
+            ["x", "x", "y", "y"],
+        ]
+        index = pd.MultiIndex.from_arrays(arrays, names=["Level 0", "Level 1", "Level 2"])
+        df = pd.DataFrame(
+            [[0, 0, 0], [0, 0, 0], [0, 0, 0], [0, 0, 0]], 
+            index=index, 
+            columns=["C1", "C2", "C3"]
+        )
+        result = (
+            df.style.hide(axis="index", level=[0, 1, 2])
+            .map_index(lambda v: "textbf:--rwrap;", axis="columns")
+            .to_latex()
+        )
+        expected = _dedent(r"""
+            \begin{tabular}{rrr}
+            \textbf{C1} & \textbf{C2} & \textbf{C3} \\
+            0 & 0 & 0 \\
+            0 & 0 & 0 \\
+            0 & 0 & 0 \\
+            0 & 0 & 0 \\
+            \end{tabular}
+            """
+        )
+        assert result == expected