python Argparse模塊的使用

python Argparse模塊的使用

標簽:python

參考:

Argparse Tutorial:https://docs.python.org/2.7/howto/argparse.html#id1

argparse - Parser for command-line options, arguments and sub-commands:https://docs.python.org/2.7/library/argparse.html?highlight=argparse#module-argparse

#################################################

模塊argparse是Python標準庫中推薦的命令行解析模塊

有兩個其他模塊也能完成相同的功能,即getopt和optparse模塊。其中,argparse是基于optparse模塊實現(xiàn)的,因此在用法上也非常相似。

##################################################

基礎

最簡單的一個例子,編寫prog.py:

[python]view plaincopy

importargparse

parser?=?argparse.ArgumentParser()

parser.parse_args()

運行上述腳本并且沒有添加任何參數(shù)時,輸出無結果

添加--help參數(shù)時,argparse模塊會自動輸出幫助信息,這是它的優(yōu)勢(參數(shù)--help也可以簡化為-h,這兩個參數(shù)是不需要我們編輯的)

當指定其他參數(shù)時報錯,即便如此,也可以得到一個有用的用法信息

###############################################################

位置參數(shù)

指定位置參數(shù)后,程序必須輸入該參數(shù)值才能運行

修改prog.py腳本:

[python]view plaincopy

importargparse

parser?=?argparse.ArgumentParser()

parser.add_argument("echo")

args?=?parser.parse_args()

printargs.echo

增加add_argument()方法,在這個方法中,我們指定程序將要去接受的命令行選項,在這里我們命名為echo(只是一個名字,可以隨意更改)

此時,調用程序需要輸入一個參數(shù)

方法parse_args()實際上會返回一些選項指定的值,上述例子中,選項為echo

在argparse模塊中,當你指定位置參數(shù)名為echo時,此時想要提取該參數(shù)值,必須指定為args.echo

可以在add_argument()方法中添加help選項以便該位置參數(shù)更加明了

[python]view plaincopy

importargparse

parser?=?argparse.ArgumentParser()

parser.add_argument("echo",?help="echo?the?string?you?use?here")

args?=?parser.parse_args()

printargs.echo

一般我們要求輸出位置參數(shù)是為了以后程序的運算。由于argparse將輸入的參數(shù)均視為strings類型,而有時我們需要的是int類型,比如

[python]view plaincopy

importargparse

parser?=?argparse.ArgumentParser()

parser.add_argument("square",?help="display?a?square?of?a?given?number")

args?=?parser.parse_args()

printargs.square**2

當我們需要輸入一個int類型值時,由于默認輸入?yún)?shù)值均為str類型,所以出錯。此時有兩種方法

第一種,在計算過程中轉換str為int類型:

[python]view plaincopy

printint(args.square)**2

第二種,可以在add_argument()方法中用關鍵字type指定輸入類型,即可解決:

[python]view plaincopy

parser.add_argument("square",?help="display?a?square?of?a?given?number",?help=int)

推薦使用第二種

######################################################################33

可選參數(shù)

一個簡單的例子

[python]view plaincopy

importargparse

parser?=?argparse.ArgumentParser()

parser.add_argument("--verbosity",?help="increase?output?verbosity")

args?=?parser.parse_args()

ifargs.verbosity:

print"verbosity?turned?on"

設置可選參數(shù)后,程序可以不輸入該參數(shù)值運行

如果程序運行時要賦值可選參數(shù),必須先指定該可選參數(shù)。例如

[python]view plaincopy

python?prog.py?--verbosity?hel

如果程序運行時沒有賦值給可選參數(shù)verbosity,則相應的args.verbosity也會為空(None)

如果程序使用了可選參數(shù),后面必須跟參數(shù)值,不然會報錯

默認可選參數(shù)會被賦值為str,而在實際情況中,很多時候,可選參數(shù)是作為一個標識而不是一個確切的值,僅需要確定是true或false即可,可以指定關鍵字action,賦值為"store_true"

[python]view plaincopy

importargparse

parser?=?argparse.ArgumentParser()

parser.add_argument("--verbosity",?help="increase?output?verbosity",?action="store_true")

args?=?parser.parse_args()

ifargs.verbosity:

print"verbosity?turned?on"

這樣--verbosity就可以像--help類似,只需輸入?yún)?shù)即可

############################################################################3

快捷選項(短選項)

類似命令行用法,長選項總會有相應的短選項,只需在相應的add_argument()方法中加上即可:

[python]view plaincopy

importargparse

