How to write a class which naturally produce a latex output in jupyter notebook

I asked this question first on stackoverflow, they rightfully recommended me to ask it here, this is it :slight_smile:

If we put these 3 lines in a jupyter notebook and run them, we get a nicely formatted equation with a MathJax/Latex format, which is nice.

import sympy

a, b, c = sympy.symbols('a b c')
(- b - sympy.sqrt(b**2 - 4*a*c)) / (2*a)

If we have a look at the source of the notebook, we can see that the output saved is made of two parts text/plain and text/latex :

"outputs": [
 {
  "data": {
   "text/latex": [
    "$\\displaystyle \\frac{- b - \\sqrt{- 4 a c + b^{2}}}{2 a}$"
   ],
   "text/plain": [
    "(-b - sqrt(-4*a*c + b**2))/(2*a)"
   ]
  },
  "execution_count": 22,
  "metadata": {},
  "output_type": "execute_result"
 }
],
"source": [
 "import sympy\n",
 "\n",
 "a, b, c = sympy.symbols('a b c')\n",
 "(-b - sympy.sqrt(b**2 - 4*a*c)) / (2*a)"
]

Even if I know how to print in a latex form (cf. another stackoverflow page), I would like to make a custom object behave as a sympy one, and able to be displayed naturally as latex math in a jupyter notebook (and if I could fill both the text/latex and text/plain fields it would be nice !). If anyone had a clue on how to do it :slight_smile:

The key resource for the plumbing is in the Integrating your objects with IPython section of the docs.

Here’s an example:

class Foo:
    def __init__(self, x, y):
        self.x = x
        self.y = y

    def _repr_mimebundle_(self, **kwargs):
        return {
            "text/plain": f"x={self.x}",
            "text/latex": f"$x = {self.x}\\\\y = {self.y}$"
        }

foo = Foo(1, 2)
foo

yields:

4 Likes

Thank you so much for this perfect answer and the link to the documentation!

1 Like