A fast way to calculate binomial coefficients in python (Andrew Dalke)

binomial.py

def binomial(n, k):
    """
    A fast way to calculate binomial coefficients by Andrew Dalke.
    See http://stackoverflow.com/questions/3025162/statistics-combinations-in-python
    """
    if 0 <= k <= n:
        ntok = 1
        ktok = 1
        for t in xrange(1, min(k, n - k) + 1):
            ntok *= n
            ktok *= t
            n -= 1
        return ntok // ktok
    else:
        return 0

Try this....a short way by importing comb from math library

from math import comb
def coefficient(n,r):
return comb(n,r)
print(coefficient(5,2))

Change coefficient values to your desire