介紹

argparse 是 Python 內建的一個用於命令項選項與引數解析的模組，通過在程式中定義好需要的引數，argparse 將會從 sys.argv 中解析出這些引數，並自動生成幫助和使用資訊。

使用步驟

import argparse                     # 匯入模板
parser = argparse.ArgumentParser()  # 建立parser
parser.add_argument()               # 新增引數
args = parser.parse_args()          # 引數解析

ArgumentParser

建立一個解析物件

• description：命令列幫助的開始文字

add_argument

新增引數

• type：指定傳進來的引數的型別
• default：如果命令列沒有出現這個選項，那麼使用default指定的預設值
• help：設定這個選項的幫助資訊
• action：在命令列中遇到這個引數時要採取的基本行動型別

示例

import argparse

def parse_args():
    parser = argparse.ArgumentParser(description='This is a description')
    parser.add_argument('--ver', '-v', action='store_true', help='haha')
    parser.add_argument('--num', '-n', default=128, type=int)
    return parser.parse_args()

def main(args):
    print(args.ver)
    print(args.num)

if __name__ == '__main__':
    main(parse_args())
 

在命令列輸入：

    python train.py -v -n 12

列印結果：

    True
    12

參考部落格：