python

关注公众号 jb51net

关闭
首页 > 脚本专栏 > python > Python的argparse的应用场景

关于Python参数解析器argparse的应用场景

作者:Taylor不想被展开

这篇文章主要介绍了关于Python参数解析器argparse的应用场景,argparse 模块使编写用户友好的命令行界面变得容易,程序定义了所需的参数,而 argparse 将找出如何从 sys.argv 中解析这些参数,需要的朋友可以参考下

一、什么是argparse

argparse 是一个用来解析命令行参数的 Python 库,它是 Python 标准库的一部分。

argparse 模块使编写用户友好的命令行界面变得容易

程序定义了所需的参数,而 argparse 将找出如何从 sys.argv 中解析这些参数。

argparse 模块还会自动生成帮助和使用消息,并在用户为程序提供无效参数时发出错误。

简而言之,就是汇总函数接口,方便传参。

二、argparse的使用场景

在深度学习代码中,特别是在train.py或test.py中,argparse最常被使用。

因为在这些文件中,通常要对其他.py文件中定义的class或function进行调用,因此必然要传入大量的参数,如果不对这些参数进行集中编写,那么在冗长的代码中去找参数的定义位置或传参的位置将会十分麻烦。

argparse为用户提供了集中编写所需参数的机会,下面的使用教程中我会结合简单而具体的例子进行说明。

三、argparse的使用流程

1.创建一个解析器

即创建 ArgumentParser() 对象

parser = argparse.ArgumentParser(description='test')

description:大多数对 ArgumentParser 构造方法的调用都会使用 description= 关键字参数。

这个参数简要描述这个程度做什么以及怎么做。

在帮助消息中,这个描述会显示在命令行用法字符串和各种参数的帮助消息之间。

2.添加参数

即调用 add_argument() 方法添加参数

parser.add_argument('--binary', action='store_true', default=False, help='action=store_true表示默认为False,当在命令行中使用--binary时自动变为True')
parser.add_argument('--number', type=int, default=6, help='普通数据,type可指定类型.')
parser.add_argument('--epochs', type=int, default=300, help='Number of epochs to train.')

这里说一下如何定义参数的格式,可以暂时略过不看,粗体部分为比较常用的参数,其实每个都很常用:

ArgumentParser.add_argument(name or flags...[, action][, nargs][, const][, default][, type][, choices][, required][, help][, metavar][, dest])

各个参数解释如下:

3.解析参数

即使用 parse_args() 解析添加的参数

args = parser.parse_args()

4.简单而具体的例子

例如我现在有一个.py文件,我现在定义了好多好多函数(为了方便我只定义了两个),我现在需要集中管理我传入的参数,使用argparse就会非常方便,对于本.py文件中的传参一目了然。

import argparse
parser = argparse.ArgumentParser(description='test')
parser.add_argument('--spring', action='store_true', default=False, help='nothing.')
parser.add_argument('--summer', type=int, default=6, help='num.')
parser.add_argument('--fall', type=int, default=11, help='num.')
parser.add_argument('--winter', type=int, default=100, help='Number of epochs to train.')
args = parser.parse_args()
# FLAGS, unparsed = parser.parse_known_args() 这种解析方法用的比较少,有兴趣可以了解一下
def sum_function(num2sum_1,num2sum_2):
    sum_result = num2sum_1 + num2sum_2
    return sum_result
def multiply_function(num2multiply_1,num2multiply_2):
    multiply_result = num2multiply_1 * num2multiply_2
    return multiply_result
print('args.spring的值是{}'.format(args.spring))
print('args.summer的值是{}'.format(args.summer))
print(sum_function(args.summer,args.fall))
print(multiply_function(args.summer,args.winter))

# 运行结果
args.spring的值是False
args.summer的值是6
17
600

你可以发现,对本文件中的函数进行传参,无论有多少参数需要传,我们只要在argparse中写好需要传的参数,在传参时传入args.参数名,需要修改传参值时,统一在argparse中进行修改即可,这样就方便了参数的管理。

5.如何修改参数值

运行结果:
args.spring的值是True
args.summer的值是10
30
1000

到此这篇关于关于Python参数解析器argparse的应用场景的文章就介绍到这了,更多相关Python的argparse的应用场景内容请搜索脚本之家以前的文章或继续浏览下面的相关文章希望大家以后多多支持脚本之家!

您可能感兴趣的文章:
阅读全文