Mercurial > hg > config
diff python/example/factorial.py @ 784:47a434dd5068
example: factorial using reduce
author | Jeff Hammel <k0scist@gmail.com> |
---|---|
date | Thu, 01 Sep 2016 15:15:23 -0700 |
parents | |
children |
line wrap: on
line diff
--- /dev/null Thu Jan 01 00:00:00 1970 +0000 +++ b/python/example/factorial.py Thu Sep 01 15:15:23 2016 -0700 @@ -0,0 +1,34 @@ +#!/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()