列表理解中的缓存变量答案

【问题标题】：Cache variable in list comprehension列表理解中的缓存变量
【发布时间】：2020-05-22 00:14:36
【问题描述】：

假设我有一个昂贵的操作expensive(x: int) -> int 和以下列表理解：

# expensive(x: int) -> int
# check(x: int) -> bool
[expensive(i) for i in range(LARGE_NUMBER) if check(expensive(i))]

如果我想避免为每个i 运行两次expensive(i)，有没有办法通过列表理解来保存它的价值？

【问题讨论】：

【解决方案1】：

您可以模拟嵌套推导：

 [val for i in range(LARGE_NUMBER) for val in [expensive(i)] if check(val)]

【讨论】：

【解决方案2】：

使用海象：

[cache for i in range(LARGE_NUMBER) if check(cache := expensive(i))]

【讨论】：