Question

我的要求是将元组作为命令行参数传递，如

--data (1,2,3,4)

我尝试使用argparse模块，但是如果我这样传递，它就会收到字符串'(1,2,3,4)'。我尝试为type=tuple提供argparse.add_argument，但此处没有用。

我是否必须添加一个新的类型类并将其传递给类型add_argument的参数？

更新

我根据答案尝试了ast.literal_eval。感谢那。但它在结果中给出了空格，如下所示。

(1,2,3,4)
<type 'str'>
(1, 2, 3, 4)
<type 'tuple'>

Answer 1

将nargs参数的data设置为nargs="+"（表示一个或多个）并输入int，然后可以在命令行上设置这样的参数：

--data 1 2 3 4

args.data现在将成为[1, 2, 3, 4]的列表。

如果你必须有一个元组，你可以这样做：

my_tuple = tuple(args.data)

全部放在一起：

parser = argparse.ArgumentParser()
parser.add_argument('--data', nargs='+', type=int)
args = parser.parse_args()
my_tuple = tuple(args.data)

Answer 2

我有同样的要求。在标准包中找不到。所以，推出了这个。蟒蛇 3 倍。放入简单的 argv 解析器，一次调用一个 arg：

var 是默认传递的，类型正确。
argval 是找到的命令行字符串。逗号分隔，没有空格。即 1、2、3 或一、二、三。

与 bash shell 一起使用，可能需要引号，其他 shell、操作系统的其他调整。相同的技术适用于非列表类型。此版本基于 var[0] 创建同构列表或元组。

示例：

./foo.py   --mytuple 1,2,3,4,5

代码：

def chk_arg( var, argstr, dbg=False): 
    ### loop on sys.argv here .....
    # if list/tuple type, use list comprehension
    # converts csv list to list or tuple 'type(var)' of 'type(var[0])' items. 
    if type(var) is list or type(var) is tuple:  
        val = type(var)([type(var[0])(i) for i in argval.split(',')])

Answer 3

如果你想扩展它，而不是处理语法树，并在参数中提供元组转换，你需要一个自定义的 argparse 类型。（我需要这个，因为我的解析器在一个包中，我不想要求我的用户都写 my_tuple = tuple(args.data)。使用参数类型，您还可以像原始问题一样接收带括号的字符串。另外，我不会包括 nargs，因为这会将您的自定义类型输出包装到一个列表中，这基本上违背了您的自定义类型的目的。我映射了一个浮点数，因为我的输入类似于“0.9, 0.99”，但您的 int 工作原理相同：

def tuple_type(strings):
    strings = strings.replace("(", "").replace(")", "")
    mapped_int = map(int, strings.split(","))
    return tuple(mapped_int)

parser = argparse.ArgumentParser()
parser.add_argument('--data', type=tuple_type)
args = parser.parse_args()

这样您就可以发送 --data "(1,2,3,4)" 或 --data "1,2,3,4"

将元组作为命令行参数传递

3 个答案: