Python中的多变量超几何分布

Question

是否有用于多元超几何分布(包括计算CDF(的python包？

我本想从scipy那里找到一个，但没有。

我想对超几何分布使用二项式近似，然后依赖scipy支持的多项式分布。你认为这是个好办法吗？

Answer 1

multivar-hypergeom的一般用途

使用multivar-hypergeom，使用pip install multivar-hypergeom即可轻松安装。

multivar-hypergeomCDF

测试使用示例：

# CDF method tests
@given(st.lists(st.integers(min_value=1, max_value=10000), min_size=2, max_size=10))
@settings(max_examples=100, deadline=5000)
def test_cdf_summing_to_1(xs):
dist = MultivarHypergeom(xs)
assert 1 - 1e-10 < list(dist.cdf())[-1] < 1 + 1e10
for i in range(sum(xs)):
dist.sample1()
assert 1 - 1e-10 < list(dist.cdf())[-1] < 1 + 1e10

生成

使用NumPy的Generator.multivariate_hypergeometric