Написать функцию вычисления f(x) в зависимости от значения x в PYTHON

import math

print('Input x, and press enter')

x = float(input())

if x < 1.5:

y = x*math.tan(x) - math.sin(x)

elif 1.5 <= x < 2.5:

y = math.pow(x, 3) + math.sin(x)

else:

y = 3*math.pow(x, 3) + 5

print('f(x) = ' + str(y))