import math
def separar(ecu):
"""Esta funcion separara cada valor de la ecuacion"""
ecu = ecu.replace('t', '') #Se reemplazan los tabulados
nueva = ecu.split('x^2') #quitamos los valores con x^2 y x
nueva += nueva[1].split('x')
#Se elimina el indice 1 de la lista y lo dejamos
#con los datos necesarios
nueva.remove(nueva[1])
#Ahora recorremos la lista para no dejar signos sin valores
#creamos un contador para el indice
cont = 0
for h in nueva:
if h == '+' or h == '-' or h == '': #Si es un signo +/- o sino es nada
h += '1'
nueva.remove(nueva[cont])
nueva.insert(cont, h)
cont += 1
x = cuadratica(int(nueva[0]), int(nueva[1]), int(nueva[2]))
return x
##Esta es tú función
def cuadratica(a, b, c):
"""Vamos a obtener el valor de X por medio de la ecuacion cuadratica"""
raiz = math.sqrt((b**2)-(4*a*c))
x1 = (-b + raiz)/(2*a)
x2 = (-b - raiz)/(2*a)
lista = [str(x1), str(x2)]
return lista
archi = open('quadratic.txt').readlines()
suma = 0
for h in archi:
z = separar(h)
for l in z:
#Si al final sale un .0 entonces lo volvemos entero y se suma
if l[-2:] == '.0':
suma += int(l.replace('.0', ''))
print suma