RatioUniforms#
- class scipy.stats.sampling.RatioUniforms(pdf, *, umax, vmin, vmax, c=0, random_state=None)[源代码][源代码]#
使用比率均匀方法从概率密度函数生成随机样本。
- 参数:
- pdf可调用
一个签名函数 pdf(x),它与分布的概率密度函数成正比。
- umax浮动
在 u 方向上的边界矩形的上限。
- vmin浮动
边界矩形在v方向上的下限。
- vmax浮动
边界矩形在v方向上的上限。
- c浮点数,可选。
比率均匀方法的偏移参数,参见注释。默认值为 0。
- random_state{None, int,}
如果 seed 是 None(或 np.random),则使用
numpy.random.RandomState
单例。如果 seed 是 int,则使用新的RandomState
实例,并以 seed 为种子。如果 seed 已经是Generator
或RandomState
实例,则使用该实例。
方法
rvs
([size])随机变量的采样
注释
给定一个单变量概率密度函数 pdf 和一个常数 c,定义集合
A = {(u, v) : 0 < u <= sqrt(pdf(v/u + c))}
。如果(U, V)
是一个在A
上均匀分布的随机向量,那么V/U + c
服从 pdf 定义的分布。上述结果(见 [1], [2])可以用于仅使用PDF采样随机变量,即不需要CDF的反演。c 的典型选择是零或 pdf 的模式。集合
A
是矩形R = [0, umax] x [vmin, vmax]
的子集,其中umax = sup sqrt(pdf(x))
vmin = inf (x - c) sqrt(pdf(x))
vmax = sup (x - c) sqrt(pdf(x))
特别是,如果 pdf 是有界的且
x**2 * pdf(x)
是有界的(即次二次尾),这些值是有限的。可以在R
上均匀生成(U, V)
并返回V/U + c
,如果(U, V)
也在A
中,这可以直接验证。如果将 pdf 替换为任意常数 k > 0 的 k * pdf,算法不会改变。因此,通常使用与概率密度函数成比例的函数来工作是很方便的,通过去掉不必要的归一化因子。
直观上,如果
A
占据了大部分的包围矩形,使得(U, V)
位于A
中的概率很高,只要它位于R
中,否则所需的迭代次数会变得过大。更准确地说,注意到在R
上均匀分布地抽取(U, V)
使得(U, V)
也在A
中的预期迭代次数由比率area(R) / area(A) = 2 * umax * (vmax - vmin) / area(pdf)
给出,其中 area(pdf) 是 pdf 的积分(如果使用概率密度函数,则等于一,但如果使用与密度成比例的函数,则可以取其他值)。等式成立是因为A
的面积等于0.5 * area(pdf)
([1] 中的定理 7.1)。如果在 50000 次迭代后未能生成单个随机变量(即没有一次抽取在A
中),则会引发异常。如果边界矩形没有正确指定(即,如果它不包含
A
),算法会从一个不同于 pdf 给定的分布中采样。因此,建议执行如kstest
这样的测试作为检查。参考文献
[2]W. Hoermann and J. Leydold, “Generating generalized inverse Gaussian random variates”, Statistics and Computing, 24(4), p. 547–557, 2014.
[3]A.J. Kinderman 和 J.F. Monahan, “使用均匀随机数的比率生成随机变量”, ACM 数学软件汇刊, 3(3), 第 257–260 页, 1977.
示例
>>> import numpy as np >>> from scipy import stats
>>> from scipy.stats.sampling import RatioUniforms >>> rng = np.random.default_rng()
模拟正态分布的随机变量。在这种情况下,显式计算包围矩形很容易。为简单起见,我们省略了密度的归一化因子。
>>> f = lambda x: np.exp(-x**2 / 2) >>> v = np.sqrt(f(np.sqrt(2))) * np.sqrt(2) >>> umax = np.sqrt(f(0)) >>> gen = RatioUniforms(f, umax=umax, vmin=-v, vmax=v, random_state=rng) >>> r = gen.rvs(size=2500)
K-S 检验证实随机变量确实是正态分布的(在5%显著性水平下不拒绝正态性):
>>> stats.kstest(r, 'norm')[1] 0.250634764150542
指数分布提供了另一个例子,其中边界矩形可以被明确地确定。
>>> gen = RatioUniforms(lambda x: np.exp(-x), umax=1, vmin=0, ... vmax=2*np.exp(-1), random_state=rng) >>> r = gen.rvs(1000) >>> stats.kstest(r, 'expon')[1] 0.21121052054580314