scipy 假设分布未知,根据样本数据计算置信区间

kxxlusnw  于 2022-11-10  发布在  其他
关注(0)|答案(4)|浏览(118)

假设分布为非正态分布且未知,我想计算样本数据的置信区间。基本上,它看起来像是Pareto分布。

但我不确定。
正态分布的答案:
Compute a confidence interval from sample data
Correct way to obtain confidence interval with scipy

4c8rllxm

4c8rllxm1#

如果您不知道底层的分布,那么我首先想到的是使用自举:https://en.wikipedia.org/wiki/Bootstrapping_(statistics)
在伪代码中,假设x是一个包含数据的numpy数组:

import numpy as np
N = 10000
mean_estimates = []
for _ in range(N):
    re_sample_idx = np.random.randint(0, len(x), x.shape)
    mean_estimates.append(np.mean(x[re_sample_idx]))

mean_estimates现在是分布平均值的10000个估计值的列表。取这10000个值的第2.5和第97.5百分位数,就得到了数据平均值的置信区间:

sorted_estimates = np.sort(np.array(mean_estimates))
conf_interval = [sorted_estimates[int(0.025 * N)], sorted_estimates[int(0.975 * N)]]
i34xakig

i34xakig2#

您可以使用bootstrap来估计每个数量,这些数量也来自未知分布

def bootstrap_ci(
    data, 
    statfunction=np.average, 
    alpha = 0.05, 
    n_samples = 100):

    """inspired by https://github.com/cgevans/scikits-bootstrap"""
    import warnings

    def bootstrap_ids(data, n_samples=100):
        for _ in range(n_samples):
            yield np.random.randint(data.shape[0], size=(data.shape[0],))    

    alphas = np.array([alpha/2, 1 - alpha/2])
    nvals = np.round((n_samples - 1) * alphas).astype(int)
    if np.any(nvals < 10) or np.any(nvals >= n_samples-10):
        warnings.warn("Some values used extremal samples; results are probably unstable. "
                      "Try to increase n_samples")

    data = np.array(data)
    if np.prod(data.shape) != max(data.shape):
        raise ValueError("Data must be 1D")
    data = data.ravel()

    boot_indexes = bootstrap_ids(data, n_samples)
    stat = np.asarray([statfunction(data[_ids]) for _ids in boot_indexes])
    stat.sort(axis=0)

    return stat[nvals]

模拟帕累托分布中的一些数据:

np.random.seed(33)
data = np.random.pareto(a=1, size=111)
sample_mean = np.mean(data)

plt.hist(data, bins=25)
plt.axvline(sample_mean, c='red', label='sample mean'); plt.legend()

使用自举法生成样本平均值的置信区间:

low_ci, up_ci = bootstrap_ci(data, np.mean, n_samples=1000)

画出结果

plt.hist(data, bins=25)
plt.axvline(low_ci, c='orange', label='low_ci mean')
plt.axvline(up_ci, c='magenta', label='up_ci mean')
plt.axvline(sample_mean, c='red', label='sample mean'); plt.legend()

使用bootstrapping生成分布参数的置信区间:

from scipy.stats import pareto

true_params = pareto.fit(data)
low_ci, up_ci = bootstrap_ci(data, pareto.fit, n_samples=1000)

low_ci[0]up_ci[0]是形状参数的置信区间

low_ci[0], true_params[0], up_ci[0] ---> (0.8786, 1.0983, 1.4599)
jhkqcmku

jhkqcmku3#

从对另一个答案的讨论中,我假设你需要总体平均值的置信区间,是吗?(你必须有某个数量的置信区间,而不是分布本身。)
对于所有具有有限矩的分布,均值的抽样分布渐近地趋向于正态分布,其中均值等于总体均值,方差等于总体方差除以n。因此,如果有大量数据,$\mu \pm \Phi^{-1}(p)\sigma / \sqrt{n}$应该是总体平均值的p置信区间的良好近似,即使分布不是正态分布。

lf5gs5x2

lf5gs5x24#

当前的解决方案不起作用,因为randint似乎已被弃用

np.random.seed(10)
point_estimates = []         # Make empty list to hold point estimates

for x in range(200):         # Generate 200 samples
    sample = np.random.choice(a= x, size=x.shape)
    point_estimates.append( sample.mean() )
sorted_estimates = np.sort(np.array(point_estimates))
conf_interval = [sorted_estimates[int(0.025 * N)], sorted_estimates[int(0.975 * N)]]
print(conf_interval, conf_interval[1] - conf_interval[0])
pd.DataFrame(point_estimates).plot(kind="density", legend= False)

相关问题