用mu参数化R的rnbinom的Python等价物

Question

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

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

在Python中。从我看来，Numpy's negative binomial function只允许一个概率。{mpy>在Numpy中也不清楚。在

Answer 1

以下是传递给rnbinom的参数：

In [131]: num_samples = 10000

In [132]: size = 0.3

In [133]: mu = 15

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

^{pr2}$

numpy.random.negative_binomial的前两个参数对应于R函数的size和{}参数。negative_binomial的第三个参数是样本数。（注意numpy调用这个参数size；它引用要生成的样本的大小。所有numpy random函数都有一个size参数。）

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

样本的平均值应接近15。在

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