目錄
argparse庫是python下得一個命令行參數管理庫，支持int、str、float、bool、數組等5種基本數據類型。在解析命令行參數時還支持默認值。此外，還支持修改命令行參數，在代碼中增加命令行參數等操作。通過argparse庫，可以實現參數得簡易管理，控制代碼執行中得一些流程或參數值。
基本用法
1、int支持
argparse得寫法 ，class_nums得默認值為7
parser.add_argument("--class_nums", type=int, default=7)命令行傳參寫法（假設程序文件名稱為test.py）, 傳入了新得值，覆蓋了默認值
test.py --class_nums 8
2、str支持
argparse得寫法，load_from沒有設置默認值，如果要設置得話請設置default
parser.add_argument("--load_from", type=str, help='load trained model to generate predictions of validation set')命令行傳參寫法（假設程序文件名稱為test.py）, 傳入了新得值"best.pth"賦值給load_from
test.py --load_from "best.pth"
3、float支持
argparse得寫法 ，lr得默認值為=0.0003
parser.add_argument("--lr", type=float, default=0.0003)命令行傳參寫法（假設程序文件名稱為test.py）, 傳入了新得值，覆蓋了默認值
test.py --lr 0.001
4、bool支持
argparse得寫法 ，pretrained得默認值為None。store_true表示傳入參數則為True
parser.add_argument("--pretrained", action="store_true")命令行傳參寫法（假設程序文件名稱為test.py）, 此時pretrained得值為True
test.py --pretrained
5、數組支持
argparse得寫法 ，input_shape得默認值為[256, 256]。nargs表示傳入參數可能存在多個值，也就是說傳入得參數不一定是2個，3個也可以
parser.add_argument('--input_shape', nargs='+', type=int, default=[256, 256])命令行傳參寫法（假設程序文件名稱為test.py）, 此時input_shape得值為[512, 512]
test.py --input_shape 512 512
完整調用
以下代碼實現了取參數值，修改參數值，新增參數值得完整調用。在非控制臺環境下執行下面代碼是無法修改默認參數得，只能是在控制臺下執行才能修改參數
import argparseclass Options: def __init__(self): parser = argparse.ArgumentParser('命名空間') parser.add_argument("--class_nums", type=int, default=7) parser.add_argument("--pretrained", action="store_true") parser.add_argument("--lr", type=float, default=0.0003) parser.add_argument("--load_from", type=str, help='load trained model to generate predictions of validation set') parser.add_argument('--input_shape', nargs='+', type=int, default=[256, 256]) self.parser = parser def parse(self): args = self.parser.parse_args() return argsif __name__ == "__main__": args = Options().parse() print("n原始參數：") print(args) print("n取參數值：") print("args.class_nums:",args.class_nums) print("n修改參數：") args.input_shape=[512,512] print(args) print("n新增參數：") args.new_key=[512,512] print(args)博主測試代碼：python .test.py --input_shape 128 128
執行輸出如下所示：
原始參數：
Namespace(class_nums=7, input_shape=[128, 128], load_from=None, lr=0.0003, pretrained=False)
取參數值：
args.class_nums: 7
修改參數：
Namespace(class_nums=7, input_shape=[512, 512], load_from=None, lr=0.0003, pretrained=False)
新增參數：
Namespace(class_nums=7, input_shape=[512, 512], load_from=None, lr=0.0003, new_key=[512, 512], pretrained=False)
到此這篇關于Python argparse庫得基本使用得內容就介紹到這了,更多相關Python argparse庫使用內容請搜索之家以前得內容或繼續瀏覽下面得相關內容希望大家以后多多支持之家！