#!/usr/bin/python import numpy as np a,b,c=np.loadtxt("sv.dat") print a,b,c def root(a,b,c): dt2=b**2-4.*a*c if dt2 > 0.: x1=(-b+np.sqrt(dt2))/(2.*a) x2=(-b-np.sqrt(dt2))/(2.*a) return "kaks ruudu",x1,x2 if dt2 == 0.: x1=x2=-b/(2.*a) return "uks ruut",x1,x2 if dt2 <0.: x1=-b/(2.*a) x2=np.sqrt(-dt2)/(2.*a) return "complex",x1,x2 print root(a,b,c)