Python 使用 for/if 转换和过滤列表

Question

有没有办法在单个列表理解中同时进行转换和过滤，即:

def transform(el):
    if some_condition(el):
        return None

    return complex_logic(el)

def main():
    transformed = [transform(el) for el in some_list if transform(el) != None]

但要避免调用 transform 两次？ IE。将它分配给一个变量，比如(在伪 Python 中):

def main():
  transformed = [transformed for el in some_list let transformed = transform(el) if transformed != None]

Answer 1

从 Python 3.8 开始，您可以使用 walrus operator :=:

def main():
    return [res for el in some_list if (res := transform(el)) is not None]

这样，调用 transform 函数的结果将存储在 res 中，然后您可以在列表理解的表达式部分使用它。