让 Python 函数根据参数数量干净地返回标量或列表

Question

免责声明：我正在寻找一种 Python 2.6 解决方案（如果有的话）。

我正在寻找一个在传递单个值时返回单个值的函数，或者在传递多个值时返回一个序列的函数：

>>> a = foo(1)
2
>>> b, c = foo(2, 5)
>>> b
3
>>> c
6

要明确的是，这是为了使某些函数调用看起来比：

a, = foo(1)

或

a = foo(1)[0]

现在 更好，不优雅的解决方案是这样的：

def foo(*args):
    results = [a + 1 for a in args]
    return results if len(results) > 1 else results[0]

是否有任何语法糖（或函数）可以让这感觉更干净？像下面这样的东西吗？

def foo(*args):
    return *[a + 1 for a in args]

Answer 1

您始终可以编写一个装饰器来删除该 if 语句（如果这对您来说更好的话）：

import functools
def unpacked(method):
    @functools.wraps(method)
    def _decorator(*args):
        result = method(*args)
        return results if len(results) != 1 else results[0]
    return _decorator

用法：

@unpacked
def foo(*args):
    return [arg + 1 for arg in args]

Answer 2

您可以轻松地编写一个函数scalify，如果列表只有一个元素，它会从列表中返回元素，即它会尝试使其成为标量（因此得名）。

def scalify(l):
    return l if len(l) > 1 else l[0]

然后你可以在你的函数中使用它，如下所示：

def foo(*args):
    return scalify([a + 1 for a in args])

这可以解决问题，但我同意那些建议你不要这样做的人。原因之一是，除非您知道至少传入了两项，否则它会排除对结果进行迭代。另外，如果您有一个列表，则在调用该函数时必须解压该列表，从而失去其“列表性”，并且您知道您可能无法取回列表。在我看来，这些缺点掩盖了您可能看到的该技术的任何好处。

Answer 3

您的意思是您想要一个具有相同数量参数的元组吗？这不是解决办法吗？

返回元组([a + 1 for a in args])

Answer 4

def foo(*args):
    return (None, args[0]+1 if args else None, map(lambda a: a + 1, args))[len(args) if len(args) < 3 else 2]

:-) 这是地狱

Answer 5

这将处理 0 个或多个参数，我认为这就是您正在寻找的。

def foo(*args):
    return map(lambda x: x + 1, args) or [None]

编辑：我修改为添加一个 None 列表，以防解压 0 参数