将包含 Heaviside 阶跃函数的 sympy 输出函数的类型设置从 Theta(t) 更改为 u(t)

Question

运行 Jupyter 笔记本或 Co-lab 笔记本并返回包含阶跃函数的函数时，它将阶跃函数显示为 Theta(t) 而不是 u(t)。虽然 Theta(t) 很常见，但在向刚接触该领域的学生教授阶跃函数概念时，切换符号可能会导致一些不必要的认知负担。

可以将排版从 Theta(t) 更改为 u(t) 吗？

我尝试了尽可能多的标题变体搜索，从排版的特定更改到通用，但我找不到任何相关资源。

Answer 1

您可以修改打印机，或者，如果这只是为了显示目的，您可以创建一个自定义函数，用函数

Heaviside

替换

u

：

>>> from sympy import Heaviside, Function
>>> from sympy.abc import x
>>> Heaviside(x).replace(lambda x: isinstance(x, Heaviside),
...                      lambda x: Function('u')(x.args[0]))
u(x)

Heaviside 的第 2 个参数确定值为 0；为了简单起见，替换只是忽略该参数；否则使用

(*x.args)