Добро пожаловать в 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
Это вывод: