求解一元二次方程组

一个简单的求解一元二次方程组的python代码，作为刚学不久的新手来说当然会有很多不足之处，希望大家多多指正!

print("请按顺序输入一元二次方程的各个系数a,b,c:")
y=input("各个系数为（中间用英文逗号隔开）")
y=y.split(",")#把用户的输入值转换成一个list
def solution(y):
        if int(y[0])==0:
                print("该方程组不是一元二次方程组，请重新输入")
                return "**"
        elif int(y[0])==1:
                 print("该方程组为x^2+({b})x+({c})".format(b=int(y[1]),c=int(y[2])))
        else:
                print("该方程组为{a}x^2+({b})x+({c})".format(a=int(y[0]),b=int(y[1]),c=int(y[2])))
        deta=int(y[1])*int(y[1])-4*int(y[0])*int(y[2])
        print("请说明在实数域上求解还是在复数域上求解")
        z=input("实数域请输入0，复数域请输入1，请输入...")
        if int(z)==0:
                import math
                if deta<0:
                        print("该方程组在实数范围内无解")
                        return "**"
                else:
                        y1=(-int(y[1])+math.sqrt(deta))/(2*int(y[0]))
                        y2=(-int(y[1])-math.sqrt(deta))/(2*int(y[0]))
                        return y1,y2
        elif int(z)==1:
                        import cmath
                        y1=(-int(y[1])+cmath.sqrt(deta))/(2*int(y[0]))
                        y2=(-int(y[1])-cmath.sqrt(deta))/(2*int(y[0]))
                        return y1,y2
        else:
                print("无法判断在实数域还是在复数域上求解，请重新输入")
                return "**"
x=solution(y)
print("该方程组的两个根分别为{x1},{x2}".format(x1=x[0],x2=x[1]))

运行结果