用 Google Colab，在线上编写和执行 Python 很方便。
又有 Colab AI 协助产生程式。
Python 算数学，输出时的数学式是用纯文字格式的，
时常不容易阅读。
如果要显示较美观直觉的数学式，
可以参考这一篇文章：
在 Colab 笔记本里呈现更美观的数学式
https://aiplusstudy.blogspot.com/2024/02/colab.html

你需要的是LaTeX(x)