scipy.stats.sampling.

SimpleRatioUniforms#

class scipy.stats.sampling.SimpleRatioUniforms(dist, *, mode=None, pdf_area=1, domain=None, cdf_at_mode=None, random_state=None)#

简单均匀比率(SROU)方法。

SROU 基于均匀比率方法,该方法使用通用不等式来构造(通用)边界矩形。它适用于 T 凹分布,其中 T(x) = -1/sqrt(x)。该方法的主要优点是设置速度快。如果需要重复生成具有不同形状参数的分布的小到中等样本,这可能是有益的。在这种情况下,NumericalInverseHermiteNumericalInversePolynomial 的设置步骤将导致性能不佳。

参数:
dist对象

具有 pdf 方法的类的实例。

  • pdf: 分布的 PDF。PDF 的签名应为:def pdf(self, x: float) -> float。即,PDF 应接受 Python float 并返回 Python float。它不需要积分到 1,即 PDF 不需要归一化。如果未归一化,则应将 pdf_area 设置为 PDF 下的面积。

modefloat,可选

分布的(精确)众数。当众数为 None 时,将使用缓慢的数值例程来近似它。默认为 None

pdf_areafloat,可选

PDF 下的面积。可选地,可以传递 PDF 下的面积的上限,代价是增加拒绝常数。默认为 1。

domain长度为 2 的列表或元组,可选

分布的支持。默认为 None。当 None

  • 如果分布对象 dist 提供了 support 方法,则使用它来设置分布的域。

  • 否则,支持被假定为 \((-\infty, \infty)\)

cdf_at_modefloat,可选

众数处的 CDF。可以给定它以提高算法的性能。当给出众数处的 CDF 时,拒绝常数会减半。默认为 None

random_state{None, int, numpy.random.Generator,

用于生成均匀随机数流的底层 NumPy 随机数生成器或种子。如果 random_state 为 None(或 np.random),则使用 numpy.random.RandomState 单例。如果 random_state 是一个 int,则使用一个新的 RandomState 实例,并使用 random_state 进行播种。如果 random_state 已经是 GeneratorRandomState 实例,则使用该实例。

参考文献

[1]

UNU.RAN 参考手册,第 5.3.16 节,“SROU - 简单均匀比率法”,http://statmath.wu.ac.at/software/unuran/doc/unuran.html#SROU

[2]

Leydold, Josef. “用于连续和离散单变量 T 凹分布的简单通用生成器。” ACM Transactions on Mathematical Software (TOMS) 27.1 (2001): 66-82

[3]

Leydold, Josef. “通过广义均匀比率方法的短通用生成器。” Mathematics of Computation 72.243 (2003): 1453-1471

示例

>>> from scipy.stats.sampling import SimpleRatioUniforms
>>> import numpy as np

假设我们有正态分布

>>> class StdNorm:
...     def pdf(self, x):
...         return np.exp(-0.5 * x**2)

请注意,PDF 不积分到 1。我们可以在生成器初始化期间传递 PDF 下的精确面积或 PDF 下的精确面积的上限。此外,建议传递分布的众数以加快设置速度

>>> urng = np.random.default_rng()
>>> dist = StdNorm()
>>> rng = SimpleRatioUniforms(dist, mode=0,
...                           pdf_area=np.sqrt(2*np.pi),
...                           random_state=urng)

现在,我们可以使用 rvs 方法从分布中生成样本

>>> rvs = rng.rvs(10)

如果可以使用众数处的 CDF,则可以设置它以提高 rvs 的性能

>>> from scipy.stats import norm
>>> rng = SimpleRatioUniforms(dist, mode=0,
...                           pdf_area=np.sqrt(2*np.pi),
...                           cdf_at_mode=norm.cdf(0),
...                           random_state=urng)
>>> rvs = rng.rvs(1000)

我们可以通过可视化其直方图来检查样本是否来自给定的分布

>>> import matplotlib.pyplot as plt
>>> x = np.linspace(rvs.min()-0.1, rvs.max()+0.1, 1000)
>>> fx = 1/np.sqrt(2*np.pi) * dist.pdf(x)
>>> fig, ax = plt.subplots()
>>> ax.plot(x, fx, 'r-', lw=2, label='true distribution')
>>> ax.hist(rvs, bins=10, density=True, alpha=0.8, label='random variates')
>>> ax.set_xlabel('x')
>>> ax.set_ylabel('PDF(x)')
>>> ax.set_title('Simple Ratio-of-Uniforms Samples')
>>> ax.legend()
>>> plt.show()
../../_images/scipy-stats-sampling-SimpleRatioUniforms-1.png

方法

rvs([size, random_state])

从分布中采样。

set_random_state([random_state])

设置底层均匀随机数生成器。