Python 命令解析 -argparse

应用场景

有时候我们需要替换Python脚本中不同的变量，比如进行模型选择。有时候需要提交到服务器，所以我们希望利用命令行就可以完成这一工作。比如在脚本后面添加模型参数，脚本中就可以自动改变模型。

sys.argv

sys包提供了一个简单的方法将参数返回到脚本内部，返回形式为一个列表。通常sys.argv[0]代表文件本身，后面添加的参数依次追加到sys.argv列表当中。

  1. import sys
  2. print(sys.argv)
  4. def model_choice():
  6.     if sys.argv[1] == 'FCN':
  7.         model = "FCN"
  8.     elif sys.argv[1] == 'Unet':
  9.         model = "Unet"
  10.     else sys.argv[1] == 'DeepLab':
  11.         model = "DeepLab"
  13.     return model
  15. if __name__ == "__main__":
  16.     print(model_choice)
  1. ~/Desktop/parser via 🐍 v3.7.3 via C base 
  2. ❯ python sysargv.py "Unet"
  3. ['sysargv.py', 'Unet']
  4. Unet

Argparse

argparse是一个Python模块：命令行选项、参数和子命令解析器。

主要有三个步骤：

  • 创建 ArgumentParser() 对象
  • 调用 add_argument() 方法添加参数
  • 使用 parse_args() 解析添加的参数
  1. import argparse
  3. def main():
  4.     parser = argparse.ArgumentParser(description="Demo of argparse")
  5.     parser.add_argument("ID",type = int,help = "输入ID") # 位置参数，按照顺序赋值，必须有这个参数
  6.     parser.add_argument('-n','--name', default=' Li ' ,type= str, help= "输入姓氏",choices = ['zhao','qian','sun','Li'])
  7.     parser.add_argument('-y','--year', default='20',type=int, help= "输入年龄")
  8.     args = parser.parse_args()
  9.     print(args)
  10.     name = args.name
  11.     year = args.year
  12.     print('Hello {}  {}'.format(name,year))
  14. if __name__ == '__main__':
  15.     main()
  1. ❯ python first.py -h
  2. usage: first.py [-h] [-n {zhao,qian,sun,Li}] [-y YEAR] ID
  4. Demo of argparse
  6. positional arguments:
  7.   ID                    输入ID
  9. optional arguments:
  10.   -h, --help            show this help message and exit
  11.   -n {zhao,qian,sun,Li}, --name {zhao,qian,sun,Li}
  12.                         输入姓氏
  13.   -y YEAR, --year YEAR  输入年龄
  15. ~/Desktop/parser via 🐍 v3.7.3 via C base 
  16. ❯ python first.py 13 -n "sun" -y "25"
  17. Namespace(ID=13, name='sun', year=25)
  18. Hello sun  25

更多细节操作，可以查询这篇博文!