使用Scipy vs Matlab拟合对数正态分布

时间:2013-03-26 06:23:22

标签: python matlab statistics scipy curve-fitting

我正在尝试使用Scipy拟合对数正态分布。我之前已经使用Matlab完成了它,但由于需要将应用程序扩展到统计分析之外,我正在尝试在Scipy中重现拟合值。

下面是我用来拟合数据的Matlab代码:

% Read input data (one value per line)
x = [];
fid = fopen(file_path, 'r'); % reading is default action for fopen
disp('Reading network degree data...');
if fid == -1
    disp('[ERROR] Unable to open data file.')
else
    while ~feof(fid)
        [x] = [x fscanf(fid, '%f', [1])];

    end
    c = fclose(fid);
    if c == 0
         disp('File closed successfully.');
    else
        disp('[ERROR] There was a problem with closing the file.');
    end
end

[f,xx] = ecdf(x);
y = 1-f;

parmhat  = lognfit(x); % MLE estimate
mu = parmhat(1);
sigma = parmhat(2);

这是合适的情节:

enter image description here

现在这是我的Python代码,目的是实现相同的目标:

import math
from scipy import stats
from statsmodels.distributions.empirical_distribution import ECDF 

# The same input is read as a list in Python
ecdf_func = ECDF(degrees)
x = ecdf_func.x
ccdf = 1-ecdf_func.y

# Fit data
shape, loc, scale = stats.lognorm.fit(degrees, floc=0)

# Parameters
sigma = shape # standard deviation
mu = math.log(scale) # meanlog of the distribution

fit_ccdf = stats.lognorm.sf(x, [sigma], floc=1, scale=scale) 

这是使用Python代码的合适。

enter image description here

正如您所看到的,两组代码都能够产生良好的效果,至少在视觉上是这样。

问题是估计的参数mu和sigma存在巨大差异。

来自Matlab:mu = 1.62 sigma = 1.29。 来自Python:mu = 2.78 sigma = 1.74。

为什么会有这样的差异?

注意:我已经仔细检查过,两组数据拟合完全相同。相同数量的点,相同的分布。

非常感谢您的帮助!提前谢谢。

其他信息:

import scipy
import numpy
import statsmodels

scipy.__version__
'0.9.0'

numpy.__version__
'1.6.1'

statsmodels.__version__
'0.5.0.dev-1bbd4ca'

Matlab的版本是R2011b。

版:

如下面的答案所示,错误在于Scipy 0.9。我可以使用Scipy 11.0从Matlab重现mu和sigma结果。

更新Scipy的简便方法是:

pip install --upgrade Scipy

如果你没有pip(你应该!):

sudo apt-get install pip

1 个答案:

答案 0 :(得分:6)

scipy 0.9.0中的fit方法中存在一个错误,该错误已在scipy的更高版本中修复。

以下脚本的输出应为:

Explicit formula:   mu = 4.99203450, sig = 0.81691086
Fit log(x) to norm: mu = 4.99203450, sig = 0.81691086
Fit x to lognorm:   mu = 4.99203468, sig = 0.81691081

但是scipy 0.9.0,它是

Explicit formula:   mu = 4.99203450, sig = 0.81691086
Fit log(x) to norm: mu = 4.99203450, sig = 0.81691086
Fit x to lognorm:   mu = 4.23197270, sig = 1.11581240

以下测试脚本显示了获得相同结果的三种方法:

import numpy as np
from scipy import stats


def lognfit(x, ddof=0):
    x = np.asarray(x)
    logx = np.log(x)
    mu = logx.mean()
    sig = logx.std(ddof=ddof)
    return mu, sig


# A simple data set for easy reproducibility
x = np.array([50., 50, 100, 200, 200, 300, 500])

# Explicit formula
my_mu, my_sig = lognfit(x)

# Fit a normal distribution to log(x)
norm_mu, norm_sig = stats.norm.fit(np.log(x))

# Fit the lognormal distribution
lognorm_sig, _, lognorm_expmu = stats.lognorm.fit(x, floc=0)

print "Explicit formula:   mu = %10.8f, sig = %10.8f" % (my_mu, my_sig)
print "Fit log(x) to norm: mu = %10.8f, sig = %10.8f" % (norm_mu, norm_sig)
print "Fit x to lognorm:   mu = %10.8f, sig = %10.8f" % (np.log(lognorm_expmu), lognorm_sig)

在标准版中使用选项ddof=1。开发。计算使用无偏方差估计:

In [104]: x
Out[104]: array([  50.,   50.,  100.,  200.,  200.,  300.,  500.])

In [105]: lognfit(x, ddof=1)
Out[105]: (4.9920345004312647, 0.88236457185021866)

matlab的lognfit documentation中有一条说明,当没有使用审查时,lognfit使用方差无偏估计的平方根计算sigma。这相当于在上面的代码中使用ddof = 1。