如何在列表理解中设置局部变量？

Question

我有一个方法，它接受一个列表并返回一个对象：

# input a list, returns an object
def map_to_obj(lst):
    a_list = f(lst)
    return a_list[0] if a_list else None

我想获取一个包含所有not的映射元素的列表None。

像这样：

v_list = [v1, v2, v3, v4]

[map_to_obj(v) for v in v_list if map_to_obj(v)]

但是map_to_obj在列表推导中两次调用该方法似乎并不好。

有没有一种方法可以在列表推导中包含局部变量，以使其具有更好的性能？

还是编译器自动对其进行优化？

这是我想要的：

(sml like)
[let mapped = map_to_obj(v) in for v in v_list if mapped end]

Answer 1

从开始Python 3.8，并引入赋值表达式（PEP
572）
（:=运算符），可以在列表推导中使用局部变量，以避免调用同一函数两次：

在我们的例子中，我们可以使用表达式的结果来过滤列表时将的评估命名map_to_obj(v)为变量o。并因此o用作映射值：

[o for v in [v1, v2, v3, v4] if (o := map_to_obj(v))]