前往小程序,Get更优阅读体验!
立即前往
首页
学习
活动
专区
工具
TVP
发布
社区首页 >专栏 >Python 中argparse模块的使用

Python 中argparse模块的使用

作者头像
用户1332428
发布2018-07-30 14:53:03
7680
发布2018-07-30 14:53:03
举报
文章被收录于专栏:人工智能LeadAI人工智能LeadAI

正文共495个字,预计阅读时间5分钟。

Python解析命令行读取参数有两种方式:sys.argv和argparse

1、sys.argv

如果脚本很简单或临时使用,没有多个复杂的参数选项,可以直接利用sys.argv将脚本后的参数依次读取(读进来的默认是字符串格式)。

代码语言:javascript
复制
1import sys
2print("输入的参数为:%s" % sys.argv[1])

命令行执行效果:

代码语言:javascript
复制
1>python demo.py 1
2输入的参数为:1

2、argparse

如果参数很多,比较复杂,并且类型不统一,那么argparse可以很好的解决这些问题,下面一个实例解释了argparse的基本使用方法。

代码语言:javascript
复制
 1import argparse
 2# 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.")
 7
 8args=parser.parse_args()
 9print(args)
10print(args.toy,args.num_epochs,args.num_layers)

命令行执行效果:

代码语言:javascript
复制
1>python demo.py --num_epochs 10 --num_layers 10
2Namespace(num_epochs=10, num_layers=10, toy=False)
3False 10 10

2.1.基本使用

代码语言:javascript
复制
1parser.add_argument('--toy','-t',action='store_true',help='Use only 50K samples of data')

--toy:为参数名称; -t:为参数别称; action='store_true':参数是否使用,如果使用则为True,否则为False。

代码语言:javascript
复制
1>python demo.py -t --num_epochs 10 --num_layers 10
2Namespace(num_epochs=10, num_layers=10, toy=True)
3True 10 10 # 对比和上次执行的区别

help:参数说明

2.2.相关参数

实例1

代码语言:javascript
复制
1parser.add_argument('--num_epochs',choices=[5,10,20],default=5,type=int,help='Number of epochs.')

choices:候选值,输出参数必须在候选值里面,否如会出现下面的结果:

代码语言:javascript
复制
1>python demo.py -t --num_epochs 30 --num_layers 10
2usage: demo.py [-h] [--toy] [--num_epochs {5,10,20}] --num_layers NUM_LAYERS
3demo.py: error: argument --num_epochs: invalid choice: 30 (choose from 5, 10, 20)

default:默认值,如果不输入参数,则使用该默认值

代码语言:javascript
复制
1>python demo.py -t  --num_layers 10
2Namespace(num_epochs=5, num_layers=10, toy=True)
3True 5 10

int:参数类型

实例2

代码语言:javascript
复制
1parser.add_argument("--num_layers", type=int, required=True, help="Network depth.")

required:为必选参数,如果不输入,则出现以下错误:

代码语言:javascript
复制
1>python demo.py -t --num_epochs 10
2usage: demo.py [-h] [--toy] [--num_epochs {5,10,20}] --num_layers NUM_LAYERS
3demo.py: error: the following arguments are required: --num_layers

实例3

-h:输出参数使用说明信息

代码语言:javascript
复制
 1>python demo.py -h
 2usage: demo.py [-h] [--toy] [--num_epochs {5,10,20}] --num_layers NUM_LAYERS
 3
 4A description of what the program does
 5
 6optional arguments:
 7-h, --help            show this help message and exit
 8--toy, -t             Use only 50K samples of data
 9--num_epochs {5,10,20}
10                Number of epochs.
11 --num_layers NUM_LAYERS
12                Network depth.

原文链接:https://www.jianshu.com/p/8f781143c524

本文参与 腾讯云自媒体分享计划,分享自微信公众号。
原始发表:2018-07-25,如有侵权请联系 cloudcommunity@tencent.com 删除

本文分享自 人工智能LeadAI 微信公众号,前往查看

如有侵权,请联系 cloudcommunity@tencent.com 删除。

本文参与 腾讯云自媒体分享计划  ,欢迎热爱写作的你一起参与!

评论
登录后参与评论
0 条评论
热度
最新
推荐阅读
目录
  • 实例1
领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档