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 )