Подгонять эмпирическое распределение к теоретическому с помощью Scipy (Python)?


139

ВВЕДЕНИЕ : У меня есть список из более чем 30000 целочисленных значений в диапазоне от 0 до 47 включительно, например, [0,0,0,0,..,1,1,1,1,...,2,2,2,2,...,47,47,47,...]взятых из некоторого непрерывного распределения. Значения в списке не обязательно в порядке, но порядок не имеет значения для этой проблемы.

ПРОБЛЕМА : Исходя из моего распределения, я хотел бы рассчитать значение p (вероятность увидеть большие значения) для любого данного значения. Например, как вы можете видеть, значение p для 0 будет приближаться к 1, а значение p для больших чисел будет стремиться к 0.

Я не знаю, прав ли я, но для определения вероятностей я думаю, что мне нужно согласовать мои данные с теоретическим распределением, наиболее подходящим для описания моих данных. Я предполагаю, что для определения наилучшей модели необходим какой-то тест на пригодность.

Есть ли способ реализовать такой анализ в Python ( Scipyили Numpy)? Не могли бы вы привести примеры?

Спасибо!


2
У вас есть только дискретные эмпирические значения, но вы хотите непрерывное распределение? Я правильно понимаю?
Майкл Дж. Барбер

1
Это кажется бессмысленным. Что представляют цифры? Измерения с ограниченной точностью?
Майкл Дж. Барбер

1
Майкл, я объяснил, что представляют цифры в моем предыдущем вопросе: stackoverflow.com/questions/6615489/…
s_sherly

6
Это данные подсчета. Это не непрерывное распределение.
Майкл Дж. Барбер

1
Проверьте принятый ответ на этот вопрос stackoverflow.com/questions/48455018/…
Ахмад Сулиман

Ответы:


209

Распределительная арматура с ошибкой суммы квадратов (SSE)

Это обновление и модификация ответа Саулло , который использует полный список текущих scipy.statsраспределений и возвращает распределение с наименьшим SSE между гистограммой распределения и гистограммой данных.

Пример примерки

Используя набор данных Эль-Ниньо изstatsmodels , распределения соответствуют и определяется ошибка. Распределение с наименьшей ошибкой возвращается.

Все Распределения

Все приспособленные распределения

Best Fit Distribution

Best Fit Distribution

Пример кода

%matplotlib inline

import warnings
import numpy as np
import pandas as pd
import scipy.stats as st
import statsmodels as sm
import matplotlib
import matplotlib.pyplot as plt

matplotlib.rcParams['figure.figsize'] = (16.0, 12.0)
matplotlib.style.use('ggplot')

# Create models from data
def best_fit_distribution(data, bins=200, ax=None):
    """Model data by finding best fit distribution to data"""
    # Get histogram of original data
    y, x = np.histogram(data, bins=bins, density=True)
    x = (x + np.roll(x, -1))[:-1] / 2.0

    # Distributions to check
    DISTRIBUTIONS = [        
        st.alpha,st.anglit,st.arcsine,st.beta,st.betaprime,st.bradford,st.burr,st.cauchy,st.chi,st.chi2,st.cosine,
        st.dgamma,st.dweibull,st.erlang,st.expon,st.exponnorm,st.exponweib,st.exponpow,st.f,st.fatiguelife,st.fisk,
        st.foldcauchy,st.foldnorm,st.frechet_r,st.frechet_l,st.genlogistic,st.genpareto,st.gennorm,st.genexpon,
        st.genextreme,st.gausshyper,st.gamma,st.gengamma,st.genhalflogistic,st.gilbrat,st.gompertz,st.gumbel_r,
        st.gumbel_l,st.halfcauchy,st.halflogistic,st.halfnorm,st.halfgennorm,st.hypsecant,st.invgamma,st.invgauss,
        st.invweibull,st.johnsonsb,st.johnsonsu,st.ksone,st.kstwobign,st.laplace,st.levy,st.levy_l,st.levy_stable,
        st.logistic,st.loggamma,st.loglaplace,st.lognorm,st.lomax,st.maxwell,st.mielke,st.nakagami,st.ncx2,st.ncf,
        st.nct,st.norm,st.pareto,st.pearson3,st.powerlaw,st.powerlognorm,st.powernorm,st.rdist,st.reciprocal,
        st.rayleigh,st.rice,st.recipinvgauss,st.semicircular,st.t,st.triang,st.truncexpon,st.truncnorm,st.tukeylambda,
        st.uniform,st.vonmises,st.vonmises_line,st.wald,st.weibull_min,st.weibull_max,st.wrapcauchy
    ]

    # Best holders
    best_distribution = st.norm
    best_params = (0.0, 1.0)
    best_sse = np.inf

    # Estimate distribution parameters from data
    for distribution in DISTRIBUTIONS:

        # Try to fit the distribution
        try:
            # Ignore warnings from data that can't be fit
            with warnings.catch_warnings():
                warnings.filterwarnings('ignore')

                # fit dist to data
                params = distribution.fit(data)

                # Separate parts of parameters
                arg = params[:-2]
                loc = params[-2]
                scale = params[-1]

                # Calculate fitted PDF and error with fit in distribution
                pdf = distribution.pdf(x, loc=loc, scale=scale, *arg)
                sse = np.sum(np.power(y - pdf, 2.0))

                # if axis pass in add to plot
                try:
                    if ax:
                        pd.Series(pdf, x).plot(ax=ax)
                    end
                except Exception:
                    pass

                # identify if this distribution is better
                if best_sse > sse > 0:
                    best_distribution = distribution
                    best_params = params
                    best_sse = sse

        except Exception:
            pass

    return (best_distribution.name, best_params)

