python argparse中使用True或者False

最新推荐文章于 2024-04-18 10:15:01 发布

Tchunren

最新推荐文章于 2024-04-18 10:15:01 发布

阅读量1.3k

点赞数 3

文章标签： str2bool argparse 布尔参数命令行解析自定义转换

本文链接：https://blog.csdn.net/t20134297/article/details/112007557

版权

def str2bool(v):
    if isinstance(v,bool):
        return v
    if v == 'True':
        return True
    if v == 'False':
        return False

import argparse
parser = argparse.ArgumentParser()

parser.add_argument('--drop', type=str2bool, default=False)
cfg = parser.parse_args()

if cfg.drop:
    print('niubi')
else:
    print('walalala')

用这个人方式就能替代 --store_true了