Python argparse库的基本使用步骤

2024-04-02 19:55

短信预约 -IT技能 免费直播动态提醒

argparse库是python下的一个命令行参数管理库，支持int、str、float、bool、数组等5种基本数据类型。在解析命令行参数时还支持默认值。此外，还支持修改命令行参数，在代码中增加命令行参数等操作。通过argparse库，可以实现参数的简易管理，控制代码执行中的一些流程或参数值。

基本用法

1、int支持

argparse的写法，class_nums的默认值为7

parser.add_argument("--class_nums", type=int, default=7)

命令行传参写法（假设程序文件名称为test.py）, 传入了新的值，覆盖了默认值

test.py --class_nums 8

2、str支持

argparse的写法，load_from没有设置默认值，如果要设置的话请设置default

parser.add_argument("--load_from", type=str, help='load trained model to generate predictions of validation set')

命令行传参写法（假设程序文件名称为test.py）, 传入了新的值"best.pth"赋值给load_from

test.py --load_from "best.pth"

3、float支持

argparse的写法，lr的默认值为=0.0003

parser.add_argument("--lr", type=float, default=0.0003)

命令行传参写法（假设程序文件名称为test.py）, 传入了新的值，覆盖了默认值

test.py --lr 0.001

4、bool支持

argparse的写法，pretrained的默认值为None。store_true表示传入参数则为True

parser.add_argument("--pretrained", action="store_true")

命令行传参写法（假设程序文件名称为test.py）, 此时pretrained的值为True

test.py --pretrained

5、数组支持

argparse的写法，input_shape的默认值为[256, 256]。nargs表示传入参数可能存在多个值，也就是说传入的参数不一定是2个，3个也可以

parser.add_argument('--input_shape', nargs='+', type=int, default=[256, 256])

命令行传参写法（假设程序文件名称为test.py）, 此时input_shape的值为[512, 512]

test.py --input_shape 512 512

完整调用

以下代码实现了取参数值，修改参数值，新增参数值的完整调用。在非控制台环境下执行下面代码是无法修改默认参数的，只能是在控制台下执行才能修改参数

import argparse
class Options:
    def __init__(self):
        parser = argparse.ArgumentParser('命名空间')
        parser.add_argument("--class_nums", type=int, default=7)
        parser.add_argument("--pretrained", action="store_true")
        parser.add_argument("--lr", type=float, default=0.0003)
        parser.add_argument("--load_from", type=str,
                            help='load trained model to generate predictions of validation set')
        parser.add_argument('--input_shape', nargs='+', type=int, default=[256, 256])
        self.parser = parser

    def parse(self):
        args = self.parser.parse_args()
        return args
if __name__ == "__main__":
    args = Options().parse()
    print("\n原始参数：")
    print(args)
    
    print("\n取参数值：")
    print("args.class_nums:",args.class_nums)
    
    print("\n修改参数：")
    args.input_shape=[512,512]
    print(args)
    
    print("\n新增参数：")
    args.new_key=[512,512]
    print(args)

博主测试代码：python .\test.py --input_shape 128 128
执行输出如下所示：

原始参数：
Namespace(class_nums=7, input_shape=[128, 128], load_from=None, lr=0.0003, pretrained=False)
取参数值：
args.class_nums: 7
修改参数：
Namespace(class_nums=7, input_shape=[512, 512], load_from=None, lr=0.0003, pretrained=False)
新增参数：
Namespace(class_nums=7, input_shape=[512, 512], load_from=None, lr=0.0003, new_key=[512, 512], pretrained=False)

到此这篇关于Python argparse库的基本使用的文章就介绍到这了,更多相关Python argparse库使用内容请搜索编程网以前的文章或继续浏览下面的相关文章希望大家以后多多支持编程网！

免责声明：

① 本站未注明“稿件来源”的信息均来自网络整理。其文字、图片和音视频稿件的所属权归原作者所有。本站收集整理出于非商业性的教育和科研之目的，并不意味着本站赞同其观点或证实其内容的真实性。仅作为临时的测试数据，供内部测试之用。本站并未授权任何人以任何方式主动获取本站任何信息。

② 本站未注明“稿件来源”的临时测试数据将在测试完成后最终做删除处理。有问题或投稿请发送至: 邮箱/279061341@qq.com QQ/279061341

阅读原文内容投诉