从 Matlab 到 Python：矩形内具有高斯分布的 N 点

Question

我在 Matlab 中编写了这段代码，我正在寻找 Python 中的等效代码。有人可以帮忙吗？ 代码是在一个边长LxL的矩形内生成N个点，具有高斯分布。 代码来源于此链接：gaussian_inside_rectangle

function ans = randn_rect( N, sigma, L )
ans = zeros(0,2);
while size(ans,1) < N,
   pts = sigma * randn( ceil(1.25*(N-size(ans,1))), 2 );    
   pts = pts(all(abs(pts)<L/2,2),:);
   ans = [ ans ; pts ];
end
ans = ans(1:N,:);

Answer 1

所以你想从truncated normal distribution 中采样N 二维点？很高兴，scipy comes with that one built-in。

这会生成一个带有随机点的 N*2 numpy 数组 points：

from scipy.stats import truncnorm
import numpy as np

N = 1000
L = 5.0
mu = 4.0
sigma = 0.8

lower, upper = 0, L

X = truncnorm(
    (lower - mu) / sigma, (upper - mu) / sigma, loc=mu, scale=sigma)

points = np.reshape(X.rvs(2*N), (N, 2))

代码改编自from this answer。