Возможные дубликаты:
Статистика: комбинации в Python
эффективный подсчет комбинаций и перестановок
Проблема эйлера проекта в python (проблема 53)
Я ищу, чтобы встроить в math-библиотеку в python функцию nCr (n Select r):
Я понимаю, что это можно запрограммировать, но я подумал, что я проверил бы, если он уже встроен, прежде чем я это сделаю.
Следующая программа вычисляет 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
Вы хотите итерации? 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 __name__ == '__main__':
print nCr(4,2)
В Python 3 используйте целочисленное деление // вместо /, чтобы избежать переполнения:
return f(n) // f(r) // f(n-r)
6