numpy.random.Generator.poisson#

方法

random.Generator.poisson(lam=1.0, size=None)#

从泊松分布中抽取样本。

泊松分布是二项分布在 N 很大时的极限。

参数:
lam浮点数或类似浮点数的数组

固定时间间隔内事件发生的预期数量,必须 >= 0。序列必须可以在请求的大小上进行广播。

size整数或整数元组,可选

输出形状。如果给定形状是例如 (m, n, k),则会抽取 m * n * k 个样本。如果 size 为 None(默认),则当 lam 是标量时返回单个值。否则,将抽取 np.array(lam).size 个样本。

返回:
outndarray 或标量

从参数化的泊松分布中抽取的样本。

备注

泊松分布的概率质量函数 (PMF) 为

\[f(k; \lambda)=\frac{\lambda^k e^{-\lambda}}{k!}\]

对于预期间隔为 \(\lambda\) 的事件,泊松分布 \(f(k; \lambda)\) 描述了在观测间隔 \(\lambda\) 内发生 \(k\) 个事件的概率。

由于输出受限于 C int64 类型的范围,当 lam 在可表示的最大值的 10 倍标准差范围内时,会引发 ValueError。

参考

[1]

Weisstein, Eric W. “泊松分布。” 选自 MathWorld–A Wolfram Web Resource. https://mathworld.net.cn/PoissonDistribution.html

[2]

维基百科,“泊松分布”,https://en.wikipedia.org/wiki/Poisson_distribution

示例

从分布中抽取样本

>>> rng = np.random.default_rng()
>>> lam, size = 5, 10000
>>> s = rng.poisson(lam=lam, size=size)

验证均值和方差,它们应该近似于 lam

>>> s.mean(), s.var()
(4.9917 5.1088311)  # may vary

显示直方图和概率质量函数

>>> import matplotlib.pyplot as plt
>>> from scipy import stats
>>> x = np.arange(0, 21)
>>> pmf = stats.poisson.pmf(x, mu=lam)
>>> plt.hist(s, bins=x, density=True, width=0.5)
>>> plt.stem(x, pmf, 'C1-')
>>> plt.show()
../../../_images/numpy-random-Generator-poisson-1_00_00.png

为 lambda 100 和 500 各抽取 100 个值

>>> s = rng.poisson(lam=(100., 500.), size=(100, 2))