Números complexos em python?

Estou fazendo uma simples função de cálculo de bhaskara. Porém quando calculo a raiz de delta negativo o programa para (como esperado). Então como faço pra resolver esse problema com números complexos?
Desde já agradeço
Obs: código abaixo

def calcular_baskhara(a,b,c):
    z=(b ** 2) - (4 * a * c)
    if z<0:

        delta=(sqrtcomplex((z)))
        z1 = complex((-b / (2 * a)) + (delta / (2 * a)))
        z2 = complex((-b / (2 * a)) - (delta / (2 * a)))
        return (z1, z2)
    else:
        delta=sqrt(z)
    if delta==0:
        x=-b/(2*a)
        return x
    elif delta>0:
        x1=(-b+delta)/(2*a)
        x2=(-b-delta)/(2*a)
        return (x1,x2)

print(calcular_baskhara(1,-14,50))

use a biblioteca cmath para trabalhar com complexos

1 curtida

Obrigado pela dica. Consegui resolver

Oi matheususb pode postar aqui o programa completo??
Estou passando por dificuldades de resolver o mesmo