常用编程记录——parser = argparse.ArgumentParser()

在很多pytorch写的人工智能项目中，都能看到这样一句：parser = argparse.ArgumentParser()，本文就简单记录下它们的使用方法

部分内容参考：

https://blog.csdn.net/lizhiyuanbest/article/details/104975848

方式1—单个参数

创建a.py：

import argparse

parser = argparse.ArgumentParser()
parser.add_argument("square", help="display a square of a given number", type=int)
args = parser.parse_args()

s = args.square ** 2
print(s)

命令行执行：

$ python a.py 2
4

方式2—多个参数

创建b.py：

import argparse

parser = argparse.ArgumentParser()
parser.add_argument("square", help="display a square of a given number", type=int)
parser.add_argument("add1", help="display a sum of two given numbers", type=int)
parser.add_argument("add2", help="display a sum of two given numbers", type=int)
args = parser.parse_args()

s = args.square ** 2
a = args.add1 + args.add2
print(s)
print(a)

命令行执行结果：

$ python b.py 2 10 20
4
30

可见是将三个整数按顺序赋给三个参数

方式3—乱序赋值

创建c.py：

import argparse

parser = argparse.ArgumentParser()
parser.add_argument("--square", help="display a square of a given number", type=int)
parser.add_argument("--add1", help="display a sum of two given numbers", type=int)
parser.add_argument("--add2", help="display a sum of two given numbers", type=int)
args = parser.parse_args()

s = args.square ** 2
a = args.add1 + args.add2
print(s)
print(a)

⚠️注意参数前面有--

命令行运行方式1：

$ python c.py --square 2 --add1 10 --add2 20
4
30

命令行运行方式2（打乱顺序）：

$ python c.py --add1 2 --add2 10 --square 20
400
12

敲pytorch的同学掌握这些应该就足够使用了