Добро пожаловать в StackOverflow!
Вам нужна кусочная функция. Вот краткий фрагмент, который поможет вам набрать скорость.
from sympy import Piecewise # import the Piecewise function
p = Piecewise((x, x < 0), (x**2+2, x > 0)) # the general form is: Piecewise((branch, condition), (branch, condition), ...)
p # display the equation
Это вывод:
![enter image description here](https://i.stack.imgur.com/DzPQB.jpg)