web-dev-qa-db-ja.com

Pythonに数学nCr関数はありますか?

可能性のある重複:
統計:Pythonの組み合わせ
組み合わせと順列を効率的にカウント
pythonのプロジェクトオイラー問題(問題53)

pythonの数学ライブラリに組み込まれているのがnCr(n Choose r)関数かどうかを確認したい:

enter image description here

これはプログラムできることは理解していますが、実行する前に既に組み込まれているかどうかを確認したいと思いました。

136
James Mertz

次のプログラムは、nCrname__を効率的な方法で計算します(階乗などの計算と比較して)

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
143
dheerosaur

繰り返したいですか? 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 を使用します。

import math

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

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

Python 3では、オーバーフローを避けるために、//の代わりに整数除算/を使用します。

return f(n) // f(r) // f(n-r)

出力

6
152
Mark Tolonen