海象操作员:如果连续出现奇怪的结果

Question

我正在使用海象运算符:=，该运算符将在Python 3.8中可用。

跟

def f(x):
    return x**3
old_list =  = list(range(9))

这按预期工作(不要介意该示例毫无意义...

new_list = [fx for x in old_list if (fx := f(x))  in {1, 8, 343}]
# [1, 8, 343]

这是有效的并运行，但返回一些意外的东西：

new_list = [fx := f(x) for x in old_list if fx in {1, 8, 343}]
# []

这是怎么回事？

Answer 1

TL;DR 你的第二个不是"那个"有效：

>>> old_list = list(range(9))
>>> f = lambda x: x ** 3
>>> [fx := f(x) for x in old_list if fx in {1, 8, 343}]
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
  File "<stdin>", line 1, in <listcomp>
NameError: name 'fx' is not defined
>>> 

解释

在你的第一个列表推导中，[fx for x in old_list if (fx := f(x)) in {1, 8, 343}] ，fx :=在其外部创建一个变量：

>>> old_list = list(range(9))
>>> f = lambda x: x ** 3
>>> [fx for x in old_list if (fx := f(x))  in {1, 8, 343}]
[1, 8, 343]
>>> fx
512

如果之后运行[fx := f(x) for x in old_list if fx in {1, 8, 343}]，它将如下所示：

• for x in old_list将 x 绑定到 0
• if fx in {1, 8, 343}，fx是旧的512，这是错误的
• 由于 if 为 false，则不执行"选择"(左(部分，不调用 f，不重新绑定 fx，fx 仍然是 512。
• 将 x 指定为 old_list 的第二个元素，即 1
• 。

我们可以像这样仔细检查这种解释：

>>> [print("no no no") for x in old_list if fx in {1, 8, 343}]
[]

no no no永远不会打印，所以这边永远不会被执行(因为fx是 512，而不是在set中(。