Python数据拟合

Question

当我试图将抛物线拟合到这些数据时，我感觉非常糟糕。 我最初制作数据的直方图，该数据是对象的位置，然后使用抛物线拟合将直方图箱计数的负对数值绘制到位置。 我正在使用的代码是：

time,pos=postime()
plt.plot(time, pos)
poslen=len(pos)
plt.xlabel('Time')
plt.ylabel('Positions')
plt.show()


n,bins,patches = plt.hist(pos,bins=100)
n=n.tolist()
plt.show()
l=len(bins)   
s=len(n)
posx=[]
i=0
j=0
pbin=[]
sig=[]
while j < (l-1): 
    pbin.append((bins[j]+bins[j+1])/2)
    j=j+1

while i < s:

    if n[i]==0:
        pbin[i]=0
    else:
        sig.append(np.power(1/n[i],2))
        n[i]=n[i]/poslen

        n[i]=np.log(n[i])
        n[i]=n[i]*(-1)
    i=i+1


n[:]=[y for y in n if y != 0] 
pbin[:]=[y for y in pbin if y != 0]       

from scipy.optimize import curve_fit
def parabola(x, a , b):
    return a * (np.power(x,2)) + b

popt, pcov = curve_fit(parabola, pbin, n)
print popt

plt.plot(pbin,n)
plt.plot(pbin, parabola(pbin, *popt), 'r-')

Answer 1

我不确定你为什么要计算直方图......但是这里有一个不需要直方图计算的工作示例。

import numpy as np
from scipy.optimize import curve_fit
from matplotlib import pyplot


time_ = np.arange(-5, 5, 0.1)
pos = time_**2 + np.random.rand(len(time_))*5

def parabola(x, a, b):
    return a * (np.power(x, 2)) + b

popt, pcov = curve_fit(parabola, time_, pos)
yfit = parabola(time_, *popt)

pyplot.plot(time_, pos, 'o')
pyplot.plot(time_, yfit)

此外，如果您的time_向量未统一采样，并且您希望对其进行统一采样，则可以执行以下操作：fittime_ = np.linsapce(np.min(time_), np.max(time_))然后yfit = parabola(fittime_, *popt)。

Answer 2

您也可以使用矩阵求逆。

import numpy as np
import matplotlib.pyplot as plt 
x = np.linspace(-5,5,100)
Y = (np.power(x,2) + np.random.normal(0,1,x.shape)).reshape(-1,1)

X = np.c_[np.ones(x.shape), x, np.power(x,2)]
A = np.linalg.inv(X.transpose().dot(X)).dot(X.transpose().dot(Y))
Yp = X.dot(A)

fig = plt.figure()
ax = fig.add_subplot()
plt.plot(x,Y,'o',alpha=0.5)
plt.plot(x,Yp)
plt.show()

矩阵形式是

X*A=Y
A=(Xt*X)-1*Xt*Y

如果需要，您可以更好地了解here。它并不总是有用，你可能想要应用某种形式的regularization。