Python之parser.add_argument解读

Python中的argparse模块是用于解析命令行参数和选项的标准模块。add_argument方法是argparse.ArgumentParser类中的一个方法，用于向该类中添加命令行选项和参数。

1. add_argument方法的基本用法

add_argument方法的基本用法如下：

parser.add_argument('name', action, nargs, const, default, type, choices, required, help, metavar)

参数说明：

• name：参数名。
• action：参数的行为。
• nargs：参数接受的值的个数。
• const：常数值，一般与nargs结合使用。
• default：参数的默认值。
• type：参数的数据类型。
• choices：参数值的限制。
• required：是否必填。
• help：参数的帮助信息。
• metavar：参数的显示名称。

2. add_argument方法的实际应用

下面是两个示例，演示add_argument方法的具体应用：

2.1 示例一

import argparse

parser = argparse.ArgumentParser()
parser.add_argument('--model', type=str, default='vgg16', help='model type')
parser.add_argument('--dataset', type=str, default='cifar10', help='dataset type')
parser.add_argument('--lr', type=float, default=0.01, help='initial learning rate')

args = parser.parse_args()
print(args.model)
print(args.dataset)
print(args.lr)

解析：

上述代码中，add_argument方法的第一个参数都以--为前缀，代表的是选项参数。如果不加该前缀，则认为是位置参数。

这段代码中定义了三个选项参数：

• model：模型类型，默认为vgg16。
• dataset：数据集类型，默认为cifar10。
• lr：学习率，类型为float，默认为0.01。

解析命令行参数时，使用parser.parse_args()方法，返回的是一个对象，通过这个对象就可以获得对应参数的值了。打印输出的是对应参数的值。

2.2 示例二

import argparse

parser = argparse.ArgumentParser(description='Process some integers.')
parser.add_argument('integers', metavar='N', type=int, nargs='+',help='an integer for the accumulator')

args = parser.parse_args()
print(args.integers)
print(sum(args.integers))

解析：

上述代码中，add_argument方法的第一个参数integers代表的是位置参数。参数类型为int，可以接受多个值，由于使用了nargs='+'的参数，所以表示这些值是一组的，用空格隔开，程序将读取所有的值并放入列表作为参数值。

运行上述代码，并在命令行输入python demo.py 1 2 3 4 5，则输出结果为：

[1, 2, 3, 4, 5]
15

3. 总结

本篇攻略详细讲解了argparse模块中的add_argument方法，包含了基本用法和实际应用的两个示例。通过学习此文可以更好地掌握add_argument方法的使用。