Python argparse模块

介绍

argparse 是 Python 内置的一个用于命令项选项与参数解析的模块，通过在程序中定义好需要的参数，argparse 将会从 sys.argv 中解析出这些参数，并自动生成帮助和使用信息。

使用步骤

import argparse                     # 导入模板
parser = argparse.ArgumentParser()  # 创建parser
parser.add_argument()               # 添加参数
args = parser.parse_args()          # 参数解析

ArgumentParser

创建一个解析对象

description：命令行帮助的开始文字

add_argument

添加参数

type：指定传进来的参数的类型
default：如果命令行没有出现这个选项，那么使用default指定的默认值
help：设置这个选项的帮助信息
action：在命令行中遇到这个参数时要采取的基本行动类型

示例

import argparse

def parse_args():
    parser = argparse.ArgumentParser(description='This is a description')
    parser.add_argument('--ver', '-v', action='store_true', help='haha')
    parser.add_argument('--num', '-n', default=128, type=int)
    return parser.parse_args()

def main(args):
    print(args.ver)
    print(args.num)

if __name__ == '__main__':
    main(parse_args())

在命令行输入：

    python train.py -v -n 12

打印结果：

    True
    12

参考博客：