课程咨询: 400-996-5531 / 投诉建议: 400-111-8989
认真做教育 专心促就业
Python解析命令行读取参数有两种方式:sys.argv和argparse
如果脚本很简单或临时使用,没有多个复杂的参数选项,可以直接利用sys.argv将脚本后的参数依次读取(读进来的默认是字符串格式)。
1import sys2print("输入的参数为:%s" % sys.argv[1])命令行执行效果:
1>python demo.py 12输入的参数为:12、argparse如果参数很多,比较复杂,并且类型不统一,那么argparse可以很好的解决这些问题,下面一个实例解释了argparse的基本使用方法。
1import argparse2# description参数可以用于描述脚本的参数作用,默认为空3parser=argparse.ArgumentParser(description="A description of what the program does")4parser.add_argument('--toy','-t',action='store_true',help='Use only 50K samples of data')5 parser.add_argument('--num_epochs',choices=[5,10,20],default=5,type=int,help='Number of epochs.')6parser.add_argument("--num_layers", type=int, required=True, help="Network depth.")78args=parser.parse_args()9print(args)10print(args.toy,args.num_epochs,args.num_layers)命令行执行效果:
1>python demo.py --num_epochs 10 --num_layers 102Namespace(num_epochs=10, num_layers=10, toy=False)3False 10 102.1.基本使用
1parser.add_argument('--toy','-t',action='store_true',help='Use only 50K samples of data')
--toy:为参数名称;
-t:为参数别称;
action='store_true':参数是否使用,如果使用则为True,否则为False。
help:参数说明
2.2.相关参数
choices:候选值,输出参数必须在候选值里面,否如会出现下面的结果:
1>python demo.py -t --num_epochs 30 --num_layers 102usage: demo.py [-h] [--toy] [--num_epochs {5,10,20}] --num_layers NUM_LAYERS3demo.py: error: argument --num_epochs: invalid choice: 30 (choose from 5, 10, 20)default:默认值,如果不输入参数,则使用该默认值
1>python demo.py -t --num_layers 102Namespace(num_epochs=5, num_layers=10, toy=True)3True 5 10int:参数类型
实例2
1parser.add_argument("--num_layers", type=int, required=True, help="Network depth.")required:为必选参数,如果不输入,则出现以下错误:
1>python demo.py -t --num_epochs 102usage: demo.py [-h] [--toy] [--num_epochs {5,10,20}] --num_layers NUM_LAYERS3demo.py: error: the following arguments are required: --num_layers实例3
-h:输出参数使用说明信息
1>python demo.py -h2usage: demo.py [-h] [--toy] [--num_epochs {5,10,20}] --num_layers NUM_LAYERS34A description of what the program does56optional arguments:7-h, --help show this help message and exit8--toy, -t Use only 50K samples of data9--num_epochs {5,10,20}10 Number of epochs.11 --num_layers NUM_LAYERS12 Network depth.