如何在Python中计算二项分布概率?
关于bin_dist函数实现的问题解答
可以使用scipy.special.comb实现该二项分布概率计算需求。
二项分布下,n次试验恰好取得x次成功的概率计算公式为:
$P(X=x) = C(n,x) \times p^x \times (1-p)^{n-x}$
公式中$C(n,x)$指从n个元素中取x个的组合数,正好是scipy.special.comb(n, x)的计算结果。
你可以按照如下方式完成函数编写:
from scipy.special import comb def bin_dist(n, p, x): # 入参合法性校验,避免非法参数返回错误结果 if not (0 <= x <= n and 0 <= p <= 1): return 0.0 return comb(n, x) * (p ** x) * ((1 - p) ** (n - x))
小提示:如果试验次数n极大、或者p极接近0/1的场景下,直接按上述公式计算可能出现数值精度问题,对精度要求高的场景可以直接调用scipy.stats.binom.pmf(x, n, p)获取结果,但仅从实现需求的角度,用comb函数的方案是完全可行的。
内容的提问来源于stack exchange,提问作者user16759613
相关产品推荐
相关产品推荐

