scipy.stats.nbinom#

scipy.stats.nbinom = <scipy.stats._discrete_distns.nbinom_gen object>[source]#

负二项分布离散随机变量。

作为 rv_discrete 类的实例，nbinom 对象从其继承了一组泛型方法（有关完整列表，请参见下方），并补充了针对此特定分布的详细信息。

另请参见

hypergeom、binom、nhypergeom

注释

负二项分布描述了一系列独立同分布的伯努利试验，直至达到一个预先定义的、非随机的成功次数。

对于 nbinom，故障次数的概率质量函数为

\[f(k) = \binom{k+n-1}{n-1} p^n (1-p)^k\]

针对 \(k \ge 0\)、\(0 < p \leq 1\)

nbinom以形状参数 \(n\) 和 \(p\) 作为参数，其中 \(n\) 为成功次数，\(p\) 为一次成功的概率，并且 \(1-p\) 为一次失败的概率。

负二项分布的另一种常见参数化形式是使用失败次均值 \(\mu\) 达到 \(n\) 次成功。均值 \(\mu\) 与成功概率之间的关系为

\[p = \frac{n}{n + \mu}\]

成功次数 \(n\) 也可以使用一个“离散”、“异质”或“聚合”参数 \(\alpha\) 来表示，该参数将均值 \(\mu\) 与方差 \(\sigma^2\) 联系起来，例如 \(\sigma^2 = \mu + \alpha \mu^2\)。无论对 \(\alpha\) 使用哪种约定，

\[\begin{split}p &= \frac{\mu}{\sigma^2} \\ n &= \frac{\mu^2}{\sigma^2 - \mu}\end{split}\]

上述概率质量函数在“标准化”形式中定义。使用 loc 参数可改变分布。具体来说， nbinom.pmf(k, n, p, loc) 与 nbinom.pmf(k - loc, n, p) 在同等情况下是完全等效的。

示例

>>> import numpy as np
>>> from scipy.stats import nbinom
>>> import matplotlib.pyplot as plt
>>> fig, ax = plt.subplots(1, 1)

计算前四个矩

>>> n, p = 5, 0.5
>>> mean, var, skew, kurt = nbinom.stats(n, p, moments='mvsk')

显示概率质量函数 (pmf)

>>> x = np.arange(nbinom.ppf(0.01, n, p),
...               nbinom.ppf(0.99, n, p))
>>> ax.plot(x, nbinom.pmf(x, n, p), 'bo', ms=8, label='nbinom pmf')
>>> ax.vlines(x, 0, nbinom.pmf(x, n, p), colors='b', lw=5, alpha=0.5)

或者，可以调用分布对象（作为函数）来修复形状和位置。这将返回一个“冻结的”RV对象，其中包含给定的固定参数。

冻结分布并显示冻结后的 pmf

>>> rv = nbinom(n, p)
>>> ax.vlines(x, 0, rv.pmf(x), colors='k', linestyles='-', lw=1,
...         label='frozen pmf')
>>> ax.legend(loc='best', frameon=False)
>>> plt.show()
../../_images/scipy-stats-nbinom-1_00_00.png

检查 cdf 和 ppf 的准确性

>>> prob = nbinom.cdf(x, n, p)
>>> np.allclose(x, nbinom.ppf(prob, n, p))
True

生成随机数

>>> r = nbinom.rvs(n, p, size=1000)

方法

rvs(n, p, loc=0, size=1, random_state=None)

随机变量。

pmf(k, n, p, loc=0)

概率质量函数。

logpmf(k, n, p, loc=0)

概率质量函数的对数。

cdf(k, n, p, loc=0)

累积分布函数。

logcdf(k, n, p, loc=0)

累积分布函数对数。

sf(k, n, p, loc=0)

生存函数（也定义为 1 - cdf，但有时候 sf 更准确）。

logsf(k, n, p, loc=0)

生存函数对数。

ppf(q, n, p, loc=0)

百分位数函数（cdf 的反函数 - 百分位）。

isf(q, n, p, loc=0)

逆生存函数（sf 的反函数）。

stats(n, p, loc=0, moments=’mv’)

平均值(‘m’)、方差(‘v’)、偏度(‘s’) 和/或峰度(‘k’)。

entropy(n, p, loc=0)

RV 的（微分）熵。

expect(func, args=(n, p), loc=0, lb=None, ub=None, conditional=False)

对于分布的一个函数（一个参数）的期望值。

median(n, p, loc=0)

分布的中位数。

mean(n, p, loc=0)

分布的平均值。

var(n, p, loc=0)

分布的方差。

std(n, p, loc=0)

分布的标准差。

interval(confidence, n, p, loc=0)

中位数周围面积相等的置信区间。
本页内容