оператор моржа: если у него странные результаты - PullRequest
1 голос
/ 10 июля 2019

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

3 голосов
/ 10 июля 2019

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 ложно, часть "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).

...