前往小程序,Get更优阅读体验!
立即前往
首页
学习
活动
专区
工具
TVP
发布
社区首页 >专栏 >人生要找对路,代码要找对路口 | argparse实例详解

人生要找对路,代码要找对路口 | argparse实例详解

作者头像
AI算法与图像处理
发布2019-05-22 20:12:06
5860
发布2019-05-22 20:12:06
举报

导读

刚开始在GitHub找别人源码的时候,经常遇到一个开源项目有好多star,但却不知道怎么用,因为我不懂入口在哪,现在不一样了,我膨胀了。。。不,我学会了这个,而且还发现了一个超叼的东西,准备下周整理一下发出来,超叼的分类器集合,我调试好之后会说明如何使用,敬请期待哈哈哈~~~~

进入正题

什么是argsparse

argsparse是python的命令行解析的标准模块,内置于python,不需要安装。这个库可以让我们直接在命令行中就可以向程序中传入参数并让程序运行。

如何传入参数

我们先在桌面新建“arg学习”的文件夹,在该文件夹中新建demo.py文件,来看一个最简单的argsparse库的使用的例子。

代码语言:javascript
复制
import argparse

parser = argparse.ArgumentParser(description='命令行中传入一个数字')
#type是要传入的参数的数据类型  help是该参数的提示信息
parser.add_argument('integers', type=str, help='传入的数字')

args = parser.parse_args()

#获得传入的参数
print(args)

在这个代码中,我们在命令行传入一个数字。使用方法是打开命令行,先将工作目录cd到 arg学习

代码语言:javascript
复制
cd desktop/arg学习

然后再命令行中输入 python demo.py-h 或者 python demo.py--help , 这里我输入的是

代码语言:javascript
复制
python demo.py -h

在命令行中看到demo.py的运行结果如下

代码语言:javascript
复制
usage: demo.py [-h] integers

命令行中传入数字

positional arguments:
  integers    传入的数字

optional arguments:
  -h, --help  show this help message and exit

现在我们在命令行中给demo.py 传入一个参数5,

代码语言:javascript
复制
python demo.py 5

运行,得到的运行结果是

代码语言:javascript
复制
Namespace(integers='5')

操作args字典

其实得到的这个结果 Namespace(integers='5')是一种类似于python字典的数据类型。

我们可以使用 arg.参数名来提取这个参数

代码语言:javascript
复制
import argparse

parser = argparse.ArgumentParser(description='命令行中传入一个数字')
#type是要传入的参数的数据类型  help是该参数的提示信息
parser.add_argument('integers', type=str, help='传入的数字')

args = parser.parse_args()

#获得integers参数
print(args.integers)

在命令行中运行 python demo.py5 , 运行结果为

代码语言:javascript
复制
5

传入多个参数

现在在命令行中给demo.py 传入多个参数,例如传入1,2,3,4四个数字

代码语言:javascript
复制
python demo.py 1 2 3 4

运行报错

代码语言:javascript
复制
usage: demo.py [-h] integers
demo.py: error: unrecognized arguments: 2 3 4

不能识别2 3 4,看源代码我们知道integers这个参数是位置参数,说明第一个数 1 是能识别。这里我们需要重新更改demo.py代码

代码语言:javascript
复制
import argparse

parser = argparse.ArgumentParser(description='命令行中传入一个数字')
parser.add_argument('integers', type=str, nargs='+',help='传入的数字')
args = parser.parse_args()

print(args.integers)

nargs是用来说明传入的参数个数,'+' 表示传入至少一个参数。这时候再重新在命令行中运行 python demo.py1234 得到

代码语言:javascript
复制
['1', '2', '3', '4']

改变数据类型

add_argument中有type参数可以设置传入参数的数据类型。我们看到代码中有type这个关键词,该关键词可以传入list, str, tuple, set, dict等。例如我们把上面的type=str,改成type=int,这时候我们就可以进行四则运算。

代码语言:javascript
复制
import argparse

parser = argparse.ArgumentParser(description='命令行中传入一个数字')
parser.add_argument('integers', type=int, nargs='+',help='传入的数字')
args = parser.parse_args()

#对传入的数据进行加总
print(sum(args.integers)

在命令行中输入 python demo.py1234, 运行结果为

代码语言:javascript
复制
10

位置参数

在命令行中传入参数时候,传入的参数的先后顺序不同,运行结果往往会不同,这是因为采用了位置参数,例如

代码语言:javascript
复制
import argparse

parser = argparse.ArgumentParser(description='姓名')
parser.add_argument('param1', type=str,help='姓')
parser.add_argument('param2', type=str,help='名')
args = parser.parse_args()

#打印姓名
print(args.param1+args.param2)

在命令行中分别输入 python demo.py张三python demo.py三张,得到的 运行结果分别为

代码语言:javascript
复制
张三

代码语言:javascript
复制
三张

如果我们将代码 parser.add_argument('param1',type=str,help='姓')

parser.add_argument('param2',type=str,help='名') 互换位置,即第4行和第五行代码,再重新运行

python demo.py张三python demo.py三张,得到的 运行结果分别为

代码语言:javascript
复制
三张

代码语言:javascript
复制
张三

*** 可选参数(重点——因为经常会用到)

为了在命令行中避免上述位置参数的bug(容易忘了顺序),可以使用可选参数,这个有点像关键词传参,但是需要在关键词前面加 --,例如

代码语言:javascript
复制
import argparse

parser = argparse.ArgumentParser(description='姓名')
parser.add_argument('--family', type=str,help='姓')
parser.add_argument('--name', type=str,help='名')
args = parser.parse_args()

#打印姓名
print(args.family+args.name)

在命令行中输入

代码语言:javascript
复制
python demo.py --family=张 --name=三
# 或者输入 python demo.py --family 张 --name 三

运行结果

代码语言:javascript
复制
张三

可选参数虽然写法比较繁琐,但是增加了命令行中的可读性,不容易因为参数传入顺序导致数据错乱。

默认值

add_argument中有一个default参数。有的时候需要对某个参数设置默认值,即如果命令行中没有传入该参数的值,程序使用默认值。如果命令行传入该参数,则程序使用传入的值。具体请看下面的例子

代码语言:javascript
复制
import argparse

parser = argparse.ArgumentParser(description='姓名')
parser.add_argument('--family', type=str, default='张',help='姓')
parser.add_argument('--name', type=str, default='三', help='名')
args = parser.parse_args()

#打印姓名
print(args.family+args.name)

在命令行中分别输入 python demo.pypython demo.py--family=李

运行结果分别为

代码语言:javascript
复制
张三

代码语言:javascript
复制
李三

必需参数

add_argument有一个required参数可以设置该参数是否必需。

代码语言:javascript
复制
import argparse

parser = argparse.ArgumentParser(description='姓名')
parser.add_argument('--family', type=str, help='姓')
parser.add_argument('--name', type=str, required=True, default='', help='名')
args = parser.parse_args()

#打印姓名
print(args.family+args.name)

在命令行中输入 python demo.py--family=张,运行结果

代码语言:javascript
复制
usage: demo.py [-h] [--family FAMILY] --name NAME
demo.py: error: the following arguments are required: --name

因为可选参数 name的required=True,所以必须要传入。如果我们将其更改为False,程序运行结果

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

本文分享自 AI算法与图像处理 微信公众号,前往查看

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

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

评论
登录后参与评论
0 条评论
热度
最新
推荐阅读
目录
  • 进入正题
  • 如何传入参数
  • 操作args字典
  • 传入多个参数
  • 改变数据类型
  • 位置参数
  • *** 可选参数(重点——因为经常会用到)
  • 默认值
  • 必需参数
领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档