用Python解决非线性方程组(scipy.optimize.fsolve)

时间:2016-11-24 09:59:42

标签: python numpy scipy

我正在尝试解决以下简单的非线性方程组(Source(second example)):

(I) y - x^2 = 7 - 5x
(II) 4y - 8x = -21

应该只有一个解决方案(x = 3.5,y = 1.75)。

我目前使用scipy堆栈的方法如下:

from scipy.optimize import fsolve

def equations(p):
    x, y = p
    return (y - x**2 -7 + 5*x, 4*y - 8*x + 21)

x, y =  fsolve(equations, (5, 5))

print(equations((x, y)))

并产生以下结果(不是结果):

(0.0, 0.0)

我已经尝试了不同的初始估算值,但它没有提供正确的解决方案。

我的方法出了什么问题?我错过了什么吗?

提前致谢!

1 个答案:

答案 0 :(得分:5)

这完全没问题:

In [1]: %paste
from scipy.optimize import fsolve

def equations(p):
    x, y = p
    return (y - x**2 -7 + 5*x, 4*y - 8*x + 21)

x, y =  fsolve(equations, (5, 5))

print(equations((x, y)))

## -- End pasted text --
(0.0, 0.0)

In [2]: x
Out[2]: 3.5000000414181831

In [3]: y
Out[3]: 1.7500000828363667

equations(x, y)(0, 0)表示y - x**2 -7 + 5*x4*y - 8*x + 21均为0。

也许你感到困惑并且意味着print(x, y)而不是print(equations(x, y))