Оператор моржа: если у него странные результаты
Я играл (в связи с этим вопросом) с оператором моржа :=
который будет доступен в 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}]
# []
что здесь происходит?
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
связывает х с 0if fx in {1, 8, 343}
FX старая512
что неверно- так как if имеет значение false, часть "select" (left) не выполняется, нет вызова 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
).