def make_pdf(dist, params, size=10000):
    """Generate distributions's Probability Distribution Function """

    # Separate parts of parameters
    arg = params[:-2]
    loc = params[-2]
    scale = params[-1]

    # Get sane start and end points of distribution
    start = dist.ppf(0.01, *arg, loc=loc, scale=scale) if arg else dist.ppf(0.01, loc=loc, scale=scale)
    end = dist.ppf(0.99, *arg, loc=loc, scale=scale) if arg else dist.ppf(0.99, loc=loc, scale=scale)

    # Build PDF and turn into pandas Series
    x = np.linspace(start, end, size)
    y = dist.pdf(x, loc=loc, scale=scale, *arg)
    pdf = pd.Series(y, x)

    return pdf

# Load data from statsmodels datasets
data = pd.Series(sm.datasets.elnino.load_pandas().data.set_index('YEAR').values.ravel())

# Plot for comparison
plt.figure(figsize=(12,8))
ax = data.plot(kind='hist', bins=50, normed=True, alpha=0.5, color=plt.rcParams['axes.color_cycle'][1])
# Save plot limits
dataYLim = ax.get_ylim()

# Find best fit distribution
best_fit_name, best_fit_params = best_fit_distribution(data, 200, ax)
best_dist = getattr(st, best_fit_name)

# Update plots
ax.set_ylim(dataYLim)
ax.set_title(u'El Niño sea temp.\n All Fitted Distributions')
ax.set_xlabel(u'Temp (°C)')
ax.set_ylabel('Frequency')

# Make PDF with best params 
pdf = make_pdf(best_dist, best_fit_params)

# Display
plt.figure(figsize=(12,8))
ax = pdf.plot(lw=2, label='PDF', legend=True)
data.plot(kind='hist', bins=50, normed=True, alpha=0.5, label='Data', legend=True, ax=ax)

param_names = (best_dist.shapes + ', loc, scale').split(', ') if best_dist.shapes else ['loc', 'scale']
param_str = ', '.join(['{}={:0.2f}'.format(k,v) for k,v in zip(param_names, best_fit_params)])
dist_str = '{}({})'.format(best_fit_name, param_str)

ax.set_title(u'El Niño sea temp. with best fit distribution \n' + dist_str)
ax.set_xlabel(u'Temp. (°C)')
ax.set_ylabel('Frequency')

2
Потрясающие. Рассмотрите возможность использования density=Trueвместо normed=Trueв np.histogram(). ^^
Пек

1
@tmthydvnprt Возможно, вы могли бы отменить изменения в .plot()методах, чтобы избежать путаницы в будущем. ^^
Пек

