scipy.stats.geom#
- scipy.stats.geom = <scipy.stats._discrete_distns.geom_gen object>[源代码]#
几何离散随机变量。
作为
rv_discrete
类的实例,geom
对象继承了该类的一系列通用方法(完整列表见下文),并用该特定分布的详细信息对其进行了完善。另请参阅
注释
geom
的概率质量函数为\[f(k) = (1-p)^{k-1} p\]对于 \(k \ge 1\),\(0 < p \leq 1\)
geom
将 \(p\) 作为形状参数,其中 \(p\) 是单次成功的概率,\(1-p\) 是单次失败的概率。请注意,当绘制随机样本时,随着 $p$ 减小到 $10^{-17}$ 以下,超过
np.iinfo(np.int64).max
的观测值的概率会迅速增加。对于 $p < 10^{-20}$,几乎所有观测值都会超过最大int64
值;但是,输出 dtype 始终为int64
,因此这些值会被裁剪为最大值。上面的概率质量函数是以“标准化”形式定义的。要移动分布,请使用
loc
参数。具体来说,geom.pmf(k, p, loc)
与geom.pmf(k - loc, p)
完全等效。示例
>>> import numpy as np >>> from scipy.stats import geom >>> import matplotlib.pyplot as plt >>> fig, ax = plt.subplots(1, 1)
计算前四个矩
>>> p = 0.5 >>> mean, var, skew, kurt = geom.stats(p, moments='mvsk')
显示概率质量函数 (
pmf
)>>> x = np.arange(geom.ppf(0.01, p), ... geom.ppf(0.99, p)) >>> ax.plot(x, geom.pmf(x, p), 'bo', ms=8, label='geom pmf') >>> ax.vlines(x, 0, geom.pmf(x, p), colors='b', lw=5, alpha=0.5)
或者,可以调用分布对象(作为函数)来固定形状和位置。这将返回一个“冻结”的 RV 对象,该对象保持给定的参数固定。
冻结分布并显示冻结的
pmf
>>> rv = geom(p) >>> ax.vlines(x, 0, rv.pmf(x), colors='k', linestyles='-', lw=1, ... label='frozen pmf') >>> ax.legend(loc='best', frameon=False) >>> plt.show()
检查
cdf
和ppf
的准确性>>> prob = geom.cdf(x, p) >>> np.allclose(x, geom.ppf(prob, p)) True
生成随机数
>>> r = geom.rvs(p, size=1000)
方法
rvs(p, loc=0, size=1, random_state=None)
随机变量。
pmf(k, p, loc=0)
概率质量函数。
logpmf(k, p, loc=0)
概率质量函数的对数。
cdf(k, p, loc=0)
累积分布函数。
logcdf(k, p, loc=0)
累积分布函数的对数。
sf(k, p, loc=0)
生存函数(也定义为
1 - cdf
,但 sf 有时更准确)。logsf(k, p, loc=0)
生存函数的对数。
ppf(q, p, loc=0)
百分点函数(
cdf
的逆函数 - 百分位数)。isf(q, p, loc=0)
逆生存函数(
sf
的逆函数)。stats(p, loc=0, moments='mv')
均值('m')、方差('v')、偏度('s')和/或峰度('k')。
entropy(p, loc=0)
RV 的(微分)熵。
expect(func, args=(p,), loc=0, lb=None, ub=None, conditional=False)
关于分布的函数(一个参数)的期望值。
median(p, loc=0)
分布的中位数。
mean(p, loc=0)
分布的均值。
var(p, loc=0)
分布的方差。
std(p, loc=0)
分布的标准差。
interval(confidence, p, loc=0)
中位数周围等面积的置信区间。