python?argparse如何使用

這篇文章主要介紹“python argparse如何使用”的相關(guān)知識(shí)，小編通過(guò)實(shí)際案例向大家展示操作過(guò)程，操作方法簡(jiǎn)單快捷，實(shí)用性強(qiáng)，希望這篇“python argparse如何使用”文章能幫助大家解決問(wèn)題。

一、前言

argparse是python的一個(gè)命令行參數(shù)解析包，在代碼需要頻繁修改參數(shù)時(shí)，方便使用，主要用法就是在命令行輸入自己想要修改的參數(shù)。

二、關(guān)于argparse用法常用框架

import argparse

def get_parser():

    # argparse.ArgumentParser生成argparse對(duì)象 description為描述信息，當(dāng)在命令行輸入需要顯示幫助信息時(shí)，會(huì)顯示

    parser = argparse.ArgumentParser(description="pytorch unet training")

    # 路徑參數(shù)設(shè)置 help為參數(shù)的幫助信息

    parser.add_argument("--data_path", default="./", help="DRIVE root")

    # 預(yù)測(cè)類(lèi)別數(shù)量 type如果不指定需要輸入的是str類(lèi)型

    parser.add_argument("--num_classes", default=1, type=int)

    # 指定設(shè)備使用

    parser.add_argument("--device", default="cuda", help="training device")

    # 指定batch size大小 "-b", "--batch_size"表示兩個(gè)都可以在命令行使用

    parser.add_argument("-b", "--batch_size", default=4, type=int)

    return parser

if __name__ =='__main__':

    parser = get_parser()

    args = parser.parse_args()

    print(args)

如上get_parser()為本文會(huì)介紹的一些使用方法，這個(gè)python文件名為python_argparse_test1.py，其中

parser = argparse.ArgumentParser(description="pytorch unet training")用來(lái)創(chuàng)建parser對(duì)象

add_argument()用來(lái)增加參數(shù)

args = parser.parse_args()中parse_args()獲取解析的參數(shù)

1.獲取參數(shù)列表

當(dāng)命令行輸入python python_argparse_test1.py時(shí)，打印args獲取到的參數(shù)得到：

Namespace(batch_size=4, data_path='./', device='cuda', num_classes=1)

表示這個(gè)參數(shù)解析器解析到的參數(shù)列表

2.獲取幫助信息

輸入python python_argparse_test1.py -h或者python python_argparse_test1.py --help顯示信息，其中usage顯示了它的用法，pytorch unet training為創(chuàng)建對(duì)象時(shí)的description,再下面是各個(gè)參數(shù)信息與用法

3.命令行修改參數(shù)

import argparse

def get_parser():

    # argparse.ArgumentParser生成argparse對(duì)象 description為描述信息，當(dāng)在命令行輸入需要顯示幫助信息時(shí)，會(huì)顯示

    parser = argparse.ArgumentParser(description="pytorch unet training")

    # 路徑參數(shù)設(shè)置 help為參數(shù)的幫助信息 default為默認(rèn)參數(shù)

    parser.add_argument("--data_path", default="./", help="DRIVE root")

    # 預(yù)測(cè)類(lèi)別數(shù)量 type如果不指定需要輸入的是str類(lèi)型

    parser.add_argument("--num_classes", default=1, type=int)

    # 指定設(shè)備使用

    parser.add_argument("--device", default="cuda", help="training device")

    # 指定batch size大小 "-b", "--batch_size"表示兩個(gè)都可以在命令行使用

    parser.add_argument("-b", "--batch_size", default=4, type=int)

    return parser

if __name__ =='__main__':

    parser = get_parser()

    args = parser.parse_args()

    print("data_path: ",args.data_path)

    print("num_classes: ", args.num_classes)

命令行輸入：python python_argparse_test1.py --data_path Desktop --num_classer 4，得到結(jié)果如下：

data_path:  Desktop

num_classes:  4

可以看到通過(guò)命令行確實(shí)修改了參數(shù)

4.'_'與"__"的使用

if __name__ =='__main__':

    parser = get_parser()

    args = parser.parse_args()

    print("-b: ",args.b)

    print("--batch_size: ", args.batch_size)

命令行輸入python python_argparse_test1.py -b 10 --batch_size 20，這時(shí)會(huì)報(bào)錯(cuò)：

這是因?yàn)楫?dāng)'_'和'__'同時(shí)存在時(shí)，系統(tǒng)默認(rèn)后者為參數(shù)名

把上述代碼改為：

if __name__ =='__main__':

    parser = get_parser()

    args = parser.parse_args()

    print("--batch_size: ",args.batch_size)

但命令行不受影響，繼續(xù)執(zhí)行命令 python python_argparse_test1.py -b 10得到：

--batch_size:  10

5.type的使用

type會(huì)把輸入的命令行字符強(qiáng)制轉(zhuǎn)化為type的類(lèi)型

if __name__ =='__main__':

    parser = get_parser()

    args = parser.parse_args()

    print("--batch_size type: ",type(args.batch_size))

命令行輸入：python python_argparse_test1.py --batch_size  '10'得到：

--batch_size type:  <class 'int'>

6.required:用來(lái)表示這個(gè)參數(shù)是否需要提供

parser.add_argument("--num_classes", default=1, type=int, required=True)

如果輸入命令python python_argparse_test1.py

這將會(huì)報(bào)錯(cuò)提示確實(shí)需要的參數(shù)

python_argparse_test1.py: error: the following arguments are required: --num_classes

7.choices選擇參數(shù)    

parser.add_argument('-arch', required=True, choices=['alexnet', 'vgg'])

如果運(yùn)行命令：python python_argparse_test1.py -arch cnn

這將會(huì)報(bào)錯(cuò)

python_argparse_test1.py: error: argument -arch: invalid choice: 'cnn' (choose from 'alexnet', 'vgg')

關(guān)于“python argparse如何使用”的內(nèi)容就介紹到這里了，感謝大家的閱讀。如果想了解更多行業(yè)相關(guān)的知識(shí)，可以關(guān)注億速云行業(yè)資訊頻道，小編每天都會(huì)為大家更新不同的知識(shí)點(diǎn)。