argparse 是 Python 内置的一个用于命令项选项与参数解析的模块,通过在程序中定义好需要的参数,argparse 将会从 sys.argv 中解析出这些参数,并自动生成帮助和使用信息。
import argparse # 导入模板 parser = argparse.ArgumentParser() # 创建parser parser.add_argument() # 添加参数 args = parser.parse_args() # 参数解析
创建一个解析对象
添加参数
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
参考博客: