问题描述
我想从二元 Possion 分布中抽取 N 次。
是否有类似于 R 中的包 bivpois
的 Python 模块?
在 Python 中,我只知道 scipy.stats.poisson
和 numpy.random.possion
库,它们允许我根据单个参数 lambda 从单变量泊松分布进行绘制,但不知道从双变量或多变量。
解决方法
你可以很容易地自己完成,因为我没有看到任何内置方法:
https://en.wikipedia.org/wiki/Poisson_distribution#Bivariate_Poisson_distribution
步骤:
- 使用参数 lambda_i 生成 3 个独立的泊松变量 Z_i
- 生成两个 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)