Python walrus运算符在范围内失败

Question

为什么我可以用这样的方法

import numpy as np

print( (g := np.arange(256)**2) / np.max(g) )

但是下面这些都失败了？

foo = list(range(256))
for i in range( (l := len(foo)) // 16 + 0 if l%16 == 0 else 1 ):
    print(i)

Traceback (most recent call last):
  File "test.py", line 8, in <module>
    for i in range( (l := len(foo)) // 16 + 0 if l%16 == 0 else 1 ):
NameError: name 'l' is not defined

Answer 1

在参数列表中，表达式从左到右计算，因此在：

plt.plot( range( len( (Y := foo) )), Y )

第一个参数range( len( (Y := foo) ))是在第二个参数Y之前计算的，因此在将Y作为第二个参数引用之前，Y是用foo定义的。

但是，在条件表达式中，if子句中的表达式在计算输出表达式之前对表达式进行计算，因此如下所示：

(l := len(foo)) // 16 + 0 if l%16 == 0 else 1

首先对l%16 == 0进行评估，因为此时还没有定义l，因此它会引发所述的NameError。

相反，您可以首先在l子句中定义if，如果这确实是您想要的逻辑：

for i in range( l // 16 + 0 if (l := len(foo))%16 == 0 else 1 ):