海象运算符：如果条件有奇怪的结果

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]

这是有效的并且可以运行，但是 returns 出乎意料的事情：

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

这里发生了什么？

Answer 1

TL;DR 你的第二个 "that" 无效：

>>> 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，这是 false
因为if为false，"select"（左）部分没有执行，没有调用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）。

海象运算符：如果条件有奇怪的结果

walrus operator: if contition with strange results

python

python-3.x

python-3.8

说明