python中有数学nCr函数吗?

我想看看python的数学库中是否内置了nCr (n Choose r)函数:

enter image description here

我知道这是可以编程的，但我想在我这样做之前，我应该检查一下它是否已经内置了。

419489 次浏览
小开

你想要迭代吗?itertools.combinations。常见用法:

>>> import itertools
>>> itertools.combinations('abcd',2)
<itertools.combinations object at 0x01348F30>
>>> list(itertools.combinations('abcd',2))
[('a', 'b'), ('a', 'c'), ('a', 'd'), ('b', 'c'), ('b', 'd'), ('c', 'd')]
>>> [''.join(x) for x in itertools.combinations('abcd',2)]
['ab', 'ac', 'ad', 'bc', 'bd', 'cd']

如果你只需要计算公式，可以使用math.factorial，但对于大的组合不是很快，但请参阅下面的math.comb，以获得Python 3.8+中可用的优化计算:

import math


def nCr(n,r):
f = math.factorial
return f(n) // f(r) // f(n-r)


if __name__ == '__main__':
print nCr(4,2)

输出:

6

从Python 3.8开始，可以使用math.comb，并且速度更快:

>>> import math
>>> math.comb(4,2)
6
小开
最佳答案

下面的程序以高效的方式计算nCr(与计算阶乘等相比)

import operator as op
from functools import reduce


def ncr(n, r):
r = min(r, n-r)
numer = reduce(op.mul, range(n, n-r, -1), 1)
denom = reduce(op.mul, range(1, r+1), 1)
return numer // denom  # or / in Python 2

从Python 3.8开始，二项式系数可以在标准库中以math.comb的形式使用:

>>> from math import comb
>>> comb(10,3)
120

提交答案