学习使用python包 argparse

简介

argparse --> Argument Parser
argparse 模块可以让人轻松编写用户友好的命令行接口。

应用

import argparse
def arg_parse():
    argparser = argparse.ArgumentParser()

    argparser.add_argument("--step",type=int ,default=1000)
    argparser.add_argument("--device",default= 0,help="device of deeplearning")

    opt = argparser.parse_args()
    print(type(opt))
    return opt

# print(opt)
def main(opt):
    print(opt)
if __name__ =="__main__":
    opt = arg_parse()
    main(opt)

结果

<class 'argparse.Namespace'>
Namespace(device=0, step=1000)

进程已结束,退出代码0