10
Для того, чтобы получить имена распределения: from scipy.stats._continuous_distns import _distn_names. Затем вы можете использовать что-то подобное getattr(scipy.stats, distname)для каждого distnameв _distn_names`. Полезно, потому что дистрибутивы обновляются с различными версиями SciPy.
Брэд Соломон

1
Не могли бы вы объяснить, почему этот код проверяет только наилучшее соответствие непрерывных распределений и не может проверять дискретные или многомерные распределения. Спасибо.
Адам Шредер

6
Очень круто. Мне пришлось обновить параметр цвета -ax = data.plot(kind='hist', bins=50, normed=True, alpha=0.5, color=list(matplotlib.rcParams['axes.prop_cycle'])[1]['color'])
basswaves

147

В SciPy 0.12.0 реализовано 82 функции распределения . Вы можете проверить, как некоторые из них соответствуют вашим данным, используя их fit()метод . Проверьте код ниже для более подробной информации:

введите описание изображения здесь

import matplotlib.pyplot as plt
import scipy
import scipy.stats
size = 30000
x = scipy.arange(size)
y = scipy.int_(scipy.round_(scipy.stats.vonmises.rvs(5,size=size)*47))
h = plt.hist(y, bins=range(48))

dist_names = ['gamma', 'beta', 'rayleigh', 'norm', 'pareto']

for dist_name in dist_names:
    dist = getattr(scipy.stats, dist_name)
    param = dist.fit(y)
    pdf_fitted = dist.pdf(x, *param[:-2], loc=param[-2], scale=param[-1]) * size
    plt.plot(pdf_fitted, label=dist_name)
    plt.xlim(0,47)
plt.legend(loc='upper right')
plt.show()

Ссылки:

- Распределение фитингов, качество посадки, значение p. Возможно ли это сделать с помощью Scipy (Python)?

- Распределительная арматура со Scipy

А вот список с именами всех функций распределения, доступных в Scipy 0.12.0 (VI):

dist_names = [ 'alpha', 'anglit', 'arcsine', 'beta', 'betaprime', 'bradford', 'burr', 'cauchy', 'chi', 'chi2', 'cosine', 'dgamma', 'dweibull', 'erlang', 'expon', 'exponweib', 'exponpow', 'f', 'fatiguelife', 'fisk', 'foldcauchy', 'foldnorm', 'frechet_r', 'frechet_l', 'genlogistic', 'genpareto', 'genexpon', 'genextreme', 'gausshyper', 'gamma', 'gengamma', 'genhalflogistic', 'gilbrat', 'gompertz', 'gumbel_r', 'gumbel_l', 'halfcauchy', 'halflogistic', 'halfnorm', 'hypsecant', 'invgamma', 'invgauss', 'invweibull', 'johnsonsb', 'johnsonsu', 'ksone', 'kstwobign', 'laplace', 'logistic', 'loggamma', 'loglaplace', 'lognorm', 'lomax', 'maxwell', 'mielke', 'nakagami', 'ncx2', 'ncf', 'nct', 'norm', 'pareto', 'pearson3', 'powerlaw', 'powerlognorm', 'powernorm', 'rdist', 'reciprocal', 'rayleigh', 'rice', 'recipinvgauss', 'semicircular', 't', 'triang', 'truncexpon', 'truncnorm', 'tukeylambda', 'uniform', 'vonmises', 'wald', 'weibull_min', 'weibull_max', 'wrapcauchy'] 

7
Что если normed = Trueпри построении гистограммы? Ты бы не умножил pdf_fittedна size, верно?
алоха

3
Посмотрите этот ответ, если вы хотите узнать, как выглядят все дистрибутивы, или узнать, как получить доступ ко всем из них.
tmthydvnprt

@SaulloCastro Что представляют собой 3 значения в param в выводе dist.fit
shaifali Gupta

2
Для того, чтобы получить имена распределения: from scipy.stats._continuous_distns import _distn_names. Затем вы можете использовать что-то подобное getattr(scipy.stats, distname)для каждого distnameв _distn_names`. Полезно, потому что дистрибутивы обновляются с различными версиями SciPy.
Брэд Соломон

1
Я бы удалил color = 'w' из кода, иначе гистограмма не отображается.
Eran

12

fit()Метод, упомянутый @Saullo Castro, обеспечивает оценку максимального правдоподобия (MLE). Лучшее распределение ваших данных - это то, которое дает вам самое высокое, может быть определено несколькими различными способами: например,

1, тот, который дает вам наибольшую вероятность регистрации.

