python - 从python中的几何分布生成随机样本

Question

我想使用 python 从几何分布生成随机样本。

我使用的是 numpy.random.geometric(p=0.3, size=100) 函数。但后来我意识到这个函数是用pmf从几何分布生成样本的:p*q^(x-1), x=1,2,3....

但我想要来自具有 pmf 的几何分布的随机样本:p*q^(x), x=0,1,2,...

请帮忙。

Answer 1

But I want random samples from geometric distribution with pmf: p*q^(x), x=0,1,2,...

您可以使用 NumPy 的 geometric 实现，然后从样本中减去一个:

In [37]: rng = np.random.default_rng()

In [38]: nsamples = 1000

In [39]: r = rng.geometric(0.3, size=nsamples) - 1

r 是您想要的分布中的样本。