parser?=?argparse.ArgumentParser()

parser.add_argument("-v","--verbose",?help="increase?verbosity",?action="store_true")

args?=?parser.parse_args()

ifargs.verbose:

print"verbosity?turned?on"

####################################################################

結合使用位置參數(shù)(positional arguments)和可選參數(shù)(optional arguments)

[python]view plaincopy

importargparse

parser?=?argparse.ArgumentParser()

parser.add_argument("square",?type=int,?help="display?a?square?of?a?given?number")

parser.add_argument("-v","--verbose",?action="store_true",?help="increase?output?verbosity")

args?=?parser.parse_args()

answer?=?args.square**2

ifargs.verbose:

print"the?square?of?{}?equals?{}".format(args.square,?answer)

else:

printanswer

注意,參數(shù)順序沒有關系

[python]view plaincopy

importargparse

parser?=?argparse.ArgumentParser()

parser.add_argument("square",?type=int,?help="display?a?square?of?a?given?number")

parser.add_argument("-v","--verbosity",?type=int,?help="increase?output?verbosity")

args?=?parser.parse_args()

answer?=?args.square**2

ifargs.verbosity?==2:

print"the?square?of?{}?equals?{}".format(args.square,?answer)

elifargs.verbosity?==1:

print"{}^2?==?{}".format(args.square,?answer)

else:

printanswer

上述程序中還可改進以下,可以指定可選參數(shù)的取值范圍,用關鍵字choices

[python]view plaincopy

parser.add_argument("-v","--verbosity",?type=int,?choices=[0,1,2],?help="increase?output?verbosity")

此處還有另一種方法來執(zhí)行程序,這也很普遍,并且這種方式匹配CPython可執(zhí)行程序

[python]view plaincopy

parser.add_argument("-v","--verbosity",?action="count",?help="increase?output?verbosity")

還可以指定關鍵字default來指定選項默認的值

[python]view plaincopy

parser.add_argument("-v","--verbosity",?default=2,?action="count",?help="increase?output?verbosity")

############################################################################

進階用法

可以使用多個位置參數(shù):

[python]view plaincopy

importargparse

parser?=?argparse.ArgumentParser()

parser.add_argument("x",?type=int,?help="the?base")

parser.add_argument("y",?type=int,?help="the?exponent")

parser.add_argument("-v","--verbosity",?action="count",?default=0)

args?=?parser.parse_args()

answer?=?args.x**args.y

ifargs.verbosity?>=2:

print"{}?to?the?power?{}?equals?{}".format(args.x,?args.y,?answer)

elifargs.verbosity?>=1:

print"{}^{}?==?{}".format(args.x,?args.y,?answer)

else:

printanswer

可以使用可選參數(shù)來顯示不同級別的信息:

[python]view plaincopy

parser?=?argparse.ArgumentParser()

parser.add_argument("x",?type=int,?help="the?base")

parser.add_argument("y",?type=int,?help="the?exponent")

parser.add_argument("-v","--verbosity",?action="count",?default=0)

args?=?parser.parse_args()

answer?=?args.x**args.y

ifargs.verbosity?>=2:

print"Running?'{}'".format(__file__)

ifargs.verbosity?>=1:

print"{}^{}?==?".format(args.x,?args.y)

printanswer

##################################################

以下是DIGITS中的一段源碼:

[python]view plaincopy

if__name__?=='__main__':

script_start_time?=?time.time()

parser?=?argparse.ArgumentParser(description='Classification?example?-?DIGITS')

###?Positional?arguments

parser.add_argument('caffemodel',???help='Path?to?a?.caffemodel')

parser.add_argument('deploy_file',??help='Path?to?the?deploy?file')

parser.add_argument('image',????????help='Path?to?an?image')

###?Optional?arguments

parser.add_argument('-m','--mean',

help='Path?to?a?mean?file?(*.npy)')

parser.add_argument('-l','--labels',

help='Path?to?a?labels?file')

parser.add_argument('--nogpu',

action='store_true',

help="Don't?use?the?GPU")

args?=?vars(parser.parse_args())

image_files?=?[args['image']]

classify(args['caffemodel'],?args['deploy_file'],?image_files,

args['mean'],?args['labels'],notargs['nogpu'])

print'Script?took?%s?seconds.'%?(time.time()?-?script_start_time,)

最后編輯于
?著作權歸作者所有,轉載或內容合作請聯(lián)系作者
平臺聲明:文章內容(如有圖片或視頻亦包括在內)由作者上傳并發(fā)布,文章內容僅代表作者本人觀點,簡書系信息發(fā)布平臺,僅提供信息存儲服務。
  • 序言:七十年代末,一起剝皮案震驚了整個濱河市,隨后出現(xiàn)的幾起案子,更是在濱河造成了極大的恐慌,老刑警劉巖,帶你破解...
    沈念sama閱讀 228,443評論 6 532
  • 序言:濱河連續(xù)發(fā)生了三起死亡事件,死亡現(xiàn)場離奇詭異,居然都是意外死亡,警方通過查閱死者的電腦和手機,發(fā)現(xiàn)死者居然都...
    沈念sama閱讀 98,530評論 3 416
  • 文/潘曉璐 我一進店門,熙熙樓的掌柜王于貴愁眉苦臉地迎上來,“玉大人,你說我怎么就攤上這事。” “怎么了?”我有些...
    開封第一講書人閱讀 176,407評論 0 375
  • 文/不壞的土叔 我叫張陵,是天一觀的道長。 經(jīng)常有香客問我,道長,這世上最難降的妖魔是什么? 我笑而不...
    開封第一講書人閱讀 62,981評論 1 312
  • 正文 為了忘掉前任,我火速辦了婚禮,結果婚禮上,老公的妹妹穿的比我還像新娘。我一直安慰自己,他們只是感情好,可當我...
    茶點故事閱讀 71,759評論 6 410
  • 文/花漫 我一把揭開白布。 她就那樣靜靜地躺著,像睡著了一般。 火紅的嫁衣襯著肌膚如雪。 梳的紋絲不亂的頭發(fā)上,一...
    開封第一講書人閱讀 55,204評論 1 324
  • 那天,我揣著相機與錄音,去河邊找鬼。 笑死,一個胖子當著我的面吹牛,可吹牛的內容都是我干的。 我是一名探鬼主播,決...
    沈念sama閱讀 43,263評論 3 441
  • 文/蒼蘭香墨 我猛地睜開眼,長吁一口氣:“原來是場噩夢啊……” “哼!你這毒婦竟也來了?” 一聲冷哼從身側響起,我...
    開封第一講書人閱讀 42,415評論 0 288
  • 序言:老撾萬榮一對情侶失蹤,失蹤者是張志新(化名)和其女友劉穎,沒想到半個月后,有當?shù)厝嗽跇淞掷锇l(fā)現(xiàn)了一具尸體,經(jīng)...
    沈念sama閱讀 48,955評論 1 336
  • 正文 獨居荒郊野嶺守林人離奇死亡,尸身上長有42處帶血的膿包…… 初始之章·張勛 以下內容為張勛視角 年9月15日...
    茶點故事閱讀 40,782評論 3 354
  • 正文 我和宋清朗相戀三年,在試婚紗的時候發(fā)現(xiàn)自己被綠了。 大學時的朋友給我發(fā)了我未婚夫和他白月光在一起吃飯的照片。...
    茶點故事閱讀 42,983評論 1 369
  • 序言:一個原本活蹦亂跳的男人離奇死亡,死狀恐怖,靈堂內的尸體忽然破棺而出,到底是詐尸還是另有隱情,我是刑警寧澤,帶...
    沈念sama閱讀 38,528評論 5 359
  • 正文 年R本政府宣布,位于F島的核電站,受9級特大地震影響,放射性物質發(fā)生泄漏。R本人自食惡果不足惜,卻給世界環(huán)境...
    茶點故事閱讀 44,222評論 3 347
  • 文/蒙蒙 一、第九天 我趴在偏房一處隱蔽的房頂上張望。 院中可真熱鬧,春花似錦、人聲如沸。這莊子的主人今日做“春日...
    開封第一講書人閱讀 34,650評論 0 26
  • 文/蒼蘭香墨 我抬頭看了看天上的太陽。三九已至,卻和暖如春,著一層夾襖步出監(jiān)牢的瞬間,已是汗流浹背。 一陣腳步聲響...
    開封第一講書人閱讀 35,892評論 1 286
  • 我被黑心中介騙來泰國打工, 沒想到剛下飛機就差點兒被人妖公主榨干…… 1. 我叫王不留,地道東北人。 一個月前我還...
    沈念sama閱讀 51,675評論 3 392
  • 正文 我出身青樓,卻偏偏與公主長得像,于是被迫代替她去往敵國和親。 傳聞我的和親對象是個殘疾皇子,可洞房花燭夜當晚...
    茶點故事閱讀 47,967評論 2 374

推薦閱讀更多精彩內容