我想将一个名字列表传递给我从控制台用Python编写的程序。例如,我想使用类似于此的方法(我知道它不应该因为bash
而起作用):
$ python myprog.py -n name1 name2
所以,我尝试了这段代码:
# myprog.py
from argparse import ArgumentParser
parser = ArgumentParser()
parser.add_argument('-n', '--names-list', default=[])
args = parser.parse_args()
print(args.names_list) # I need ['name1', 'name2'] here
导致错误:
usage: myprog.py [-h] [-n NAMES_LIST]
myprog.py: error: unrecognized arguments: name2
我知道我可以使用引号"name1 name2"
传递名称并将其拆分为我的代码args.names_list.split()
。但我很好奇,有没有更好的方法通过argparse
模块传递字符串列表。
任何想法都会受到赞赏。
谢谢!
答案 0 :(得分:6)
您需要定义--names-list
以获取任意数量的参数。
parser.add_argument('-n', '--names-list', nargs='+', default=[])
请注意,具有任意数量参数的选项通常不适用于位置参数:
# Is this 4 arguments to -n, or
# 3 arguments and a single positional argument, or ...
myprog.py -n a b c d
答案 1 :(得分:2)
您需要使用nargs
:
parser.add_argument('-n', '--names-list', nargs="*")
答案 2 :(得分:1)
parser.add_argument('-n', '--names-list', default=[], nargs='+')