Argparse函数的用法

1. argparse基本用法

通过argpares函数，我们可以在.py文件外部执行的时候调整内部参数。

import math
import argparse  # 导入argparse模块
# 用来装载参数的容器，后面的description可以采用xxx.py -h的方法查看，这个方法也会显示后面add_argument里面的信息。
parser = argparse.ArgumentParser(description='Calculate volume of a cylinder')
# 用法1,这里radius前面没加--，说明是必选参数
parser.add_argument('radius', type=int, help='Radius of cylinder')

# 用法2：--dt表示可选参数，dest表示调用args.device_type来获得外部输入参数，而不是调用args.dt
parser.add_argument("--dt", dest='device_type', type=str,
        help="grid, ourense, sycamore, rochester, tokyo, aspen-4, or eagle")

# 用法3：这里内部action后面的参数指定了如果在命令行里采用--sabre声明了一个值，就会触发的动作,
# 这里是store_true，表示将sabre设置为True
parser.add_argument("--sabre", action='store_true', default=False,
        help="Use sabre to get SWAP upper bound")


args = parser.parse_args()  # 获取所有参数,这一步是必须的

def cylinder_volume(radius, height):
    vol = (math.pi) * (radius**2) * (height)
    return vol
# 这一行让我们一会儿运行xx.py的时候会自动运行下面的函数
if __name__ == '__main__':
    print(cylinder_volume(args.radius, args.height))