Python中的二元泊松分布

如何解决Python中的二元泊松分布

我想从二元 Possion 分布中抽取 N 次。 是否有类似于 R 中的包 bivpois 的 Python 模块？

在 Python 中，我只知道 scipy.stats.poisson 和 numpy.random.possion 库，它们允许我根据单个参数 lambda 从单变量泊松分布进行绘制，但不知道从双变量或多变量。

解决方法

你可以很容易地自己完成，因为我没有看到任何内置方法：

https://en.wikipedia.org/wiki/Poisson_distribution#Bivariate_Poisson_distribution

步骤：

1. 使用参数 lambda_i 生成 3 个独立的泊松变量 Z_i
2. 生成两个 P_i = Z_i + Z_3 for i = 1,2 跟随 Poi(lambda_i + lambda_3)

代码：

import numpy
lam1 = 1
lam2 = 2
lam3 = 3
#wrap next part in a loop to generate more than 1 sample
a = np.random.poisson(lam1)
b = np.random.poisson(lam2)
c = np.random.poisson(lam3)
bivariate1 = a + c #follows Poi(lam1+lam3)
bivariate2 = b + c #follows Poi(lam2+lam3)