在 Python 中处理时，如何确保在短路点之后不会急切地评估任意 function 调用的列表？

Question

例如，给定

def expensive_call(x):
    print(x)
    if x == "d":
        return x
def expensive_call_2(x, y):
    print(x)
    print(y)
    return x + y

a = [expensive_call("a"), expensive_call_2("b", "c"), expensive_call("d")]
next((e for e in a if e is not None), 'All are Nones')

output 是

a
b
c
d
Out[22]: 'bc'

由于expensive_call("d")被急切地评估，请注意，即使next调用在第二次调用短路时使用“bc”的 output，也会打印“d”。

我正在对列表a中的调用进行硬编码，并且a不必是列表数据结构。

一种可能的解决方案如下：

a = ['expensive_call("a")', 'expensive_call_2("b", "c")', 'expensive_call("d")']
def generator():
    for e in a:
        r = eval(e)
        if r is not None:
            yield r
next(generator(), 'All are Nones')

output 是

a
b
c
Out[23]: 'bc'

如预期的。 但是，我真的不喜欢使用 eval。 我也不想使用任何最初将 function 指针和 arguments 分开的解决方案，例如(expensive_call, ("a")) 。 理想情况下，我会有类似的东西

a = lazy_magic([expensive_call("a"), expensive_call_2("b", "c"), expensive_call("d")])
next((e for e in a if e is not None), 'All are Nones')

请注意， https://stackoverflow.com/a/3405828/2750819是一个类似的问题，但仅适用于函数具有相同方法签名的情况。

Answer 1

您可以将它们全部放入 function 并产生结果：

def gen():
    yield expensive_call("a")
    yield expensive_call_2("b", "c")
    yield expensive_call("d")


result = next(
    (value for value in gen() if value is not None),
    'All are Nones')

另一种解决方案是使用partial应用程序：

from functools import partial

calls = [partial(expensive_call, 'a'),
         partial(expensive_call_2, 'b', 'c'),
         partial(expensive_call, 'd')]

然后评价：

next((result for call in calls
      for result in [call()]
      if result is not None),
     'All results None')

Answer 2

您可以使用以下装饰器：

def lazy_fn(fn):
    return lambda *args: lambda: fn(*args)

（也可以表示为lazy_fn = lambda fn: lambda *args: lambda: fn(*args)如果你喜欢 lambda.）

像这样使用它：

@lazy_fn
def expensive_call(x):
    print(x)
    if x == "d":
        return x

@lazy_fn
def expensive_call_2(x, y):
    print(x)
    print(y)
    return x + y

a = [expensive_call("a"), expensive_call_2("b", "c"), expensive_call("d")]
print(next((e for e in map(lambda i: i(), a) if e is not None), 'All are Nones'))

输出：

a
b
c
bc

请注意，您需要for e in map(lambda i: i(), a)中使用for e in a中使用 for e。