Question

我正在尝试使用scipy curve_fit将高斯函数拟合到我的数据中，网上有很多有用的例子，我试图让几个工作，但无济于事。我用一个简单的脚本编写了一个简单的脚本来诊断问题。简而言之，curve_fit没有做任何拟合，函数只是返回初始参数值，无论它们与实数有多接近。以下是简单脚本的代码：

# -*- coding: utf-8 -*-
import numpy
import matplotlib.pyplot as plt
import matplotlib.mlab as mlab
from scipy import optimize

##Fit
def Fit( datax, datay ):    
    # define your function:
    def f(x, *p): 
        p = m, b     
        return m*numpy.asarray(x) + b
    m = 0.4
    b = 2.4
    p_init = [m, b]
    Initial_model = f(datax, [m, b])    
    plt.plot(datax, Initial_model, label='Initial Model')
    plt.title("Initial Model")
#    plt.title('Initial Model')
#    plt.show()
   # fit! (given that data is an array with the data to fit)
    print optimize.curve_fit(f, datax, datay, p_init)
    coeff, var_matrix = optimize.curve_fit(f, datax, datay, p_init)
    global fit
    fit = f(datax, *coeff)
    plt.plot(datax, fit, 'r-')
    plt.show()
    print 'Fitted slope 1 = ', coeff[0]
    print 'Fitted intercept 1 = ', coeff[1]
    return fit

##Plot
def Plot( datax, datay, fit ):
    fig = plt.figure()
    ax = fig.add_subplot(111)
    plt.plot(datax, datay, 'b*')
    l = ax.plot( datax, fit, 'r-', linewidth=2)
    ax.set_xlabel('Rate')
    ax.set_ylabel('Return')
    ax.set_title("Test")
    ax.autoscale(enable=True, axis='both', tight=None)
    ax.grid(True)
    plt.show()

##data
datax = numpy.array([7.02, 20.06, 13.78, 16.92, 10.17], dtype=numpy.float64)
datay = numpy.array([5.14, 10.66, 8.44, 9.64, 6.79], dtype=numpy.float64)

##analyze
Fit( datax, datay )
Plot( datax, datay, fit )

Out:
(array([ 0.4,  2.4]), inf)
Fitted slope 1 =  0.4
Fitted intercept 1 =  2.4

我尝试过的事情： -using leastsq direct：同样的问题 - 重新安装scipy：没有变化

我在Windows 7上使用Anaconda。

可能是什么问题？

Answer 1

您对f()的定义很奇怪，并不符合您的要求。您正在将m和b分配给p，从而覆盖传入的内容。这就是为什么参数永远不会改变，因为它们不会改变！

没有必要以一种特殊的方式定义f()，只需按照通常的方式定义参数即可。对于简单的拟合线的情况，我们可以使用

def f(x, m, b) :
    return m*x + b

唯一需要的其他改变是

Initial_model = f(datax, *p_init)

您的代码将会运行。

scipy curve_fit返回初始参数估计值

1 个答案: