python日志n选择k

Question

scipy.misc.comb，返回n choose k，使用gammaln函数实现。是否有保留在日志空间中的功能？我看到没有 scipy.misc.combln 或任何类似的东西。自己实现是微不足道的，但如果它已经在某个地方的包中会很方便。我在scipy.misc里没看到，转成普通空间再转回log感觉很浪费。

Answer 1

可以使用 gammaln，但是当 N >> k 时，减法会损失精度。这可以通过与 beta 函数的关系来避免:

from numpy import log
from scipy.special import betaln

def binomln(n, k):
    # Assumes binom(n, k) >= 0
    return -betaln(1 + n - k, 1 + k) - log(n + 1)