python - "only length-1 arrays can be converted to Python scalars"在 Sage 中使用 scipy.optimize

Question

我想根据给定的数据集调整模型的参数。

我正在尝试在 Sage 中使用 scipy 的函数 curve_fit，但我不断得到

类型错误:只有长度为 1 的数组可以转换为 Python 标量

这是我的代码:

from numpy import cos,exp,pi
f = lambda x: exp(  - 1 / cos(x)  )
import numpy as np

def ang(time): return (time-12)*pi/12

def temp(x,maxtemp):
    cte=(273+maxtemp)/f(0)**(1/4)
    if 6<x and x<18:
        return float(cte*f(ang(x))**(1/4)-273)
    else:
        return -273

lT=list(np.linspace(15,40,1+24*2))
lT=[float(num) for num in lT]             #list of y data

ltimes=np.linspace(0,24,6*24+1)[1:]
ltimes=list(ltimes)                     #list of x data
u0=lT[0]

def u(time,maxtemp,k):                    #the function I want to fit to the data
    def integ(t): return k*exp(k*t)*temp(t,maxtemp)
    return exp(-k*time)*( numerical_integral(integ, 0, time)[0] + u0 )

import scipy.optimize as optimization

print optimization.curve_fit(u, ltimes, lT,[1000,0.0003])

Answer 1

scipy.optimize.curve_fit 期望模型函数被向量化:也就是说，它必须能够接收一个数组(准确地说是 ndarray)，并且返回值数组。您可以通过添加调试打印输出立即看到问题

def u(time,maxtemp,k):                    
    print time                % for debugging
    def integ(t): return k*exp(k*t)*temp(t,maxtemp)
    return exp(-k*time)*( numerical_integral(integ, 0, time)[0] + u0 )

print 的输出将是您传递给 curve_fit 的整个数组 ltimes。这不是 numerical_integral 设计用来处理的事情。你需要一个一个地给它赋值。

像这样:

def u(time,maxtemp,k):                    
    def integ(t): return k*exp(k*t)*temp(t,maxtemp)
    return [exp(-k*time_i)*( numerical_integral(integ, 0, time_i)[0] + u0) for time_i in time]

这将处理“只能转换长度为 1 的数组”错误。然后您将有另一个错误，因为您的列表 ltimes 和 lT 的长度不同，这没有意义，因为 lT 应该是输入 ltimes 的目标输出。您应该修改这些数组的定义以确定您想要的大小。