N Choose K Python