Оператор моржа: если у него странные результаты

Я играл (в связи с этим вопросом) с оператором моржа := который будет доступен в 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 связывает х с 0
  • if 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).

Другие вопросы по тегам