将整数列表传递给python

Question

我想传递2个整数列表作为python程序的输入。

对于e.g，（来自命令行）

python test.py --a 1 2 3 4 5 -b 1 2 

在此列表中的整数的范围可以从1-50，列表2是List1中的子集。
任何帮助/建议？是​​正确的模块？任何使用该问题的顾虑？

我曾尝试：

import argparse 
if __name__ == '__main__': 
    parser = argparse.ArgumentParser() 
    parser.add_argument('--a', help='Enter list 1 ') 
    parser.add_argument('--b', help='Enter list 2 ') 
    args = parser.parse_args() 
    print (args.a) 

Answer 1

您可以将它们作为字符串比转换成列表。您可以使用argparse或optparse。

import argparse 

parser = argparse.ArgumentParser() 
parser.add_argument('--l1', type=str) 
parser.add_argument('--l2', type=str) 
args = parser.parse_args() 
l1_list = args.l1.split(',') # ['1','2','3','4'] 

实施例：python prog.py --l1=1,2,3,4

另外，作为线可以传递是这样的1-50，然后分裂的“ - ”和构建范围。 事情是这样的：

import argparse 

parser = argparse.ArgumentParser() 
parser.add_argument('--l1', type=str, help="two numbers separated by a hyphen") 
parser.add_argument('--l2', type=str) 
args = parser.parse_args() 
l1_list_range = xrange(*args.l1.split('-')) # xrange(1,50) 
for i in l1_list_range: 
    print i 

例子：python prog.py --l1=1-50

逻辑我觉得你可以自己编写。 :)

Answer 2

是optparse和argparse工作是他们从读取命令行参数，参数通过空格分开，方式，所以如果你想要输入您通过从optparse或argparse命令行interfact的整数列表 -

> my_script.py --a "1 2 3 4 5" --b "1 2" 

或：

> my_script.py --a 1,2,3,4,5 --b 1,2 

你SC您可以通过删除空格，或用"围绕你的说法，例如做然后需要将这些输入转换为实际列表。

使用argparse语法（optparse非常类似）：

# with spaces and " 
a_lst = [i for i in args.a.split(' ')] 
b_lst = [i for i in args.b.split(' ')] 

# without spaces and , 
a_lst = [i for i in args.a.split(',')] 
b_lst = [i for i in args.b.split(',')] 

---

另一种方式做，这将是要么导入要运行该模块并通过对象列表的一类交易与您的代码或通过使用while循环和raw_input/input收集所需的列表。

Answer 3

如果只有参数是列表和分隔符，你可以做到这一点相对简单：

sa = sys.argv.index('-a') 
sb = sys.argv.index('-b') 
lista = [int(i) for i in sys.argv[sa+1:sb]] 
listb = [int(i) for i in sys.argv[sb+1:]] 

添加验证很简单：

aval = [i for i in lista if i>1 and i<50] 
if len(aval) < len(lista): 
    print 'The -a list contains invalid numbers.' 
bval = [i for i in listb if i>1 and i<50] 
if len(bval) < len(listb): 
    print 'The -b list contains invalid numbers.' 

产生帮助信息：

if sys.argv[1] in ['-h', '-H'] or len(sys.argv) == 1: 
    print "Usage: <name> -a [list of integers] -b [list of integers]" 

Answer 4

​​支持nargs参数，它告诉你它吃了多少参数。 当nargs="+"它接受一个或多个参数，这样你就可以通过-b 1 2 3 4，它会被指定为一个列表b参数

# args.py 
import argparse 

p = argparse.ArgumentParser() 

# accept two lists of arguments 
# like -a 1 2 3 4 -b 1 2 3 
p.add_argument('-a', nargs="+", type=int) 
p.add_argument('-b', nargs="+", type=int) 
args = p.parse_args() 

# check if input is valid 
set_a = set(args.a) 
set_b = set(args.b) 

# check if "a" is in proper range. 
if len(set_a - set(range(1, 51))) > 0: # can use also min(a)>=1 and max(a)<=50 
    raise Exception("set a not in range [1,50]") 

# check if "b" is in "a" 
if len(set_b - set_a) > 0: 
    raise Exception("set b not entirely in set a") 

# you could even skip len(...) and leave just operations on sets 
# ... 

所以，你可以运行：

$ python arg.py -a 1 2 3 4 -b 2 20 
Exception: set b not entirely in set a 

$ python arg.py -a 1 2 3 4 60 -b 2 
Exception: set a not in range [1,50] 

这是有效的：

$ python arg.py -a 1 2 3 4 -b 2 3 

Answer 5

这为我工作：

parser.add_argument('-i', '--ids', help="A comma separated list IDs", type=lambda x: x.split(','))