pytorch 命令行运行_PyTorch 40.argsparse: python和命令行之间的交互

以你之姓@ 2022-11-05 15:55 117阅读 0赞

argparse是python标准库里面用来处理命令行参数的库

import argparse 首先导入模块

parser = argparse.ArgumentParser() 创建一个解析对象

parser.add\_argument() 向该对象中添加你要关注的命令行参数和选项

return parser.parse\_args() 进行解析

假设我们创建一个“argp.py”的文件：

import argparse \# 引入模块

\# 建立解析对象

parser = argparse.ArgumentParser()

parser.add\_argument("echo") \# xx.add\_argument("aa")

\# 给xx实例增加一个aa属性

\# 把parser中设置的所有"add\_argument"给返回到args子类实例当中

\# 那么parser中增加的属性内容都会在args实例中，使用即可。

args = parser.parse\_args()

parser.parse\_args()

print(arg.echo) \# 打印定位参数echo

运行的话，在python argp.py后面加上abcd(任意)；

其将输出“abcd”。因为我们把abcd赋给了args.echo了。

给其设置help：

parser.add\_argument("echo", help = "echo is here~")

在命令行中输入：

python argp.py -h

就可以得到相关的帮助信息。

type：

假设我们写一个文件，在命令行将其编译时，同时输入一个数字，返回一个平方：

因为parse.add\_argument()对于接受的值默认其为str，如果要将之视为int类型，额外加一句“type=int”。

import argparse

parser = argparse.ArgumentParser()

parser.add\_argument("square", help = "To sqaure the number given", type = int)

\# 在命令行中，该py文件希望用户能给他一个参数，最终将之转化为：args.square

args = parser.parse\_args()

print(args.square\*\*2)

可选参数：

在add\_argument前，给属性名之前加上'--'，就能将之变为可选参数：

import argparse

parser = argparse.ArgumentParser()

parser.add\_argument("--bool",help = "Whether to pirnt sth.")

\# 这里的bool是一个可选参数，返回给args的是 args.bool

args = parser.parse\_args()

if args.bool:

print('bool = 1')

输入

python argp.py --bool 1

得到

bool = 1

既然是可选的，如果不指定(就是不使用它)的话，对应的变量会被设置为None。

对于简单程序，我们可能只需要两种值TrueorFalse：

我们可以在原有的程序的基础上，在add\_argument中多加一个参数：

action = "store\_true"

这个时候，只要在命令行中加个bool，args.bool就是True了，无需再加上之后的赋值内容，加了反而会报错。

但是，如果在parser.add\_argument(..., required=True,type=int)的话，那么这个参数就是必须的，因为他是required。

参数简写：

parser.add\_argument("-b","--bool",action="store\_true")

这样，在程序内部我们依旧使用args.bool，但是在命令行当中我们只需要多写一个“-b”就好了。

混合定位参数和可选参数：

当我们使用多个parser.add\_argument(...)的时候，若将之视为可选参数，无则为None。

import argparse

parser = argparse.ArgumentParser()

parser.add\_argument("-b","--bool",help="to show the ans in a sentencen form",action = "store\_true")

parser.add\_argument("square",type=int)

args = parser.parse\_args()

ans = args.square\*\*2

if args.bool:

print("the square of \{\} = \{\}".format(args.square,ans))

else:

print(ans)

\# 得到的效果为：一旦输入“-b”然后再输入相关的参数，就可以得到我们需要的结果了，不输入“-b”我们还是可以得到结果，只不过是另外一种方式而已。

进一步，我们可以进行以下操作，或者对之进行一些了解：限制输入的值：

如parser.add\_argument("-b","--bool",choices=\['0','1'\],help="you can only input 0 or 1.")

其中，choice可以限定我们能够输入的数。

参数值和顺序：

一般我们要把可选参数最后add，不可选参数放前面add。

而在命令行里面输入代码时，程序会先赋值“先add的参数选项”。

比如我先parser.add\_argument(“a”,…)再b;

输入命令行的时候：python xx.py x y，这个时候，args.a=x ; args.b=y。

count操作计数赋值：

parser.add\_argument("-b","--bool",action="count",type=int)

这个时候，在命令后加上 -b ——> bool=1 ; -bb ——> bool=2 ;以此类推。

但是这样的话，如果在程序里拿bool作为比较对象的时候，比如if bool > 0:,不写“-b”的时候会报错，原因是此时的bool是None,无法比较，要解决这个问题我们只需要加上一个 default 选项：parser.add\_argument("-b","--bool",action="count",type=int,default=0)就好了。