Mercurial > hg > config
view python/example/factorial.py @ 836:0fa456446fea
use module docstring for description, as intended
author | Jeff Hammel <k0scist@gmail.com> |
---|---|
date | Thu, 09 Mar 2017 14:22:32 -0800 |
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()