传递参数列表给SciPy fsolve函数

Question

我有以下功能：

def equation1(xy, d=7.62, G=8.2728, rhop=7.51, ut=399):
    ep, uc = xy     # define variables
    g = 981         # acceleration due to gravity, cm/s^2

    f1 = 2*g*d*((ep**-4.7) - 1) - 0.01*(uc/ep - ut)**2
    f2 = G - (uc/ep - ut)*rhop*(1 - ep)
    return f1, f2

其中，我通过在SciPy中使用fsolve来求解ep和{}：

^{pr2}$

我不想为G定义一个单一的值，而是希望为ep和{}求解一个范围G值，例如G = [8, 10, 12]。当我传递G的值列表时，我得到一个关于用序列设置数组元素的错误。在

是否可以使用fsolve，其中一个参数是值的列表或数组？在

Answer 1

您可以将G作为equation1的参数，并通过fsolve（使用其args参数）将其传递为：

from scipy.optimize import fsolve

def equation1(xy, G, d=7.62, rhop=7.51, ut=399):
    ep, uc = xy     # define variables
    g = 981         # acceleration due to gravity, cm/s^2

    f1 = 2*g*d*((ep**-4.7) - 1) - 0.01*(uc/ep - ut)**2
    f2 = G - (uc/ep - ut)*rhop*(1 - ep)
    return f1, f2


for G in [8, 10, 12]:
    ep1, uc1 = fsolve(equation1, [1, 500], (G, ))
    print(G, ep1, uc1)

在我的机器上显示：

^{pr2}$