view python/example/factorial.py @ 789:2064d617fe56

add a suggested fix for posterity
author Jeff Hammel <k0scist@gmail.com>
date Wed, 28 Sep 2016 17:58:48 -0700
parents 47a434dd5068
children
line wrap: on
line source

#!/usr/bin/env python
"""how many 0s are in a factorial"""

import argparse
import sys

def factorial(N):
    """factorial of `N`"""
    if N == 1:
        return 1
    return reduce(int.__mul__, range(2, N+1))

def factorial_zeros(N):
    """how many 0s are in a factorial?"""

    return N/5

def main(args=sys.argv[1:]):
    """CLI"""

    parser = argparse.ArgumentParser(description=__doc__)
    parser.add_argument('N', type=int, nargs='+')
    options = parser.parse_args(args)

    # sanity
    if any([i < 1 for i in options.N]):
        parser.error("Input values must be >= 1")

    for i in options.N:
        f =  factorial(i)
        print f

if __name__ == '__main__':
    main()