Python argparse.ArgumentParser()用法

1. 概述

        argparse是python用于解析命令行参数和选项的标准模块，一方面可以接收运行代码时用户传入的各种参数（比如传入数据路径），另一方面也可以规范的定义程序中用到的参数（比如定义epoch及batch_size等）。

2. 使用步骤

①导入包：import argparse

②创建解析器：parser = argparse.ArgumentParser()

③添加参数：parser.add_argument('--epochs', type=int, default=100, help='')

④解析参数：args = parser.parse_args()

⑤使用参数：epoch = args.epochs

3.示例

import argparse  # 导入包

parser = argparse.ArgumentParser()   # 创建解析器
# 添加参数
parser.add_argument('--test', action='store_true', default=False, help='Test module.')
parser.add_argument('--batch_size', type=int, default=32, help='Total batch size.')
parser.add_argument('--epochs', type=int, default=100, help='Num of epochs to train.')

args = parser.parse_args()  # 解析参数

# 使用
print(args.epochs)
100
print(args.batch_size)
32

4. 命令行传参

python test.py --batch_size 64

print(args.test)
False
print(args.batch_size)
64
print(args.epochs)
100

python test.py --test --epochs 50

print(args.test)
True
print(args.batch_size)
32
print(args.epochs)
50

5. action='store_true'说明

action=‘store_true’ ，只要运行时该变量有传参，该变量就为True。