python - 用于从具有不同间隔的截断正态分布中采样的矢量化代码

Question

以下代码根据具有不同间隔的截断正态分布生成大小为 100 的样本。有什么有效的(矢量化的)方法可以做到这一点吗？

from scipy.stats import truncnorm
import numpy as np
sample=[]
a_s=np.random.uniform(0,1,size=100)
b_s=a_s+0.2
for i in range(100):
    sample.append(truncnorm.rvs(a_s[i], b_s[i], size=100))
print sample

Answer 1

在不久的将来有一天，所有 NumPy/SciPy 函数都将广播它们的所有参数，您将能够执行 truncnorm.rvs(a_s, b_s, size=100)，但由于我们还没有做到这一点，您可以从均匀分布和正态分布的 CDF 和 PPF 中手动生成随机样本:

import numpy as np
from scipy.stats import truncnorm, norm

a_s = np.random.uniform(0, 1, size=100)
b_s = a_s + 0.2

cdf_start = norm.cdf(a_s)
cdf_stop = norm.cdf(b_s)
cdf_samples = np.random.uniform(0, 1, size=(100, 100))
cdf_samples *= (cdf_stop - cdf_start)[:, None]
cdf_samples +=  cdf_start[:, None]
truncnorm_samples = norm.ppf(cdf_samples)