繁体   English   中英

Python中scipy的curve_fit曲线拟合

[英]curve fitting by curve_fit from scipy in Python

我试图用指数函数拟合我的数据

import numpy as np

def exponentional(k, alpha, k0, c):
    return k0 * np.exp(k *-alpha) + c 

我使用了curve_fit的 curve_fit,

from scipy.optimize import curve_fit
import matplotlib.pyplot as plt
uniq_deg = [2,...,103,..,203,...,307,...,506] 
normalized_deg_dist = [0.99,...,0.43,..0.12,..,0.04,..., 0.01]
           
            
popt, pcov = curve_fit(exponentional, uniq_deg, normalized_deg_dist, 
                       p0 = [1,0.00001,1,1], maxfev = 6000)
           
fig = plt.figure() 
ax = fig.add_subplot(111)
             
ax.semilogy(uniq_deg, normalized_deg_dist, 'bo', label = 'Real data') 
ax.semilogy(uniq_deg,[exponentional(d,*popt) for d in uniq_deg], 'r-', label = 'Fit') 
ax.set_xlabel('Degree' ) 
ax.set_ylabel('1-CDF degree') 
ax.legend(loc='best')
ax.set_title(f'Degree distribution in {city}')
plt.show()

导致:

在此处输入图片说明

它看起来不太合身。

我错在哪里?

最后,我没有使用curve_fit 我使用了https://mathworld.wolfram.com/LeastSquaresFittingExponential.html 中指数拟合的定义

而且我还需要通过幂律拟合一些其他数据,我也这样做了。

    #%%
def fit_powerlaw(xs, ys):
    S_lnx_lny = 0.0
    S_lnx_S_lny = 0.0
    S_lny = 0.0
    S_lnx = 0.0
    S_lnx2 = 0.0
    S_ln_x_2 = 0.0
    n = len(xs)
    for (x,y) in zip(xs, ys):
        S_lnx += np.log(x)
        S_lny += np.log(y)
        S_lnx_lny += np.log(x) * np.log(y)
        S_lnx_S_lny = S_lnx * S_lny
        S_lnx2 += np.power(np.log(x),2)
        S_ln_x_2 = np.power(S_lnx,2)
    #end
    b = (n * S_lnx_lny - S_lnx_S_lny ) / (n * S_lnx2 - S_ln_x_2)
    a = (S_lny - b * S_lnx)  / (n)
    return (np.exp(a), b)
#%%
def fit_exp(xs, ys):
    S_x2_y = 0.0
    S_y_lny = 0.0
    S_x_y = 0.0
    S_x_y_lny = 0.0
    S_y = 0.0
    for (x,y) in zip(xs, ys):
        S_x2_y += x * x * y
        S_y_lny += y * np.log(y)
        S_x_y += x * y
        S_x_y_lny += x * y * np.log(y)
        S_y += y
    #end
    a = (S_x2_y * S_y_lny - S_x_y * S_x_y_lny) / (S_y * S_x2_y - S_x_y * S_x_y)
    b = (S_y * S_x_y_lny - S_x_y * S_y_lny) / (S_y * S_x2_y - S_x_y * S_x_y)
    return (np.exp(a), b)

在此处输入图片说明

在此处输入图片说明

第一个指数拟合图和第二个幂律图。 我认为结果令人信服。

暂无
暂无

声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.

 
粤ICP备18138465号  © 2020-2024 STACKOOM.COM