如何使用 scipy.stats.truncnorm 限制 numpy.random.normal 生成的值范围?
numpy.random.normal 中如何限制值的范围
在使用 numpy.random.normal 时,有时您可能会希望生成的值落在特定的范围内。虽然您可以重复调用该函数并丢弃不符合条件的值,但存在一种更有效的方法可以实现此目的。让我们探讨使用 scipy.stats.truncnorm 的方法。
使用 scipy.stats.truncnorm 生成截断正态分布
scipy.stats.truncnorm 允许您从截断正态分布中生成随机变量。此分布由其下限、上限、均值和标准差定义。在 scipy.stats.truncnorm 的上下文中:
以下代码片段展示了如何使用 scipy.stats.truncnorm 生成随机变量:
import matplotlib.pyplot as pltimport scipy.stats as statslower, upper = 3.5, 6mu, sigma = 5, 0.7X = stats.truncnorm( (lower - mu) / sigma, (upper - mu) / sigma, loc=mu, scale=sigma)fig, ax = plt.subplots(2, sharex=True)ax[0].hist(X.rvs(10000), normed=True)ax[1].hist(N.rvs(10000), normed=True)plt.show()
此代码将生成一个截断正态分布,其下限为 3.5、上限为 6、均值为 5 和标准差为 0.7。生成的随机变量将仅落在 3.5 和 6 之间。