在argparse的add_argument方法中使用nargs参数

我使用nargs='*'作为add_argument参数。如果我没有传递任何显式参数，我特别使用nargs='*'选项来选择默认值

包括一个代码片段作为示例:

例如:temp_args1.py

请注意:下面的示例代码是用python3编写的。通过改变打印语句的格式，可以在python2中运行

#!/usr/local/bin/python3.6

from argparse import ArgumentParser

description = 'testing for passing multiple arguments and to get list of args'
parser = ArgumentParser(description=description)
parser.add_argument('-i', '--item', action='store', dest='alist',
                    type=str, nargs='*', default=['item1', 'item2', 'item3'],
                    help="Examples: -i item1 item2, -i item3")
opts = parser.parse_args()

print("List of items: {}".format(opts.alist))

注意:我正在收集多个字符串参数，这些参数存储在list - opt . list中 如果需要整数列表，请更改解析器上的类型参数。Add_argument到int

执行结果:

python3.6 temp_agrs1.py -i item5 item6 item7
List of items: ['item5', 'item6', 'item7']

python3.6 temp_agrs1.py -i item10
List of items: ['item10']

python3.6 temp_agrs1.py
List of items: ['item1', 'item2', 'item3']

在add_argument()中，type只是一个可调用的对象，它接收字符串并返回选项值。

import ast

def arg_as_list(s):                                                            
    v = ast.literal_eval(s)                                                    
    if type(v) is not list:                                                    
        raise argparse.ArgumentTypeError("Argument \"%s\" is not a list" % (s))
    return v                                                                   


def foo():
    parser.add_argument("--list", type=arg_as_list, default=[],
                        help="List of values")

这将允许:

$ ./tool --list "[1,2,3,4]"

我更喜欢传递一个带分隔符的字符串，稍后在脚本中解析。原因是;列表可以是任何类型的int或str，有时使用nargs，如果有多个可选参数和位置参数，我会遇到问题。

parser = ArgumentParser()
parser.add_argument('-l', '--list', help='delimited list input', type=str)
args = parser.parse_args()
my_list = [int(item) for item in args.list.split(',')]

然后,

python test.py -l "265340,268738,270774,270817" [other arguments]

or,

python test.py -l 265340,268738,270774,270817 [other arguments]

会很好。分隔符也可以是空格，这将强制在参数值周围加上引号，就像问题中的示例一样。

或者你也可以使用Chepner评论中建议的lambda类型:

parser.add_argument('-l', '--list', help='delimited list input', 
    type=lambda s: [int(item) for item in s.split(',')])

如果你有一个嵌套列表，其中内部列表有不同的类型和长度，你想要保留类型，例如:

[[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]])

我想处理传递多个列表，整数值和字符串。

如何将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变量按顺序包含参数。

请注意，如果你传递action='append'和默认参数，Argparse将尝试附加到提供的默认值，而不是替换默认值，这可能是你所期望的，也可能不是。

下面是Argparse Docs中给出的一个action='append示例。 在这种情况下，事情将如预期的那样工作:

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

Out[2]: Namespace(foo=['1', '2'])

然而，如果你选择提供一个默认值，Argparse的"append"操作将尝试附加到提供的默认值，而不是替换默认值:

import argparse
REASONABLE_DEFAULTS = ['3', '4']
parser = argparse.ArgumentParser()
parser.add_argument('--foo', default=REASONABLE_DEFAULTS,action='append')
parser.parse_args('--foo 1 --foo 2'.split())

Out[6]: Namespace(foo=['3', '4', '1', '2'])

如果你希望Argparse替换默认值——比如传入一个元组作为默认值，而不是一个列表——这可能会导致一些令人困惑的错误:

import argparse
REASONABLE_DEFAULTS = ('3', '4')
parser = argparse.ArgumentParser()
parser.add_argument('--foo', default=REASONABLE_DEFAULTS,action='append')
parser.parse_args('--foo 1 --foo 2'.split())

AttributeError: 'tuple' object has no attribute 'append'

有一个跟踪这种意外行为的bug，但由于它可以追溯到2012年，所以不太可能得到解决。