python 中的argparse模块——自定义程序参数

最新推荐文章于 2024-04-15 11:05:12 发布

原创最新推荐文章于 2024-04-15 11:05:12 发布 · 667 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#深度学习 #python

本文介绍了Python的argparse模块，用于在程序运行时接收命令行参数。通过示例展示了如何创建解析器，添加不同类型的参数如int、str，并设置了默认值和帮助信息。最后演示了如何解析参数并使用它们，例如获取批处理大小(batch)和过期时间(expire_time)。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

官方文档

argparse模块可以在运行程序时传参到程序中，一般的步骤如下：

#导入模块
import argparse

#创建解析器
parser = argparse.ArgumentParser(description = 'parameter')
#通过add_argument()函数添加参数，默认type为str型
parser.add_argument('--batch', type = int, default = 100, help = 'the batch size')
#解析参数
args = parser.parse_args()

使用示例：

#导入模块
import argparse

def parameter():
    #创建解析器
    parser = argparse.ArgumentParser(description = 'parameter')
    #添加参数
    parser.add_argument('--batch', type = int, default = 100, help = 'the batch size')
    parser.add_argument('--expire_time', type = int,  default = 60, help = 'the expire_time')
    parser.add_argument('--table_name', type = str, help = 'the table_name')
    parser.add_argument('--date', type = int, default = date_default, help = 'the date')
    parser.add_argument('--name', type = str, default = 'driver_recall_condition', help = 'the name')
    parser.add_argument('--key', type = str, help = 'the key')
    parser.add_argument('--val', type = str, help = 'the val')

    args = parser.parse_args()
    return args
 
 if __name__=='__main__':
    
    arg = parameter()
    batch = arg.batch
    expire_time = arg.expire_time
    print(expire_time )