一、什么是argparse

argparse 是一個用來解析命令行參數(shù)的 Python 庫，它是 Python 標準庫的一部分。

argparse 模塊使編寫用戶友好的命令行界面變得容易。

程序定義了所需的參數(shù)，而 argparse 將找出如何從 sys.argv 中解析這些參數(shù)。

argparse 模塊還會自動生成幫助和使用消息，并在用戶為程序提供無效參數(shù)時發(fā)出錯誤。

簡而言之，就是匯總函數(shù)接口，方便傳參。

二、argparse的使用場景

在深度學(xué)習(xí)代碼中，特別是在train.py或test.py中，argparse最常被使用。

因為在這些文件中，通常要對其他.py文件中定義的class或function進行調(diào)用，因此必然要傳入大量的參數(shù)，如果不對這些參數(shù)進行集中編寫，那么在冗長的代碼中去找參數(shù)的定義位置或傳參的位置將會十分麻煩。

argparse為用戶提供了集中編寫所需參數(shù)的機會，下面的使用教程中我會結(jié)合簡單而具體的例子進行說明。

三、argparse的使用流程

1.創(chuàng)建一個解析器

即創(chuàng)建 ArgumentParser() 對象

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

description：大多數(shù)對 ArgumentParser 構(gòu)造方法的調(diào)用都會使用 description= 關(guān)鍵字參數(shù)。

這個參數(shù)簡要描述這個程度做什么以及怎么做。

在幫助消息中，這個描述會顯示在命令行用法字符串和各種參數(shù)的幫助消息之間。

2.添加參數(shù)

即調(diào)用 add_argument() 方法添加參數(shù)

parser.add_argument('--binary', action='store_true', default=False, help='action=store_true表示默認為False，當在命令行中使用--binary時自動變?yōu)門rue')
parser.add_argument('--number', type=int, default=6, help='普通數(shù)據(jù)，type可指定類型.')
parser.add_argument('--epochs', type=int, default=300, help='Number of epochs to train.')

這里說一下如何定義參數(shù)的格式，可以暫時略過不看，粗體部分為比較常用的參數(shù)，其實每個都很常用：

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

各個參數(shù)解釋如下：

• name or flags：選項字符串的名字或者列表，例如 foo 或者 -f, --foo； action：命令行遇到參數(shù)時的動作，默認值是 store； store_const：表示賦值為const；
• default ：不指定參數(shù)時的默認值。
• type：命令行參數(shù)應(yīng)該被轉(zhuǎn)換成的類型；
• choices：參數(shù)可允許的值的一個容器；
• nargs ：應(yīng)該讀取的命令行參數(shù)個數(shù)，可以是具體的數(shù)字，或者是?號，當不指定值時對于 Positional argument 使用 default，對于 Optional argument 使用 const；或者是 * 號，表示 0 或多個參數(shù)；或者是 + 號表示 1 或多個參數(shù)。
• required：可選參數(shù)是否可以省略 (僅針對可選參數(shù))。
• help：參數(shù)的幫助信息，當指定為 argparse.SUPPRESS 時表示不顯示該參數(shù)的幫助信息.
• metavar：在 usage 說明中的參數(shù)名稱，對于必選參數(shù)默認就是參數(shù)名稱，對于可選參數(shù)默認是全大寫的參數(shù)名稱
• dest：解析后的參數(shù)名稱，默認情況下，對于可選參數(shù)選取最長的名稱，中劃線轉(zhuǎn)換為下劃線.
• const：action 和 nargs 所需要的常量值。
• append：將遇到的值存儲成列表，也就是如果參數(shù)重復(fù)則會保存多個值; append_const：將參數(shù)規(guī)范中定義的一個值保存到一個列表；
• count：存儲遇到的次數(shù)；此外，也可以繼承 argparse.Action 自定義參數(shù)解析；

3.解析參數(shù)

即使用 parse_args() 解析添加的參數(shù)

args = parser.parse_args()

4.簡單而具體的例子

例如我現(xiàn)在有一個.py文件，我現(xiàn)在定義了好多好多函數(shù)（為了方便我只定義了兩個），我現(xiàn)在需要集中管理我傳入的參數(shù)，使用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))

# 運行結(jié)果
args.spring的值是False
args.summer的值是6
17
600

你可以發(fā)現(xiàn)，對本文件中的函數(shù)進行傳參，無論有多少參數(shù)需要傳，我們只要在argparse中寫好需要傳的參數(shù)，在傳參時傳入args.參數(shù)名，需要修改傳參值時，統(tǒng)一在argparse中進行修改即可，這樣就方便了參數(shù)的管理。

5.如何修改參數(shù)值

• 第一，對于有default的參數(shù)，直接修改default的值即可；
• 第二，用命令行運行，例如在上面舉例的程序，我如果想把summer的值改為10，fall的值改為20，那么命令行指令就是：python train.py --summer 10 --fall 20；如果我同時還想讓spring的值成為True，那么命令行指令為：python train.py --summer 10 --fall 20 --spring

運行結(jié)果：
args.spring的值是True
args.summer的值是10
30
1000

• 第三，在Edit Configurations中進行配置，然后run即可：

到此這篇關(guān)于關(guān)于Python參數(shù)解析器argparse的應(yīng)用場景的文章就介紹到這了,更多相關(guān)Python的argparse的應(yīng)用場景內(nèi)容請搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家！

最新評論