def combination(n, k): """ Code for combination formula i.e. "n choose k" """ from math import factorial return factorial(n) / (factorial(n - k) * factorial(k))