[英]How to pass an entire list as command line argument in Python when the list comes from a json file
[英]How to pass an entire list as command line argument in 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]
有一个更好的方法吗?
不要重新发明轮子。 使用argparse 模块,明确并传入实际的参数列表
import argparse
# defined command line options
# this also generates --help and error handling
CLI=argparse.ArgumentParser()
CLI.add_argument(
"--lista", # name on the CLI - drop the `--` for positional/required parameters
nargs="*", # 0 or more values expected => creates a list
type=int,
default=[1, 2, 3], # default if nothing is provided
)
CLI.add_argument(
"--listb",
nargs="*",
type=float, # any type/callable can be used here
default=[],
)
# parse the command line
args = CLI.parse_args()
# access CLI options
print("lista: %r" % args.lista)
print("listb: %r" % args.listb)
然后你可以调用它使用
$ python my_app.py --listb 5 6 7 8 --lista 1 2 3 4
lista: [1, 2, 3, 4]
listb: [5.0, 6.0, 7.0, 8.0]
命令行参数始终作为字符串传递。 您需要自己将它们解析为所需的数据类型。
>>> 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和点击,让你定义自己的参数类型转换,但argparse
对待"[2,3,4]"
一样[
2
,
3
,
4
]
所以我怀疑这将是有益的。
编辑 2019 年 1 月这个答案似乎仍然有一些作用,所以我将添加另一个直接从 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]
我对此进行了测试,我的输入如下所示:
python foo.py "[1,2,3,4]" "[5,6,7,8,9]"
我正在执行以下操作来转换感兴趣的两个参数:
import ast
import sys
list1 = ast.literal_eval(sys.argv[1])
list2 = ast.literal_eval(sys.argv[2])
为什么不:
python foo.py 1,2,3,4 5,6,7,8
比尝试 eval python 更干净,并且不需要您的用户知道 python 格式。
import sys
list1 = sys.argv[1].split(',')
list2 = [int(c) for c in sys.argv[2].split(',')] # if you want ints
您还可以执行以下操作:
说,你有foo.py
:
import json
import sys
data = json.loads(sys.argv[1])
print data, type(data)
然后,如果您将上述内容运行为: python foo.py "[1,2,3]"
输出:
[1, 2, 3] <type 'list'>
不,无法在命令行参数中传递列表。 命令行参数始终是字符串。 但是有一种更好的方法可以将其转换为列表。 你可以这样做:
import ast
A = ast.literal_eval(strA)
B = ast.literal_eval(strB)
您可以简单地使用 argparse 的nargs='+'
选项
import argparse
parser = argparse.ArgumentParser()
parser.add_argument('--myarg', dest='myarg', required=False, help='--myarg 100 100 100 takes a list of 3 elements, each is of value 100 ', nargs='+', type=int, default=[100,100,100])
然后你可以像这样传递参数:
python your_file.py --myarg 1 2 3
这将作为 [1,2,3] 存储在 myarg 中的程序中
print(myarg)
输出:
[1,2,3]
声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.