获取传递给函数的参数的列表/元组/字典？

Question

鉴于以下功能:

def foo(a, b, c):
    pass

如何获得传入的参数的list/tuple/dict/etc,而不必自己构建结构？

具体来说,我正在寻找Python的JavaScript arguments关键字版本或PHP的func_get_args()方法.

我不想要的是使用*args或的解决方案**kwargs; 我需要在函数定义中指定参数名称(以确保它们被传入),但在函数中我希望在list-或dict-style结构中使用它们.

Answer 1

您可以使用locals()在函数中获取局部变量的dict,如下所示:

def foo(a, b, c):
    print locals()

>>> foo(1, 2, 3)
{'a': 1, 'c': 3, 'b': 2}

但是,这有点hackish,因为locals()返回本地范围内的所有变量,而不仅仅是传递给函数的参数,所以如果你不在函数的最顶层调用它,结果可能包含的信息比你想要的多:

def foo(a, b, c):
    x = 4
    y = 5
    print locals()

>>> foo(1, 2, 3)
{'y': 5, 'x': 4, 'c': 3, 'b': 2, 'a': 1}

我宁愿在你的函数顶部构建一个dict或变量列表,如其他答案所示.它更明确,并以更清晰的方式传达代码的意图,恕我直言.

Answer 2

您可以使用inspect模块:

def foo(x):
    return x

inspect.getargspec(foo)
Out[23]: ArgSpec(args=['x'], varargs=None, keywords=None, defaults=None)

这是一个重复的这个和这个.

Answer 3

如果参数不符合预期，我将使用*argsor**kwargs并抛出异常

如果你想得到与 python 检查的错误相同的错误，你可以做类似的事情

def check_arguments(function_name,args,arg_names):
    missing_count = len(arg_names) - len(args)
    if missing_count > 0:
        if missing_count == 1:
            raise TypeError(function_name+"() missing 1 required positionnal argument: "+repr(arg_names[-1]))
        else:
            raise TypeError(function_name+"() missing "+str(missing_count)+" required positionnal argument: "+", ".join([repr(name) for name in arg_names][-missing_count:-1])+ " and "+repr(arg_names[-1]))

与类似的东西一起使用

def f(*args):
    check_arguments("f",args,["a","b","c"])
    #whatever you want
    ...

Answer 4

这是您可以调用以获取当前函数的 kwargs 的函数。或者，如果您想直接在您自己的函数中使用这些行而不是调用，get_kwargs()只需删除.f_back

import inspect


def get_kwargs():
    frame = inspect.currentframe().f_back
    keys, _, _, values = inspect.getargvalues(frame)
    kwargs = {}
    for key in keys:
        if key != 'self':
            kwargs[key] = values[key]
    return kwargs


def test(x, y=100):
    z = 7
    print(get_kwargs())


test(5, 6)
# Output: {'x': 5, 'y': 6}
test(5)
# Output: {'x': 5, 'y': 100}