argparse.ArgumentParser()用法举例

1. 应用场景

我们在玩深度学习，训练模型的时候，会涉及到很多的参数，这个时候就需要用到argparse.ArgumentParser()方法，它的优点是方便在命令行调用的时候修改参数，为了快速了解该方法的应用，这里举例说明

2.应用举例

假设我们新建一个python文件叫par.py，它的作用是输入一个圆的半径，帮助我们计算圆的面积，并且要求可以在命令行中调用这个python文件 ，所以我们新建文件，编写以下代码：

# -*- coding: utf-8 -*-
# @Time    : 2024/5/24 10:05
# @Author  : WY


import argparse

parser = argparse.ArgumentParser(description='Enter the radius and calculate the area')
parser.add_argument("--r", help="input the radius", type=float)
args = parser.parse_args()

print(args.r ** 2 * 3.14)

3. 程序调用

在终端模式下输入：python par.py -h，可以查看该python文件的调用方法。

当我们输入：python par.py --r 1.0，就可以把1.0 参数传递到python文件的变量r中，运行结果如下：