Python计算分段函数

Question

当x=0时，f（x）＝1
当x＞0时，f（x）＝x+1
当x＜0时，f（x）＝x^2+1

Answer 1

x = int(input("请输入x"))
if(x==0):
    y=1
elif(x>0):
    y=x+1
else:
    y=x*x+1
print(y)