如何使用变量nargs使用argparse进行条件循环?所以,基本上,它应该有或没有争论。
我在努力:
parser = argparse.ArgumentParser(description="output parser")
group = parser.add_mutually_exclusive_group()
group.add_argument("--dos", help="get DOSCAR for plot",
nargs="?", metavar=("int"))
args = parser.parse_args()
if args.dos:
if len(args.dos) > 1:
chosen = int(args.dos[0])
chdos = "at_dos"+args.dos[0]+".dat"
else:
chosen = None
inpt = "DOSY"
print(chosen)
print(inpt)
现在,如果我有变量,那么它会打印一些值,错误的是一些值:
$python3 vasp.py --dos 111
111
None # IT SHOULDN'T BE NONE
DOSY
但没什么可以不争论的。
我也尝试过使用普通sys.argv,因为:
def get_dos():
if len(sys.argv) > 2:
chosen = int(sys.argv[2])
chdos = "at_dos"+sys.argv[2]+".dat"
else:
chosen = None
inpt = "DOSCAR"
print(sys.argv)
print(args.dos)
print(chosen)
print(inpt)
在这种情况下,当选项存在时,它给出正确的结果:
python3 vasp.py --dos 12
['vasp.py', '--dos', '12']
12
12
DOSCAR
但同样,没有什么是没有选择的:
$python3 vasp.py --dos
我已经尝试过了HaululJ的建议。它给出:
$python3 tt.py --dos 12
Namespace(dos='12')
1
DOSY
没有争论,它仍然没有打印任何东西。
最佳答案
在ipython会话中简化解析器:
In [1004]: parser=argparse.ArgumentParser()
In [1005]: parser.add_argument('--dos', nargs='?')
In [1007]: parser.parse_args('--dos 111'.split())
Out[1007]: Namespace(dos='111')
在这种情况下,
args.dos
将是字符串'111',len
3,int(args.dos[0])
是数字1。同样的情况发生在您将nargs
留空(默认为无)时。使用
nargs='?'
我也可以使用不带参数的标志,在这种情况下,该值是默认的None
。In [1013]: parser.parse_args('--dos'.split())
Out[1013]: Namespace(dos=None)
nargs=
?is most useful with a
const, which gives a convenient 3 way action. I can add
type=int`将字符串(如果有)转换为整数。In [1015]: parser.add_argument('--dos', nargs='?', type=int,
default=None, const=123)
In [1016]: parser.parse_args([]) # not used
Out[1016]: Namespace(dos=None)
In [1017]: parser.parse_args('--dos'.split()) # without argument
Out[1017]: Namespace(dos=123)
In [1018]: parser.parse_args('--dos 456'.split()) # with argument
Out[1018]: Namespace(dos=456)
其他的,比如1,“*”和“+”给你一个列表,你可以检查它的长度等。
====================
在您的
nargs
测试中if len(sys.argv) > 2:
chosen = int(sys.argv[2])
chdos = "at_dos"+sys.argv[2]+".dat"
argv
是一个列表,因此如果有足够的元素来应用sys.argv
步骤,len将计数。这不起作用,因为
sys.argv[2]
是单个字符串,而不是列表。if len(args.dos) > 1:
chosen = int(args.dos[0])
chdos = "at_dos"+args.dos[0]+".dat"
args.dos
是字符串中的字符数,len(args.dos)
是第一个字符。============
如果我定义:
def get_dos(argv=None):
parser=argparse.ArgumentParser()
parser.add_argument('--dos', type=int, nargs='?')
args = parser.parse_args(argv)
chosen = args.dos
if chosen is not None:
chdos = 'at_dos%s.dat'%chosen
else:
chdos = ''
return chosen, chdos
这些测试产生的值我认为符合您的需要:
In [1042]: get_dos([])
Out[1042]: (None, '')
In [1043]: get_dos(['--dos'])
Out[1043]: (None, '')
In [1044]: get_dos(['--dos','123'])
Out[1044]: (123, 'at_dos123.dat')