我试图将一个列表作为参数传递给命令行程序。是否有argparse选项传递一个列表作为选项?
parser.add_argument('-l', '--list',
type=list, action='store',
dest='list',
help='<Required> Set flag',
required=True)
脚本如下所示
python test.py -l "265340 268738 270774 270817"
当前回答
如果你有一个嵌套列表，其中内部列表有不同的类型和长度，你想要保留类型，例如:
[[1,2]， ["foo"， "bar"]， [3.14， "baz"， 20]]]
然后你可以使用@sam-mason提出的解决方案来解决这个问题，如下所示:
from argparse import ArgumentParser
import json
parser = ArgumentParser()
parser.add_argument('-l', type=json.loads)
parser.parse_args(['-l', '[[1,2],["foo","bar"],[3.14,"baz",20]]'])
这使:
Namespace(l=[[1, 2], ['foo', 'bar'], [3.14, 'baz', 20]])
其他回答
如果你想让一个开关有多个参数，那么你可以使用nargs='+'。如果你的例子'-l'实际上是接受整数:
a = argparse.ArgumentParser()
a.add_argument(
'-l', '--list', # either of this switches
nargs='+', # one or more parameters to this switch
type=int, # /parameters/ are ints
dest='lst', # store in 'lst'.
default=[], # since we're not specifying required.
)
print a.parse_args("-l 123 234 345 456".split(' '))
print a.parse_args("-l 123 -l=234 -l345 --list 456".split(' '))
生产
Namespace(lst=[123, 234, 345, 456])
Namespace(lst=[456]) # Attention!
如果多次指定相同的参数，默认操作('store')将替换现有数据。
另一种方法是使用追加操作:
a = argparse.ArgumentParser()
a.add_argument(
'-l', '--list', # either of this switches
type=int, # /parameters/ are ints
dest='lst', # store in 'lst'.
default=[], # since we're not specifying required.
action='append', # add to the list instead of replacing it
)
print a.parse_args("-l 123 -l=234 -l345 --list 456".split(' '))
生产
Namespace(lst=[123, 234, 345, 456])
或者您可以编写一个自定义处理程序/操作来解析逗号分隔的值，这样您就可以这样做
-l 123,234,345 -l 456
如果你有一个嵌套列表，其中内部列表有不同的类型和长度，你想要保留类型，例如:
[[1,2]， ["foo"， "bar"]， [3.14， "baz"， 20]]]
然后你可以使用@sam-mason提出的解决方案来解决这个问题，如下所示:
from argparse import ArgumentParser
import json
parser = ArgumentParser()
parser.add_argument('-l', type=json.loads)
parser.parse_args(['-l', '[[1,2],["foo","bar"],[3.14,"baz",20]]'])
这使:
Namespace(l=[[1, 2], ['foo', 'bar'], [3.14, 'baz', 20]])
我认为最优雅的解决方案是传递一个lambda函数给“类型”，正如Chepner所提到的。除此之外，如果你事先不知道列表的分隔符是什么，你也可以传递多个分隔符给re.split:
# python3 test.py -l "abc xyz, 123"
import re
import argparse
parser = argparse.ArgumentParser(description='Process a list.')
parser.add_argument('-l', '--list',
type=lambda s: re.split(' |, ', s),
required=True,
help='comma or space delimited list of characters')
args = parser.parse_args()
print(args.list)
# Output: ['abc', 'xyz', '123']
将chepner的评论应用到Lunguini的回答中:
import argparse, json
parser = argparse.ArgumentParser()
parser.add_argument('-l', '--list', type=lambda a: json.loads(a), default="[]",
help="String formatted as list wrapped in []")
args = parser.parse_args()
print(args.list)
用法:
$ python parse_list.py -l "[265340, 268738, 270774, 270817]"
[265340, 268738, 270774, 270817]
我想处理传递多个列表，整数值和字符串。
如何将Bash变量传递给Python?
def main(args):
my_args = []
for arg in args:
if arg.startswith("[") and arg.endswith("]"):
arg = arg.replace("[", "").replace("]", "")
my_args.append(arg.split(","))
else:
my_args.append(arg)
print(my_args)
if __name__ == "__main__":
import sys
main(sys.argv[1:])
秩序并不重要。如果你想传递一个列表，只需在“[”和“]之间做，并用逗号分隔它们。
然后,
python test.py my_string 3 "[1,2]" "[3,4,5]"
Output => ['my_string'， '3'， ['1'， '2']， ['3'， '4'， '5']]]， my_args变量按顺序包含参数。
