蟒蛇当量R的rnbinom与mu参数

Question

R有一个negative binomial function，可以使用mu而不是概率（例如，一个float> = 0和< = 1）参数化。我试图复制这种分布：

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

in Python。从我所看到的，Numpy's negative binomial function只允许一个概率。此外，我不清楚0123um参数将在Numpy中。

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