python编写求二次方程函数

编写求二次方程函数，求大家帮忙看一下，第一次接触，什么都不懂。

你好同学，代码供参考：

import math
def quadratic_equation(arg):
    a,b,c=arg
    if(a==0 and b~=0):
        res=-c/b
    elif(a==0 and b==0):
        res = "无解"
    d = b**2-4*a*c
    if(d>0):
        res = [(-b+math.sqrt(d))/(2*a), (-b-math.sqrt(d))/(2*a)]
    elif(d==0):
        res = -b/(2*a)
    else:
        res = "无解"
    return res

如有帮助，还望给个采纳哟谢谢啦

def quadratic_equation(arg):
    a,b,c=arg
    d=b*b-4*a*c
    if d<0:
        res='无解'
    else:
        x1=(-b+math.sqrt(d))/(2*a)
        x2=(-b-math.sqrt(d))/(2*a)
        res=[x1,x2]
    return res

 有帮助请采纳