如何解决Python argparse从选项中选择一个列表
如何使用argparse从一组选项中提供参数列表。例如,假设我想要类似的东西:
python sample.py --p1 ['a','b'] --p2 ['x','y']
p1
只能是'a','b','c'
列表中的任何一个或全部,而p2
只能是'x','y','z'
列表中的任何一个或全部
解决方法
我找到了一种获取所需行为的方法,但是语法与您所呈现的不同。您必须使用唯一的参数名称/值对指定每个选择。如果可以,则可以使用以下方法:
parser = argparse.ArgumentParser(prog='game.py')
parser.add_argument('--p1',choices=['a','b','c'],action='append')
args = parser.parse_args(['--p1','a','--p1','b'])
print(args)
结果:
Namespace(p1=['a','b'])
但这会失败:
parser = argparse.ArgumentParser(prog='game.py')
parser.add_argument('--p1','x'])
print(args)
结果:
usage: game.py [-h] [--p1 {a,b,c}]
game.py: error: argument --p1: invalid choice: 'x' (choose from 'a','c')
我在文档中找不到任何建议ArgumentParser将有效值列表作为参数值的参数,而这正是您的版本所需要的。
,将nargs
与choices
一起使用:
In [1]: import argparse
In [2]: p = argparse.ArgumentParser()
In [3]: p.add_argument('-a',nargs='+',choices=['x','y','z'])
Out[3]: _StoreAction(option_strings=['-a'],dest='a',const=None,default=None,type=None,'z'],help=None,metavar=None)
In [4]: p.parse_args('-a x y z x x'.split())
Out[4]: Namespace(a=['x','z','x','x'])
In [5]: p.parse_args('-a x y z w x'.split())
usage: ipython3 [-h] [-a {x,y,z} [{x,z} ...]]
ipython3: error: argument -a: invalid choice: 'w' (choose from 'x','z')
版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。