快捷導(dǎo)航

Python命令行參數(shù)定義及需要注意的地方

更新時間：2020年11月30日 16:44:57 作者：頎周

這篇文章主要介紹了Python命令行參數(shù)定義及需要注意的地方，幫助大家更好的理解和學(xué)習(xí)python，感興趣的朋友可以了解下

　　在命令行中運行python代碼是很常見的，下面介紹如何定義命令后面跟的參數(shù)。

1 常規(guī)用法

　　Python代碼中主要使用下面幾行代碼來定義并獲取需要在命令行中賦值的參數(shù)：

import argparse 
 
parser = argparse.ArgumentParser("Description.") # ————1————
parser.add_argument('--test1', type=str, default="1", help="Test1 help.") # ————2————
parser.add_argument('--test2', type=float, default=2.1, help="Test2 help.") # ————2————
args = parser.parse_args() # ————3————

print(args)
test1, test2 = args.test1, args.test2 # ————4———— 
print(type(test1), test1, test1 + " add test") 
print(type(test2), test2, test2 + 100)

　　注釋：

　　1、獲取外部參數(shù)對象實例?？梢詡魅胱址畞砻枋隹傮w的外部參數(shù)。

　　2、定義外部參數(shù)，可以多次調(diào)用來定義多個外部參數(shù)。函數(shù)的第一個參數(shù)是外部參數(shù)的名稱，建議以兩個減號開頭，中間不要使用數(shù)字字母下劃線以外的符號，否則容易出錯。type是你希望外部參數(shù)傳入后轉(zhuǎn)換的類型，比如--test1定義為str，即使命令行傳參不添加引號，函數(shù)也將它轉(zhuǎn)換為字符串（當(dāng)然不加引號傳字符串就不能用空格了）。而--test2是float，即使命令行傳參帶了引號，只要能轉(zhuǎn)換為數(shù)字，函數(shù)都能轉(zhuǎn)換為float類型。default是外部參數(shù)的默認(rèn)值。如果命令行不對外部參數(shù)賦值而使用默認(rèn)值，函數(shù)不會對你定義的默認(rèn)值進行type上的轉(zhuǎn)換，所以default最好直接定義為type指定的類型（比如--test1的default定義為"1"而不是1）。help傳入對當(dāng)前外部參數(shù)的描述，命令行中使用--help輸出的就是這個。

　　3、分析獲取的外部參數(shù)。傳出的是一個命名空間，類似于字典，但使用屬性的方式而不是key索引來獲取命名對應(yīng)的外部參數(shù)值。

　　4、以獲取屬性的方式獲取外部參數(shù)。

　　需要注意的是，外部參數(shù)的定義必須寫在獲取外部參數(shù)之前。也就是說，parser.add_argument()要寫在parser.parse_args()之前，否則外部參數(shù)無法被命令行捕獲。另外，如果代碼中沒有parser.parse_args()，命令行就不能傳參。

　　用命令行執(zhí)行以上代碼，命令與輸出如下（以上代碼保存為test.py）：

　　可以看出，故意以數(shù)字和字符串的方式分別對字符串和float類型的外部參數(shù)賦值，函數(shù)都能正確地轉(zhuǎn)換并正確地參與運算。

　　使用默認(rèn)值時輸出如下：

2 注意事項

　　通過以上的實驗，argparse庫似乎能將外部參數(shù)正確轉(zhuǎn)換為各種類型，只要能轉(zhuǎn)換。下面來實驗是否能傳入一個numpy數(shù)組。代碼如下：

import numpy as np
import argparse 
 
parser = argparse.ArgumentParser() 
parser.add_argument('--array', type=np.array, default=np.array([1,2])) 
args = parser.parse_args() 

print(args) 
array = args.array
print(type(array))
print(array)
print(array + 1)

　　當(dāng)使用默認(rèn)值時：