{
  "nbformat": 4,
  "nbformat_minor": 0,
  "metadata": {
    "colab": {
      "name": "SympyInColab.ipynb",
      "version": "0.3.2",
      "provenance": [],
      "include_colab_link": true
    },
    "kernelspec": {
      "name": "python3",
      "display_name": "Python 3"
    }
  },
  "cells": [
    {
      "cell_type": "markdown",
      "metadata": {
        "id": "view-in-github",
        "colab_type": "text"
      },
      "source": [
        "<a href=\"https://colab.research.google.com/github/jakevdp/colab-snippets/blob/master/SympyInColab.ipynb\" target=\"_parent\"><img src=\"https://colab.research.google.com/assets/colab-badge.svg\" alt=\"Open In Colab\"/></a>"
      ]
    },
    {
      "metadata": {
        "id": "9G8w79zS5vG4",
        "colab_type": "text"
      },
      "cell_type": "markdown",
      "source": [
        "# Sympy in Colab Example\n",
        "\n",
        "Rendering sympy equations requires MathJax to be available within each cell output. The following is a function that will make this happen."
      ]
    },
    {
      "metadata": {
        "id": "LcGMeyen43QF",
        "colab_type": "code",
        "colab": {
          "base_uri": "https://localhost:8080/",
          "height": 17
        },
        "outputId": "b5619ae5-5584-4869-bfe0-f4d0d5a7fdfe"
      },
      "cell_type": "code",
      "source": [
        "from IPython.display import Math, HTML\n",
        "def enable_sympy_in_cell():\n",
        "  display(HTML(\"<script src='https://cdnjs.cloudflare.com/ajax/libs/\"\n",
        "               \"mathjax/2.7.3/latest.js?config=default'></script>\"))\n",
        "get_ipython().events.register('pre_run_cell', enable_sympy_in_cell)"
      ],
      "execution_count": 3,
      "outputs": [
        {
          "output_type": "display_data",
          "data": {
            "text/html": [
              "<script src='https://cdnjs.cloudflare.com/ajax/libs/mathjax/2.7.3/latest.js?config=default'></script>"
            ],
            "text/plain": [
              "<IPython.core.display.HTML object>"
            ]
          },
          "metadata": {
            "tags": []
          }
        }
      ]
    },
    {
      "metadata": {
        "id": "pu1gbpMi5_pA",
        "colab_type": "text"
      },
      "cell_type": "markdown",
      "source": [
        "After registering this hook, SymPy rendering will work correctly:\n"
      ]
    },
    {
      "metadata": {
        "id": "llIM2iiJ2k8O",
        "colab_type": "code",
        "outputId": "4fc1bd54-d16d-4231-ed59-2f5b2cf9dd11",
        "colab": {
          "base_uri": "https://localhost:8080/",
          "height": 89
        }
      },
      "cell_type": "code",
      "source": [
        "import sympy\n",
        "from sympy import *\n",
        "init_printing()\n",
        "\n",
        "x, y, z = symbols('x y z')\n",
        "Integral(sqrt(1/x), (x, 0, oo))"
      ],
      "execution_count": 4,
      "outputs": [
        {
          "output_type": "display_data",
          "data": {
            "text/html": [
              "<script src='https://cdnjs.cloudflare.com/ajax/libs/mathjax/2.7.3/latest.js?config=default'></script>"
            ],
            "text/plain": [
              "<IPython.core.display.HTML object>"
            ]
          },
          "metadata": {
            "tags": []
          }
        },
        {
          "output_type": "display_data",
          "data": {
            "text/html": [
              "<script src='https://cdnjs.cloudflare.com/ajax/libs/mathjax/2.7.3/latest.js?config=default'></script>"
            ],
            "text/plain": [
              "<IPython.core.display.HTML object>"
            ]
          },
          "metadata": {
            "tags": []
          }
        },
        {
          "output_type": "execute_result",
          "data": {
            "image/png": "iVBORw0KGgoAAAANSUhEUgAAAGwAAAAnCAYAAADq+a1iAAAABHNCSVQICAgIfAhkiAAABRZJREFU\neJzt22msXVMUwPHf05aHkn5o1ZCqmX6gVWkNQR9qVtSQRqMiUYQQIWnEFI2gUnxAzCGNIUhTETV/\nIIaKEko/tJrGPFTNSuigng9rv973rjud895998a7/6Q575y1117r3H32Xmuvc0qLAc9srMATGIbj\nsARv46DGudWiFJNxH4ZiFl7FRxiDXfAG2hrl3ECjDYOqtJmFg7udL8J53c4fwIg+9mtAsVkNbdox\nBz/jN8xVfuBWiCUQJmELXIEdsDX2wU85/GxDZ5P+azoe9V8nry7Ttk0siV/iPbEMzsBX+AxTc/ow\nUczeFlXYDv+IQboUZ6a/v+pnP24QcXDAM7iKfJxCkvC4WBJfwA/YCn/Wz7UejMHyfrLV1FQbsOHp\nuFbEMDixfu6UZEd80882m5ZqScc26bim3o5U4CQ810D7TUW1ARuajn/U25EKTMLrDbTf14wSecDT\neZRrnWG/5+m8D2jH39hQRn4G7sKbYhXoxGM19DsSG5Nuf3NAOn6QR7laDGv0DDtKVEvKcS3GCv++\nFvu8WjhFPKy5nvJeMj4d38+j3Owz7HiRlZbjcuyFbXFRhn6nig38G/ldy02vZlizx7ARYgtRjtew\nUraKw7Y4EgvFstjFK6mf04vat2Fekt1So43BuAxLRYb9Ba5MfY3Ht1hdB7ueTQoP16qQg33FzRUz\nVvmKSik61BbDpqd2J5ewtxHL9Cy93Z7a31+jH5srDMIS3Cp+v7/wYLr+bB3sIuJHJ+7MopSBSeKm\nvi4huwb7ZeirQ20DNl+sGO0lZPNSH+em86vT+VNqq7tSGJTr9HwzcbhCaW92Hewi6oGduCmLUkaG\nYZ1CMO7iyYz9dKg+YO0iHs8vIx8lZsJnuCT195KYNbUwMeksLCNfluRT+tjuJpYn5auyKmbkZT2f\nuhGyp9wdqg/YlNRmeoU2cxRmwiJRgquVrkJ5uRe1byb5TnntNkuWuFDPp+4ElbPDvJyG9Xi+Qpvu\nSc55stVLjxHZ5+Iy8t3wvdKltt7Y3cSveq6t9WK0eCvQ9eTNUzrGVKJD5Rk2CD/ixQp9TE9+rEp9\n3ZvBfnvSKZeuj0/yUg9ib+z24G+lU856sBQXYggeyaHfofKAHZHkF5SRnyBm31KxJH8sKix712h/\niPi9SiVQsCDZv7GP7W5iS4U19disyjm4SRR5J2NmDv0OlQfsTpE+jywhO1QsQZ+Kt+NE2asTz2Tw\noSvmFycVVyr8lt1f4vaVXcTLyy4jh2RVzsGBIlO6X8H5apwqls95IqvqxCfdrt2W2rWJt+Bvlehj\nnFj6V2H3IllXlnxYjf6ck9qvF6vEXLwr7uvLJBtdB7uIANk1YFn2Q3lpw3d4J4PObJW/t/g8tZuQ\nzq8o0t8j2fxF6XucnPSy+HSZmDEbRDVjgYhfX4gYWi+7xirc+K5ZFHvBQ7i+Dv3erH/voyEcqjBg\nw6u0vVhs+taKKnSmqdyNCdgzp24lluPDOvTbUCaKlPOedH6WGKwNKu/XpqU254tvL+4SpZ+d6+Zp\nCxRi1p/YX2wuO0UArMRiUT/rzkqxc29RZxb6b+CutAfbXOw9ziy6frf/12v9pqF4qZshPmdbI76J\nP0VkOeUYLioIq4uur8b2feRji24UfyLwK85uhCMtaiPTu5YS/Kh09WCk2Ge06GN6O2DrRRp/dNH1\no8X/B2vRhEwTAzdTpPV3iLR+dCWlFo3lYlEGWidm3OEN9aZFixYtWgwE/gUvrYr/Q6Dv9AAAAABJ\nRU5ErkJggg==\n",
            "text/latex": "$$\\int_{0}^{\\infty} \\sqrt{\\frac{1}{x}}\\, dx$$",
            "text/plain": [
              "∞           \n",
              "⌠           \n",
              "⎮     ___   \n",
              "⎮    ╱ 1    \n",
              "⎮   ╱  ─  dx\n",
              "⎮ ╲╱   x    \n",
              "⌡           \n",
              "0           "
            ]
          },
          "metadata": {
            "tags": []
          },
          "execution_count": 4
        }
      ]
    },
    {
      "metadata": {
        "id": "3gOggSmO8ZaJ",
        "colab_type": "code",
        "colab": {}
      },
      "cell_type": "code",
      "source": [
        ""
      ],
      "execution_count": 0,
      "outputs": []
    }
  ]
}