关于python的argparse模块使用的一个例程

参考https://www.jianshu.com/p/fef2d215b91d

argparse模块可以方便地在命令行设置一些超参数

# encoding: utf-8


import argparse


parser = argparse.ArgumentParser(description="Exponent Computing")
parser.add_argument("-a", metavar='base', type=int, help="底数")
parser.add_argument("-b", metavar='exponent', type=int, help="指数", default=1)
parser.add_argument("-v", "--verbosity", type=int, choices=[0, 1, 2], default=2, help="increase output verbosity")

args = parser.parse_args()

answer = args.a**args.b

if args.verbosity == 1:
    print("the square of {} equals {}".format(args.square, answer))
elif args.verbosity == 2:
    print("{}^{} == {}".format(args.a, args.b, answer))
else:
    print(answer)

演示结果：

(L:\ProgramData\Anaconda3\envs\pytorch031) L:\>python untitled1.py -h
usage: untitled1.py [-h] [-a base] [-b exponent] [-v {0,1,2}]

Exponent Computing

optional arguments:
  -h, --help            show this help message and exit
  -a base               底数
  -b exponent           指数
  -v {0,1,2}, --verbosity {0,1,2}
                        increase output verbosity

(L:\ProgramData\Anaconda3\envs\pytorch031) L:\>python untitled1.py -a 2 -b 4
2^4 == 16