python - 如何在Python中将整个列表作为命令行参数传递？

我试图将两个包含整数的列表作为参数传递给python代码。但是sys.argv[i]将参数作为字符串列表获取。

输入看起来像

$ python filename.py [2,3,4,5] [1,2,3,4]

我发现以下黑客可以转换列表。

strA = sys.argv[1].replace('[', ' ').replace(']', ' ').replace(',', ' ').split()
strB = sys.argv[2].replace('[', ' ').replace(']', ' ').replace(',', ' ').split()
A = [float(i) for i in strA]
B = [float (i) for i in strB]

有一个更好的方法吗？

最佳答案

命令行参数始终以字符串形式传递。您将需要自己将它们解析为所需的数据类型。

>>> input = "[2,3,4,5]"
>>> map(float, input.strip('[]').split(','))
[2.0, 3.0, 4.0, 5.0]
>>> A = map(float, input.strip('[]').split(','))
>>> print(A, type(A))
([2.0, 3.0, 4.0, 5.0], <type 'list'>)

有诸如argparse和click之类的库可让您定义自己的参数类型转换，但是argparse将"[2,3,4]"与[一样对待2 , 3 , 4 ]，因此我怀疑它是否有用。
编辑Jan 2019这个答案似乎仍然有一些行动，所以我将直接从argparse文档中添加另一个选项。
您可以使用action=append将重复的参数收集到一个列表中。

>>> parser = argparse.ArgumentParser()
>>> parser.add_argument('--foo', action='append')
>>> parser.parse_args('--foo 1 --foo 2'.split())
Namespace(foo=['1', '2'])

在这种情况下，您将为每个列表项传递一次--foo ?。使用OP示例:python filename.py --foo 2 --foo 3 --foo 4 --foo 5将导致foo=[2,3,4,5]