Python命令行参数解析：sys.argv和argparse怎么用
作者：猛男炸鸡翅
还在为Python脚本无法灵活接收命令行参数而烦恼吗？本文用简单例子带你搞懂sys.argv和argparse的区别与用法,无论是简单传参还是训练模型时设置batch_size、epochs,都能轻松掌握,让你的脚本更专业
一. sys
假设有一个函数文件为test.py，内容为：
import sys a = sys.argv[0] b = sys.argv[1] print(a) # test.py print(b) # 1000
运行python test.py 1000，可以发现sys.argv[0]就是test.py本身，sys.argv[1]就是传入的第一个命令行参数1000，注意这里的1000是str类型。
如果命令输入多个参数（按空格隔开），会把多个参数存在sys.argv里，实际上sys.argv就是个列表类型，用来存储命令行参数。例子如下：
import sys print(type(sys.argv)) # list c = sys.argv[1:] print(c)
运行python test.py 1 2 3 4，返回结果如下图所示：
二. argparse
argparser模块经常会在训练模型的时候看到，例如gpu_ids的选择，batch_size，epochs等等，argparser这个模块也可以传入命令行参数。
具体用法如下所述：
- 首先导入模块，创建一个解析器:
import argparse parser = argparse.ArgumentParser(description='description') # description后面是描述信息。
- 添加参数，如下所示添加了三种类型的参数，设置了默认值，表示如果不从命令行传入，就使用默认值 。
parser.add_argument('--gpu_ids',default='0', type=str, help='gpu_ids: e.g. 0 0,1,2 0,2')
parser.add_argument('--batchsize', default=256, type=int, help='batchsize')
parser.add_argument('--use_dense', action='store_true', help='use densenet121' )
- 解析参数
args = parser.parse_args()
- 打印结果，看一下参数：
# 不传入命令行参数 print(args) print(args.gpu_ids) print(args.batchsize) print(args.use_dense)
不传入参数，则打印出默认参数。
传入命令行参数，运行python test.py --batchsize 64 --use_dense 则有
三. 总结
以上为个人经验，希望能给大家一个参考，也希望大家多多支持脚本之家。