Python最小二乘法

Question

我的代码有一点小问题：

def pval(x, p, i):
return sum(p[0]*(cos(i*x)) for i in range(1, p[1] + 1))

if __name__ == '__main__':
   main()

x, y = loadtxt("Data.txt" , unpack=True)
errfunc = lambda p, x, y: pval(p, x, i) - y # Distance to the target function
p0 = [49, 100] # Initial guess for the parameters
p1, success = optimize.leastsq(errfunc, p0[:], args=(x, y))

我出现了一个提示，说‘优化’的名字没有定义，但是我看不出这个错误来自哪里？

Answer 1

问题出在最后一行：p1, success = optimize.leastsq(...) (错误消息回溯指出了它)。

Python不知道您所说的optimize是什么意思。你必须告诉它：from scipy import optimize接近你程序的顶端。

您还可以调用一个不存在的main()函数(至少在您的代码片段中)。通常，你会像这样组织你的代码：

def pval(x, p, i):
    return sum(p[0]*(cos(i*x)) for i in range(1, p[1] + 1))

def main():
    x, y = loadtxt("Data.txt" , unpack=True)
    errfunc = lambda p, x, y: pval(p, x, i) - y # Distance to the target function
    p0 = [49, 100] # Initial guess for the parameters
    p1, success = optimize.leastsq(errfunc, p0[:], args=(x, y))
    # do whatever with p1

if __name__ == '__main__':
   main()

或者干脆从你的代码中完全去掉if __name__ == '__main__'部分。