python - "introspect"函数内预期返回量

Question

大家好，
我目前正在开发一个 python 模块，为同事提供使用工具，我试图提供的一些功能可以返回相当多的有用信息(比如 5 个返回值)，但在大多数用例中，只需要前两个。

我知道 python 非常适合编写高度动态的程序，尤其是因为我们可以询问有关当前正在运行的系统的信息，所以我想搜索一下，但我似乎找不到。 (我发现了很多有趣的讨论，但没有真正讨论我的问题。)

有没有办法从函数内部知道预期返回值的数量？

例如，可能有函数:

def lazyOdd():
    return 1, 3, 5, 7, 9, 11, 13, 15, 17, 19, 21

将使用 one, three, five = lazyOdd() 调用。

编辑:显然一点都不清楚，我寻找的看起来像:

def lazyReturn():
    return (val1, val2) if expectedReturns == 2 else (val1, val2, val3, val4)

我知道在这种情况下，这绝对不是继续进行的最佳方式，几乎任何一种可迭代对象都可以，但是对于根本没有链接的值，它不会那么简单。如果可能的话，因为大多数用例将只使用两个结果，我想避免返回一个大元组，因为许多同事甚至不知道该函数返回其他值，这将是非常错误的 -俯卧。

我可以这样做吗？

编辑:此外，我不介意解决方案是否包含一些开销，因为我很可能会通过不处理最后三个值来赢得很多时间。

Edit²:也许最好的方法真的是传递另一个指定预期值数量的参数？

Answer 1

缓存

从你的描述中很难判断，但你似乎需要caching .

这样，您就可以编写多个函数。一个用于繁重的计算，其他的输出正确的格式和值的数量。如果使用相同的输入值第二次调用繁重的计算，则返回之前的结果。

from functools import lru_cache

@lru_cache(maxsize=32)
def heavy_common_calculation(x):
    print("HEAVY CALCULATION FOR %d. Should only be done once!" % x)
    result = x**2
    return result

def get_two_values(x):
    return x, heavy_common_calculation(x)

def get_four_values(x):
    y = heavy_common_calculation(x)
    return x, y, y**2, y**3

print(get_two_values(3))
# HEAVY CALCULATION FOR 3. Should only be done once!
# (3, 9)
print(get_four_values(3))
# (3, 9, 81, 729)

生成器

你想实现一个generator吗？ ？

def lazyOdd():
    x = 1
    while True:
        yield x
        x = x + 2

odd_gen = lazyOdd()
print(next(odd_gen))
# 1
print(next(odd_gen))
# 3
print(next(odd_gen))
# 5
print(next(odd_gen))
# 7

如果你想要一个值，它会被计算出来。如果您不需要它，则不会发生任何事情，也不会白白计算任何事情。

len(odd_gen) 未定义，因为它会启动无限循环。

关键字参数

您可以使用一个参数来告诉函数您需要多少个结果:

def f(x, detail='low'):
    a, b = x, x**2
    if detail == 'high':
        return a, b, 'something', 'else'
    else:
        return a, b


print(f(3))
# (3, 9)
print(f(3, detail='high'))
# (3, 9, 'something', 'else')