Python 等效于用 mu 参数化的 R 的 rnbinom

Question

R 有一个 negative binomial function可以使用 mu 而不是概率(例如， float >= 0 和 <=1)进行参数化。我正在尝试复制此分布:

rnbinom(1000,size=0.3,mu=15)

在 Python 中。据我所知，Numpy's negative binomial function只允许一个概率。另外，我不清楚 Numpy 中的 size 参数是什么。

Answer 1

以下是您传递给 rnbinom 的参数:

In [131]: num_samples = 10000

In [132]: size = 0.3

In [133]: mu = 15

如您链接到的 R 文档中所述，您可以按如下方式计算概率:

In [134]: prob = size/(size + mu)

numpy.random.negative_binomial 的前两个参数对应于 R 函数的 size 和 prob 参数。 negative_binomial 的第三个参数是样本数。 (注意——numpy 称此参数为 size；它指的是要生成的样本的大小。所有 numpy 随机函数都有一个 size 参数。)

In [135]: sample = np.random.negative_binomial(size, prob, num_samples)

样本的均值应该接近 15。

In [136]: sample.mean()
Out[136]: 14.9032