numpy curve_fit，带有向量系数

Question

有没有办法使用numpys curve_fit（或者可能是另一个拟合例程）来拟合系数列表？

我知道可以填写表格

的功能

def func( x , y , a , b , c )

其中a，b，c表示系数。

但我想安装一个定义如下的函数：

def func( x , y , A )

其中A =（a，b，c）。

我知道MATLAB的'nlinfit'以这种方式处理它的系数但是有没有办法在python中类似地执行它？

编辑：

主要问题是系数的数量不同。

我找到了numpy.curve_fit如何接受列表中的系数的方法

将功能定义为

def func( x , y , *A )

curve_fit使A的条目适合函数。

请注意，我必须通过向curve_fit

提供系数猜测来定义系数的数量

Answer 1

如果我理解你的问题是正确的，你只需要一个包装函数：

def wrapper_func(x, y, a, b, c):
    return func(x, y, (a, b, c))

因此，您只需将包装函数传递给curve_fit。