Python计算分段函数

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

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

if 和 elif， 也可以加上else
差不多就是
result = 0
if x ==0: result = 1
elif x>0: result = x+1
if x<0: result = x^2 +1