scipy.optimize.fsolve 中的输入/输出错误答案

【问题标题】：input/output error in scipy.optimize.fsolvescipy.optimize.fsolve 中的输入/输出错误
【发布时间】：2012-05-09 04:04:24
【问题描述】：

当我在 scipy 中使用 root-finder 时，我似乎遇到了错误。我想知道是否有人能指出我做错了什么。
我要查找其根的函数只是一个简单的示例，并不是特别重要。

如果我使用 scipy 0.9.0 运行此代码：

import numpy as np
from scipy.optimize import fsolve

tmpFunc = lambda xIn: (xIn[0]-4)**2 + (xIn[1]-5)**2 + (xIn[2]-7)**3

x0 = [3,4,5]
xFinal = fsolve(tmpFunc, x0 )

print xFinal

我收到以下错误消息：

Traceback (most recent call last):
  File "tmpStack.py", line 7, in <module>
    xFinal = fsolve(tmpFunc, x0 )
  File "/usr/lib/python2.7/dist-packages/scipy/optimize/minpack.py", line 115, in fsolve
    _check_func('fsolve', 'func', func, x0, args, n, (n,))
  File "/usr/lib/python2.7/dist-packages/scipy/optimize/minpack.py", line 26, in _check_func
    raise TypeError(msg)
TypeError: fsolve: there is a mismatch between the input and output shape of the 'func' argument '<lambda>'.

【问题讨论】：

标签： python numpy scipy

【解决方案1】：

看来我试图错误地使用此例程。这个例程需要相同数量的方程和变量，而我给它的一个方程有三个变量。因此，如果要最小化的函数的输入是 3-D 数组，则输出应该是 3-D 数组。此代码有效：

import numpy as np
from scipy.optimize import fsolve

tmpFunc = lambda xIn: np.array( [(xIn[0]-4)**2 + xIn[1], (xIn[1]-5)**2 - xIn[2]) \
, (xIn[2]-7)**3 + xIn[0] ] )

x0 = [3,4,5]
xFinal = fsolve(tmpFunc, x0 )

print xFinal

代表同时解三个方程。

【讨论】：