Question

我想用数值方法求解非线性方程组，并将numpy ndarrays作为输入传递。考虑下面的任意代码：

import numpy as np
from scipy.optimize import fsolve

def eqs(A, B, C, D):
    eq1 = (A - B * np.sin(C)).tolist()
    eq2 = [5 * B + D * np.sum(A * np.cos(C))]
    return eq1 + eq2

n = 3

A = np.zeros((n))
A0 = np.random.rand(n)
B = 0.0
B0 = np.random.rand(1)[0]

C = np.random.rand(n)
D =  np.random.rand(1)[0]

sol = fsolve(func = eqs, x0 = [A0, B0], args = [C, D])

导致

缺少必需的位置参数

错误并将函数更改为：

def eqs(A, B, C, D):
    eq1 = A - B * np.sin(C)
    eq2 = C[0] * B + D * np.sum(A * np.cos(C))
    return [eq1, eq2]

也无济于事。但是，我高度怀疑该错误与传递ndarrays有关。一种方法可能是将所有ndarray来回更改为python列表。但是后来我将无法使用numpy的矢量化函数，例如np.sin() ...

如果您能帮助我知道应该怎么做，我将不胜感激。

P.S。。上述方程只是任意的，可能根本没有解决方案。

Answer 1

检查是否可以解决您的方程式

import numpy as np
from scipy.optimize import fsolve

def eqs(X, Y):
    A, B = X[:3], X[3]
    C, D = Y[:3], Y[3]
    eq1 = A - B * np.sin(C)
    eq2 = C[0] * B + D * np.sum(A * np.cos(C))
    return np.append(eq1, eq2)

n = 3

A = np.zeros((n))
A0 = np.random.rand(n)
B = 0.0
B0 = np.random.rand(1)[0]

C = np.random.rand(n)
D =  np.random.rand(1)[0]

sol = fsolve(func = eqs, x0 = np.append(A0, B0), args = np.append(C, D))
sol

输出：

array([ 0.e+000, -1.e-323,  5.e-324, -1.e-323])

Answer 2

这些scipy.optimize函数需要具有签名的函数，如

f(x, *args)

x是求解器将变化的数组（通常为1d）。 args是仅从外部传递的参数的元组。

更改您的eqs以适应此模式

In [11]: def eqs(X, C, D): 
    ...:     A, B  = X[:-1], X[-1] 
    ...:     eq1 = (A - B * np.sin(C)).tolist() 
    ...:     eq2 = [5 * B + D * np.sum(A * np.cos(C))] 
    ...:     return eq1 + eq2 

    ...: n = 3 
    ...: A0 = np.random.rand(n) 
    ...: B0 = np.random.rand(1) 
    ...:  
    ...: C = np.random.rand(n) 
    ...: D =  np.random.rand(1)

对eqs进行测试通话：

In [12]: eqs(np.concatenate((A0,B0)),C,D)                                                            
Out[12]: 
[-0.28460532658572657,
 -0.03649115738682615,
 0.7625781482352719,
 array([5.46430853])]

现在在fsolve中尝试：

In [13]: fsolve(eqs, np.concatenate((A0,B0)), args=(C,D))                                            
Out[13]: array([0., 0., 0., 0.])

传递numpy ndarray作为fsolve函数的输入

2 个答案: