快捷導(dǎo)航

Python如何使用argparse模塊處理命令行參數(shù)

更新時(shí)間：2019年12月11日 10:20:27 作者：MrDoghead

這篇文章主要介紹了Python如何使用argparse模塊處理命令行參數(shù),文中通過(guò)示例代碼介紹的非常詳細(xì)，對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值,需要的朋友可以參考下

介紹

argparse是python3標(biāo)準(zhǔn)庫(kù)里用來(lái)處理命令行參數(shù)的模塊

使用三步走：

創(chuàng)建 ArgumentParser() 對(duì)象
調(diào)用 add_argument() 方法添加參數(shù)
使用 parse_args() 解析添加的參數(shù)

實(shí)例

我們先來(lái)創(chuàng)建一個(gè)python文件

vim test.py

然后開始在輸入模式（i）下輸入代碼

import argparse # 導(dǎo)入模塊

parser = argparse.ArgumentParser() # 創(chuàng)建一個(gè)對(duì)象
parser.add_argument("square", help="pls enter a num", type=int) # 增加一個(gè)叫square的參數(shù)，類型必須是int
parser.add_argument("input_str", help="pls enter a string", type=str) # 增加一個(gè)叫input_str的參數(shù)，類型必須是str
args = parser.parse_args() # 解析
print(args.square ** 2)
print(args.input_str)

然后，讓我們來(lái)運(yùn)行一下

python test.py 3 "I am a string!"
# 結(jié)果是:
# 9
# I am a string!

如果要查看文件的參數(shù)信息

python test.py -h

如果要設(shè)定可選參數(shù)

parser = argparse.ArgumentParser()
parser.add_argument("--argument1", “-arg1”, help="display the first argument", type=int) # 參數(shù)名前加--，取別名為-arg1
parser.add_argument("--argument2", choices=[5,10,20], help="display the second argument", type=int) # choices 為限定候選列表
args = parser.parse_args()

print(args)
if args.argument1 and args.argument2:
  print(args.argument1 + args.argument2)

執(zhí)行一下

python test.py
# Namespace(argument1=None, argument2=None)

python test.py -arg1 3 --argument2 5
# Namespace(argument1=3, argument2=5)
# 8

如果要設(shè)定必選參數(shù)

使用required

parser.add_argument("--num", type=int, required=True, help="must enter a num")

參數(shù)還可以是文件

parser.add_argument('file', type=argparser.FileType('r')) # 讀取文件
parser.add_argument('file', default='text.txt', type=argparser.FileType('r')) # 讀取文件，預(yù)定義參數(shù)

以上就是本文的全部?jī)?nèi)容，希望對(duì)大家的學(xué)習(xí)有所幫助，也希望大家多多支持腳本之家。

您可能感興趣的文章: