ENH: add math mode to formatter escape=latex-math II

Author: natmokval

pandas/io/formats/style_render.py

@@ -989,9 +989,8 @@ def format(
             Use 'latex' to replace the characters ``&``, ``%``, ``$``, ``#``, ``_``,
             ``{``, ``}``, ``~``, ``^``, and ``\`` in the cell display string with
             LaTeX-safe sequences.
-            Use 'latex-math' to replace the characters ``&``, ``%``, ``#``, ``_``,
-            ``{``, ``}``, ``~``, ``^``, and ``\``
-            in the cell display string with LaTeX-safe sequences.
+            All characters remain in the string as is. LaTeX math mode starts with
+            the character ``$`` or ``\(`` and ends with ``$`` or ``\)``.
             Escaping is done before ``formatter``.
 
             .. versionadded:: 1.3.0
@@ -1108,16 +1107,30 @@ def format(
         <td .. >NA</td>
         ...
 
-        Using a ``formatter`` with LaTeX ``escape``.
+        Using a ``formatter`` with LaTeX ``escape`` in 'latex' mode.
 
-        >>> df = pd.DataFrame([["123"], ["~ ^"], ["$%#"]])
+        >>> df = pd.DataFrame([["123"], ["~ ^"], ["%#"]])
         >>> df.style.format("\\textbf{{{}}}", escape="latex").to_latex()
         ...  # doctest: +SKIP
         \begin{tabular}{ll}
-        {} & {0} \\
+         & 0 \\
         0 & \textbf{123} \\
         1 & \textbf{\textasciitilde \space \textasciicircum } \\
-        2 & \textbf{\$\%\#} \\
+        2 & \textbf{\%\#} \\
+        \end{tabular}
+
+        Using a ``formatter`` with LaTeX ``escape`` in 'latex-math' mode.
+
+        >>> df = pd.DataFrame([[r"$ \alpha = \frac{\beta}{\zeta^2} $", \
+        r"$ \$ \ \ \# \{\} $"],[r"\( \sum_{i=1}^{10} a_i \)", \
+        r"\( \int_0^\infty \mathrm{e}^{-x}\,\mathrm{d}x \)"]])
+        >>> print(df.style.format(escape="latex-math").to_latex())
+        ...  # doctest: +SKIP
+        \begin{tabular}{lll}
+         & 0 & 1 \\
+        0 & $ \alpha = \frac{\beta}{\zeta^2} $ & $ \$ \ \ \# \{\} $ \\
+        1 & \( \sum_{i=1}^{10} a_i \) & \( \int_0^\infty \mathrm{e}^{-x}\, \
+        \mathrm{d}x \) \\
         \end{tabular}
 
         Pandas defines a `number-format` pseudo CSS attribute instead of the `.format`
@@ -2354,31 +2367,17 @@ def _escape_latex(s):
 
 def _escape_latex_math(s):
     r"""
-    Replace the characters ``&``, ``%``, ``#``, ``_``, ``{``, ``}``,
-    ``~``, ``^``, and ``\`` in the string with LaTeX-safe sequences.
+    All characters in the string are preserved.
+
+    Note that LaTeX math mode starts with the character ``$`` or ``\(``
+    and ends with ``$`` or ``\)``.
 
     Parameters
     ----------
     s : str
-        Input to be escaped
 
     Return
     ------
-    str :
-        Escaped string
+    str
     """
-    return (
-        s.replace("\\", "ab2§=§8yz")  # rare string for final conversion: avoid \\ clash
-        .replace("ab2§=§8yz ", "ab2§=§8yz\\space ")  # since \backslash gobbles spaces
-        .replace("&", "\\&")
-        .replace("%", "\\%")
-        .replace("#", "\\#")
-        .replace("_", "\\_")
-        .replace("{", "\\{")
-        .replace("}", "\\}")
-        .replace("~ ", "~\\space ")  # since \textasciitilde gobbles spaces
-        .replace("~", "\\textasciitilde ")
-        .replace("^ ", "^\\space ")  # since \textasciicircum gobbles spaces
-        .replace("^", "\\textasciicircum ")
-        .replace("ab2§=§8yz", "\\textbackslash ")
-    )
+    return s