我的二次方程求解程序(Python 3)并没有给出正确的方程

时间:2016-12-10 03:02:37

标签: python python-3.x python-3.5

我的代码是

#Import the module
from math import sqrt

#Using while loop statement to make the program not finish before the user close the program.
while True:

#Print out the introduction message, and get the input value to solve the quadratic equation.
    print("ax^2+bx+c=0의 꼴로 된 방정식을 풀 수 있습니다. a, b, c의 값을 차례대로 입력하세요.")
    a = input("a를 입력하세요 : ")
    b = input("b를 입력하세요 : ")
    c = input("c를 입력하세요 : ")

#Define function that checks whether the input values are natural number or negative number
    def func_num(n):
        if n[0] == '-':
            n = -int(n[1:])
            return n
        else:
            n = int(n)
            return n

#Execute the function for the input value a, b, c
    a = func_num(a); b = func_num(b); c = func_num(c);

#This if statement chekcs whether the solution of the quadratic equation going to be real number or imaginary number.
    if b ** 2 > 4*a*c:
        solution1 = ((sqrt((b ** 2)-(4*a*c)))-b) / (2*a)
        solution2 = (-(sqrt((b ** 2)-(4*a*c)))-b) / (2*a)
    else:
        square_root = sqrt( -(b**2 - 4*a*c) ) + 1j
        solution1 = ( (square_root)  - b  ) / (2*a)
        solution2 = ( -(square_root)  - b  ) / (2*a)

#Prints out the solution of the quadratic equation.
    print("정답은 바로바로... {}, {} 이거다!".format(solution1, solution2))

它并没有给出正确的答案。对于某些方程式,它给出了解的负值(解* -1),有时甚至解是错误的(不仅是正/负号),但有时它给出了正确的答案。

如何改进它,以及代码的哪一部分出现问题?

1 个答案:

答案 0 :(得分:2)

您的公式中存在虚构平方根情况的错误。当您打算乘以1j时,您添加 1j。它应该改为:

square_root = sqrt( -(b**2 - 4*a*c) ) * 1j
                                      ^

另外需要注意的是:Python完全能够计算假想的平方根 - 您只需要在sqrt(复杂数学)包中使用cmath的版本而不是math 1}}包:

import cmath
print(cmath.sqrt(4))
print(cmath.sqrt(-4))

这样,您可以避免处理负数平方根的特殊情况。

最后一项改进:事实证明int函数处理代表负数的字符串就好了(例如,试试int("-5")并且它工作正常),因此您可以替换func_num(a)函数调用只需调用int(a)等(或者更好,float(a)将处理浮点数。)