scipy.stats.truncnorm#

scipy.stats.truncnorm = <scipy.stats._continuous_distns.truncnorm_gen object>[源代码]#

截断正态连续随机变量。

作为 rv_continuous 类的实例,truncnorm 对象继承了它的一系列通用方法(完整列表见下文),并使用此特定分布的详细信息对其进行补充。

注释

此分布是以 loc (默认为 0) 为中心的正态分布,标准差为 scale (默认为 1),并在距离 locab标准差处截断。 对于任意的 locscaleab 不是移位和缩放后的分布被截断的横坐标。

注意

如果 a_truncb_trunc 是我们希望截断分布的横坐标(而不是距离 loc 的标准差数),那么我们可以按如下方式计算分布参数 ab

a, b = (a_trunc - loc) / scale, (b_trunc - loc) / scale

这是一个常见的混淆点。 如需更多说明,请参阅下面的示例。

示例

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

计算前四个矩

>>> a, b = 0.1, 2
>>> mean, var, skew, kurt = truncnorm.stats(a, b, moments='mvsk')

显示概率密度函数 (pdf)

>>> x = np.linspace(truncnorm.ppf(0.01, a, b),
...                 truncnorm.ppf(0.99, a, b), 100)
>>> ax.plot(x, truncnorm.pdf(x, a, b),
...        'r-', lw=5, alpha=0.6, label='truncnorm pdf')

或者,可以调用分布对象(作为函数)来固定形状、位置和比例参数。 这将返回一个“冻结”的 RV 对象,其中保存着给定的固定参数。

冻结分布并显示冻结的 pdf

>>> rv = truncnorm(a, b)
>>> ax.plot(x, rv.pdf(x), 'k-', lw=2, label='frozen pdf')

检查 cdfppf 的准确性

>>> vals = truncnorm.ppf([0.001, 0.5, 0.999], a, b)
>>> np.allclose([0.001, 0.5, 0.999], truncnorm.cdf(vals, a, b))
True

生成随机数

>>> r = truncnorm.rvs(a, b, size=1000)

并比较直方图

>>> ax.hist(r, density=True, bins='auto', histtype='stepfilled', alpha=0.2)
>>> ax.set_xlim([x[0], x[-1]])
>>> ax.legend(loc='best', frameon=False)
>>> plt.show()
../../_images/scipy-stats-truncnorm-1_00_00.png

在上面的示例中,loc=0scale=1,因此图在左侧的 a 和右侧的 b 处截断。 但是,假设我们要使用 loc = 1scale=0.5 生成相同的直方图。

>>> loc, scale = 1, 0.5
>>> rv = truncnorm(a, b, loc=loc, scale=scale)
>>> x = np.linspace(truncnorm.ppf(0.01, a, b),
...                 truncnorm.ppf(0.99, a, b), 100)
>>> r = rv.rvs(size=1000)
>>> fig, ax = plt.subplots(1, 1)
>>> ax.plot(x, rv.pdf(x), 'k-', lw=2, label='frozen pdf')
>>> ax.hist(r, density=True, bins='auto', histtype='stepfilled', alpha=0.2)
>>> ax.set_xlim(a, b)
>>> ax.legend(loc='best', frameon=False)
>>> plt.show()
../../_images/scipy-stats-truncnorm-1_01_00.png

请注意,分布不再看起来是在横坐标 ab 处截断。 这是因为标准正态分布首先在 ab 处截断,然后将所得分布按 scale 缩放,并按 loc 移动。 如果我们希望移动和缩放后的分布在 ab 处截断,我们需要在将这些值作为分布参数传递之前对其进行转换。

>>> a_transformed, b_transformed = (a - loc) / scale, (b - loc) / scale
>>> rv = truncnorm(a_transformed, b_transformed, loc=loc, scale=scale)
>>> x = np.linspace(truncnorm.ppf(0.01, a, b),
...                 truncnorm.ppf(0.99, a, b), 100)
>>> r = rv.rvs(size=10000)
>>> fig, ax = plt.subplots(1, 1)
>>> ax.plot(x, rv.pdf(x), 'k-', lw=2, label='frozen pdf')
>>> ax.hist(r, density=True, bins='auto', histtype='stepfilled', alpha=0.2)
>>> ax.set_xlim(a-0.1, b+0.1)
>>> ax.legend(loc='best', frameon=False)
>>> plt.show()
../../_images/scipy-stats-truncnorm-1_02_00.png

方法

rvs(a, b, loc=0, scale=1, size=1, random_state=None)

随机变量。

pdf(x, a, b, loc=0, scale=1)

概率密度函数。

logpdf(x, a, b, loc=0, scale=1)

概率密度函数的对数。

cdf(x, a, b, loc=0, scale=1)

累积分布函数。

logcdf(x, a, b, loc=0, scale=1)

累积分布函数的对数。

sf(x, a, b, loc=0, scale=1)

生存函数(也定义为 1 - cdf,但 sf 有时更准确)。

logsf(x, a, b, loc=0, scale=1)

生存函数的对数。

ppf(q, a, b, loc=0, scale=1)

百分点函数(cdf 的反函数 — 百分位数)。

isf(q, a, b, loc=0, scale=1)

逆生存函数(sf 的反函数)。

moment(order, a, b, loc=0, scale=1)

指定阶的非中心矩。

stats(a, b, loc=0, scale=1, moments='mv')

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

entropy(a, b, loc=0, scale=1)

RV 的(微分)熵。

fit(data)

通用数据的参数估计。有关关键字参数的详细文档,请参阅 scipy.stats.rv_continuous.fit

expect(func, args=(a, b), loc=0, scale=1, lb=None, ub=None, conditional=False, **kwds)

函数(一个参数)相对于分布的期望值。

median(a, b, loc=0, scale=1)

分布的中位数。

mean(a, b, loc=0, scale=1)

分布的均值。

var(a, b, loc=0, scale=1)

分布的方差。

std(a, b, loc=0, scale=1)

分布的标准差。

interval(confidence, a, b, loc=0, scale=1)

中位数周围区域相等的置信区间。