Составте программу для вычесления у у= х^2 при х < 2 корень х при 100 > x > =2

Вот программа на языке python2

x = input()

if x < 2:
    y = x ** 2
elif x < 100 and x >= 2:
    y = x ** 0.5

print y