如何在列表理解中按函数返回值执行过滤器？

Question

以下是基本逻辑：

result = []

for item in item_lst:
    code = foo(item)
    if code != -1:
        result.append(code)

有了列表理解，我应该写两次：

result = [foo(item) for item in item_lst]
result = [code for code in result if code != -1]

要么

result = [foo(item) for item in item_lst if foo(item) != -1]

它将两次调用函数foo。

有没有更好的解决方案？

Answer 1

您可以使用生成器表达式来避免创建第二个列表：

result = [code for code in (foo(item) for item in item_ls) if code != -1]

这里：

(foo(item) for item in item_ls) 

是一个生成器表达式。没有创建中间列表。这可能有助于节省内存。