python argparse传入布尔类型参数的几种方式

最新推荐文章于 2024-05-30 08:25:08 发布

行业边缘的摸鱼怪

最新推荐文章于 2024-05-30 08:25:08 发布

阅读量1.4k

点赞数 3

分类专栏： python 文章标签： python

本文链接：https://blog.csdn.net/dream6985/article/details/129220793

版权

python 专栏收录该内容

43 篇文章 0 订阅

订阅专栏

当使用argparse时，布尔型参数默认被视为字符串。可以使用action=store_true让未传递参数默认为False，或者自定义type转换函数，如lambdax:x.lower()==true，来确保正确解析布尔值。此外，定义一个str2bool函数能更灵活地处理所有布尔类型的变量，避免因用户输入差异导致的错误。

摘要由CSDN通过智能技术生成

问题描述

argparse无法识别布尔型参数，参数传进来时，默认为字符串。字符串强制转布尔型时为True。因此，在我们想要传入布尔类型的参数时，需要用一定的方法对参数进行一些处理。处理方法如下所示。

解决方案

1 action=‘store_true’,传入参数为True，不传为False

例子如参考链接中所示。

parser.add_argument("--isOverwrite1", action='store_true',help="Overwrite",required=False)

2 自定义函数type=lambda x: x.lower() == ‘true’，做判断

例子如参考链接中所示。

parser.add_argument("--isOverwrite2",help="Overwrite", type=lambda x: x.lower() == 'true', required=False)

3 专门写一个函数处理布尔值

举例如下：

import argparse
parser.add_argument('--replay', type=str)
 # 定义一个转换函数
def str2bool(v):
    if v.lower() in ('yes', 'true', 't', 'y', '1'):
        return True
    elif v.lower() in ('no', 'false', 'f', 'n', '0'):
        return False
    else:
        raise argparse.ArgumentTypeError('Boolean value expected.')

replay = str2bool(args.replay)
args.replay = str2bool(args.replay)

这种方法的好处是，可以很方便的处理所有布尔类型的变量，不用担心用户因对布尔类型变量的表示不同造成的参数误传错误。