什么是有条件地返回函数的最pythonic方式

Question

说我有2个功能.我希望func2返回func1 UNLESS func1返回None,在这种情况下func2返回其他内容.我有两种方法可以做到这一点,但他们都觉得有点不对劲.

我可以说:

def func1(n):
    if (condition):
        return foo

def func2(n):
    if func1(n) is not None:
        return func1(n)
    else:
        return something_else

但这感觉不对,因为我必须两次调用func1(n)(并且func1(n)是一个更大的计算).为了解决这个问题,我可以说:

def func1(n):
    if (condition):
        return foo

def func2(n):
    foo = func1(n)
    if foo is not None:
        return foo
    else:
        return something_else

但这感觉不对,因为我认为我不应该分配一个永远不会再次使用的新变量,只是为了检查func1是否返回None.

有没有更简单的方法来执行此操作,我不必再调用func1两次,而且我不必创建新变量？如果这是唯一的方法,你会推荐哪两个？我目前使用第二种方式(我将foo设置为func1返回的位置,而不是返回foo,除非foo == None)

另外,请记住,在我的实际代码中,我调用了几个不同的函数,我想返回第一个不是None的函数,这只是一个更简单的代码版本,可以解决问题.

Answer 1

自None评估以来False,你可以这样做:

def func2(n):
    return func1(n) or something_else

但是,应该指出的是,这将导致func2返回something_else,如果func1(n)返回任何falsey( ,0,[]等)

对于许多函数,您可以使用next和一些生成器表达式:

def myfunc(n):
    vals = (f(n) for f in (func1, func2, func3...))
    return next((v for v in vals if v is not None), something_else)

Answer 2

给调用func1的结果命名是相对便宜的,所以我会这样做,但是写这样的函数:

def func2(n):
    ret = func1(n)
    return ret if ret is not None else something_else