Scipy fmin论证传递

时间:2015-01-09 09:49:35

标签: python scipy mathematical-optimization

所以我对scipy.optimize.fmin如何接受参数感到有点困惑

说我有两个功能:

def func_1(x, y):
   return (x*x) + (y*y)

def func_2(x, y, a):
   return ((x-a)*(x-a)) + ((y-a) *(y-a))

在func_1中我只想要x和y的min值,并且在func_2中我理想地想要为a传递一个值,然后找到x和y的最小值。我已经阅读了有关此问题的其他一些问题,我仍然有点困惑。

尝试致电:fmin(fund_1, [1,1])会导致:fmin() takes at least 2 arguments (1 given)

非常感谢!

1 个答案:

答案 0 :(得分:2)

首先,将func_1func_2的签名更改为:

def func_1(t):
   x,y=t
   return (x*x) + (y*y)

def func_2(t, a):
   x,y=t
   return ((x-a)*(x-a)) + ((y-a) *(y-a))

在第一种情况下,请使用

from scipy.optimize import fmin
from numpy import array
fmin(func_1,array([1,1]))

在第二种情况下,您必须传递fmin参数args=(a,)。它将完全按照你的描述进行,

fmin(func_2,array([1,1]),args=(3,))