2, тот, который дает вам наименьшие значения AIC, BIC или BICc (см. Вики: http://en.wikipedia.org/wiki/Akaike_information_criterion) , в основном можно рассматривать как логарифмическую вероятность, скорректированную на количество параметров, как распределение с большим количеством параметров. Ожидается, что параметры будут соответствовать лучше)

3, тот, который максимизирует байесовскую апостериорную вероятность. (см. вики: http://en.wikipedia.org/wiki/Posterior_probability )

Конечно, если у вас уже есть дистрибутив, который должен описывать ваши данные (основанные на теориях в вашей конкретной области) и вы хотите придерживаться этого, вы пропустите этап определения наиболее подходящего распределения.

scipyне поставляется с функцией для вычисления правдоподобия журнала (хотя предусмотрен метод MLE), но жесткий код один прост: см. Медленнее ли встроенные функции плотности вероятности `scipy.stat.distributions`, чем предоставленные пользователем?


1
Как бы я применил этот метод к ситуации, когда данные уже были объединены - это уже гистограмма, а не генерирование гистограммы из данных?
Пит

@pete, это была бы ситуация данных с интервальной цензурой, для этого есть метод максимального правдоподобия, но в настоящее время он не реализован вscipy
CT Zhu

Не забудьте доказательства
jtlz2

5

AFAICU, ваше распределение дискретно (и ничего, кроме дискретного). Поэтому достаточно просто подсчитать частоты различных значений и нормализовать их. Итак, пример, чтобы продемонстрировать это:

In []: values= [0, 0, 0, 0, 0, 1, 1, 1, 1, 2, 2, 2, 3, 3, 4]
In []: counts= asarray(bincount(values), dtype= float)
In []: cdf= counts.cumsum()/ counts.sum()

Таким образом, вероятность увидеть значения выше, чем 1просто (согласно дополнительной кумулятивной функции распределения (ccdf) :

In []: 1- cdf[1]
Out[]: 0.40000000000000002

Обратите внимание, что ccdf тесно связан с функцией выживания (sf) , но он также определяется с помощью дискретных распределений, тогда как sf определяется только для смежных распределений.


2

Для меня это похоже на проблему оценки плотности вероятности.

from scipy.stats import gaussian_kde
occurences = [0,0,0,0,..,1,1,1,1,...,2,2,2,2,...,47]
values = range(0,48)
kde = gaussian_kde(map(float, occurences))
p = kde(values)
p = p/sum(p)
print "P(x>=1) = %f" % sum(p[1:])

Также см. Http://jpktd.blogspot.com/2009/03/using-gaussian-kernel-density.html .


1
Для будущих читателей: это решение (или, по крайней мере, идея) дает простейший ответ на вопросы OP («что такое p-значение») - было бы интересно узнать, как это можно сравнить с некоторыми из более сложных методов, которые подходят известное распределение.
Грег

Работают ли гауссовы регрессии ядра для всех распределений?

@mikey Как правило, регрессии не работают для всех дистрибутивов. Они не плохи, хотя
Эколог

2

Попробуйте distfitбиблиотеку.

pip install distfit

# Create 1000 random integers, value between [0-50]
X = np.random.randint(0, 50,1000)

# Retrieve P-value for y
y = [0,10,45,55,100]

# From the distfit library import the class distfit
from distfit import distfit

# Initialize.
# Set any properties here, such as alpha.
# The smoothing can be of use when working with integers. Otherwise your histogram
# may be jumping up-and-down, and getting the correct fit may be harder.
dist = distfit(alpha=0.05, smooth=10)

# Search for best theoretical fit on your empirical data
dist.fit_transform(X)

> [distfit] >fit..
> [distfit] >transform..
> [distfit] >[norm      ] [RSS: 0.0037894] [loc=23.535 scale=14.450] 
> [distfit] >[expon     ] [RSS: 0.0055534] [loc=0.000 scale=23.535] 
> [distfit] >[pareto    ] [RSS: 0.0056828] [loc=-384473077.778 scale=384473077.778] 
> [distfit] >[dweibull  ] [RSS: 0.0038202] [loc=24.535 scale=13.936] 
> [distfit] >[t         ] [RSS: 0.0037896] [loc=23.535 scale=14.450] 
> [distfit] >[genextreme] [RSS: 0.0036185] [loc=18.890 scale=14.506] 
> [distfit] >[gamma     ] [RSS: 0.0037600] [loc=-175.505 scale=1.044] 
> [distfit] >[lognorm   ] [RSS: 0.0642364] [loc=-0.000 scale=1.802] 
> [distfit] >[beta      ] [RSS: 0.0021885] [loc=-3.981 scale=52.981] 
> [distfit] >[uniform   ] [RSS: 0.0012349] [loc=0.000 scale=49.000] 

# Best fitted model
best_distr = dist.model
print(best_distr)

# Uniform shows best fit, with 95% CII (confidence intervals), and all other parameters
> {'distr': <scipy.stats._continuous_distns.uniform_gen at 0x16de3a53160>,
>  'params': (0.0, 49.0),
>  'name': 'uniform',
>  'RSS': 0.0012349021241149533,
>  'loc': 0.0,
>  'scale': 49.0,
>  'arg': (),
>  'CII_min_alpha': 2.45,
>  'CII_max_alpha': 46.55}

# Ranking distributions
dist.summary

# Plot the summary of fitted distributions
dist.plot_summary()

введите описание изображения здесь

# Make prediction on new datapoints based on the fit
dist.predict(y)

# Retrieve your pvalues with 
dist.y_pred
# array(['down', 'none', 'none', 'up', 'up'], dtype='<U4')
dist.y_proba
array([0.02040816, 0.02040816, 0.02040816, 0.        , 0.        ])

# Or in one dataframe
dist.df

# The plot function will now also include the predictions of y
dist.plot()

Наиболее подходящий

Обратите внимание, что в этом случае все точки будут значимыми из-за равномерного распределения. Вы можете фильтровать с помощью dist.y_pred, если требуется.


1

С OpenTURNS я бы использовал критерии BIC, чтобы выбрать лучшее распределение, подходящее для таких данных. Это потому, что этот критерий не дает слишком большого преимущества для распределений, которые имеют больше параметров. Действительно, если у распределения есть больше параметров, подобранному распределению легче быть ближе к данным. Кроме того, Колмогоров-Смирнов может не иметь смысла в этом случае, потому что небольшая ошибка в измеренных значениях окажет огромное влияние на значение p.

Чтобы проиллюстрировать этот процесс, я загружаю данные Эль-Ниньо, которые содержат 732 ежемесячных измерения температуры с 1950 по 2010 год:

import statsmodels.api as sm
dta = sm.datasets.elnino.load_pandas().data
dta['YEAR'] = dta.YEAR.astype(int).astype(str)
dta = dta.set_index('YEAR').T.unstack()
data = dta.values

Легко получить 30 встроенных одномерных фабрик распределений GetContinuousUniVariateFactoriesстатическим методом. После этого BestModelBICстатический метод возвращает лучшую модель и соответствующий показатель BIC.

sample = ot.Sample(data, 1)
tested_factories = ot.DistributionFactory.GetContinuousUniVariateFactories()
best_model, best_bic = ot.FittingTest.BestModelBIC(sample,
                                                   tested_factories)
print("Best=",best_model)

который печатает:

Best= Beta(alpha = 1.64258, beta = 2.4348, a = 18.936, b = 29.254)

Для того, чтобы графически сравнить подгонку к гистограмме, я использую drawPDFметоды наилучшего распределения.

import openturns.viewer as otv
graph = ot.HistogramFactory().build(sample).drawPDF()
bestPDF = best_model.drawPDF()
bestPDF.setColors(["blue"])
graph.add(bestPDF)
graph.setTitle("Best BIC fit")
name = best_model.getImplementation().getClassName()
graph.setLegends(["Histogram",name])
graph.setXTitle("Temperature (°C)")
otv.View(graph)

Это производит:

Бета-версия соответствует температуре Эль-Ниньо

Более подробная информация по этой теме представлена ​​в документе BestModelBIC . Можно было бы включить дистрибутив Scipy в SciPyDistribution или даже с дистрибутивами ChaosPy с ChaosPyDistribution , но я предполагаю, что текущий скрипт выполняет большинство практических целей.


2
Вы, вероятно, должны объявить интерес?
jtlz2

0

Извините, если я не понимаю вашу потребность, но как насчет хранения ваших данных в словаре, где ключами будут числа от 0 до 47 и значения количества вхождений связанных ключей в вашем исходном списке?
Таким образом, ваша вероятность p (x) будет суммой всех значений для ключей больше x, деленной на 30000.


В этом случае p (x) будет одинаковым (равно 0) для любого значения больше 47. Мне нужно непрерывное распределение вероятностей.
s_sherly

2
@s_sherly - Было бы , наверное, хорошо , если вы могли бы изменить и уточнить ваш вопрос, как и в самом деле «вероятность увидеть большие значения» - как вы выразились - IS ноль для значений , которые находятся выше самого высокого значения в бассейне ,
Мак
Используя наш сайт, вы подтверждаете, что прочитали и поняли нашу Политику в отношении файлов cookie и Политику конфиденциальности.
Licensed under cc by-sa 3.0 with attribution required.