Python之parser.add_argument

井礼骞
2023-12-01
  1. Python解析参数
import argparse

parser = argparse.ArgumentParser(description = 'TEST')
parser.add_argument ('--start-epoch', default = 0, type = int, help = 'please input number')

def main():
    global args
    args = parser.parse_args()

    print(args)
    print(args.start_epoch)

if __name__ == '__main__':
    main()

  • 注:在参数解析时,参数中并不区分字符‘-’和‘_’;如上所述,在添加是使用的是’–start-epoch’,但是在解析时使用的是args.start_epoch。
  1. parser.add_argument参数action

在使用parser进行传参时,遇到参数action,一直不得其解,终有所获,解释如下:

parser.add_argument(‘--is_train’, action=’store_true’, default=False)

其中“–is_train”参数在调用时,如下:

python demo1.py 						#无参数传入时,使用默认值False 
python demo1.py --is_train 		#当使用此参数时,表示该参数是True, 注意在这里并没有对其赋值,
													#其作用是作为开关使用,来控制action动作
 类似资料: