我的代码是
#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),有时甚至解是错误的(不仅是正/负号),但有时它给出了正确的答案。
如何改进它,以及代码的哪一部分出现问题?
答案 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)
将处理浮点数。)