一个 If 语句中的两个海象运算符

Question

是否有在 1 个 if 语句中包含两个海象运算符的正确方法？

if (three:= i%3==0) and (five:= i%5 ==0):
    arr.append("FizzBuzz")
elif three:
    arr.append("Fizz")
elif five:
    arr.append("Buzz")
else:
    arr.append(str(i-1))

此示例适用于 three，但 five 将“未定义”。

Answer 1

您遇到的问题是 five 仅在 three 在此语句中为真时才分配，因为 short circuiting:

if (three:= i%3==0) and (five:= i%5 ==0)

因此 five 通常未分配，导致 NameError 或使用非当前值。

您可以通过在其中使用海象赋值形成一个非空元组然后在该元组之后使用 three 和 five 来强制使用 True 值。

它并不比在 if 之前分配 three 和 five 更漂亮，但这有效：

arr=[]
for i in range(1,26):
    if (three:=i%3==0, five:=i%5==0) and three and five:
        arr.append(f"{i} FizzBuzz")
    elif three:
        arr.append(f"{i} Fizz")
    elif five:
        arr.append(f"{i} Buzz")
    else:
        arr.append(f"{i}")

 >>> arr
 ['1', '2', '3 Fizz', '4', '5 Buzz', '6 Fizz', '7', '8', '9 Fizz', '10 Buzz', '11', '12 Fizz', '13', '14', '15 FizzBuzz', '16', '17', '18 Fizz', '19', '20 Buzz', '21 Fizz', '22', '23', '24 Fizz', '25 Buzz']

任何非空元组都是 Python 中的 True。形成它会导致 (three:=i%3==0, five:=i%5==0) 始终为真，并且每次都会分配 3 和 5。由于该元组为真，因此必须使用正确的值三和五来评估表达式的其余部分。

或者，使用 if all((three:=i%3==0, five:=i%5==0)): 因为元组是在测试其内容之前形成的——即使 all 短路；这只会在元组形成后发生。

这两种形式中的任何一种都可以轻松重构为推导式：

arr=[f"{i} FizzBuzz" if three and five 
             else f"{i} Fizz" if three 
             else f"{i} Buzz" if five 
             else f"{i}" 
                 for i in range(1,26) if (three:=i%3==0, five:=i%5==0)]

或者，

arr=[f"{i} FizzBuzz" if all((three:=i%3==0, five:=i%5==0)) 
                     else f"{i} Fizz" if three 
                     else f"{i} Buzz" if five 
                     else f"{i}" for i in range(1,26)]

如果 each 元素的结果不是布尔值，请注意构造 if (three := i % 3 == 0) & (five := i % 5 == 0):。你可能会遇到一些意想不到的失败：

>>> bool((x:=3) & (y:=4))
False
>>> bool((x:=3) and (y:=4))
True

解决此问题的唯一方法是对每个应用 bool：

>>> bool(x:=3) & bool(y:=4)
True

顺便说一句，说到元组，在 Python 中进行 FizzBuzz 类型挑战的更短方法：

fb={(True,True):"{} FizzBuzz", 
    (True,False):"{} Fizz", 
    (False,True):"{} Buzz", 
    (False,False):"{}"}

arr=[fb[(i%3==0,i%5==0)].format(i) for i in range(1,26)]

如果您正在寻找 新东西 这种类型的问题对于 Python 3.10+ pattern matching:

来说很自然

arr=[]
for i in range(1,26):
    s=f"{i}"
    match (i%3==0,i%5==0):
        case (True, (True | False) as oth):
            s+=" FizzBuzz" if oth else " Fizz"
        
        case (False, True):
            s+=" Buzz"
            
    arr.append(s)

Answer 2

logical operator and 仅有条件地评估其第二个操作数。 无条件需要条件赋值，没有正确的方法。

而是使用 "binary" operator &，它无条件地计算其第二个操作数。

arr = []
for i in range(1, 25):
    #                        v force evaluation of both operands
    if (three := i % 3 == 0) & (five := i % 5 == 0):
        arr.append("FizzBuzz")
    elif three:
        arr.append("Fizz")
    elif five:
        arr.append("Buzz")
    else:
        arr.append(str(i))

print(arr)
# ['1', '2', 'Fizz', '4', 'Buzz', 'Fizz', '7', '8', 'Fizz', 'Buzz', '11', ...]

相应地，可以使用|作为or的无条件变体。此外，“异或”运算符 ^ 根本不等同于条件求值。

值得注意的是，二元运算符将布尔值计算为纯布尔值 - 例如，False | True 是 True 而不是 1 - 但对于其他类型可能有不同的工作方式。要使用二元运算符在布尔上下文中评估任意值，例如 lists，请在赋值后将它们转换为 bool：

#  |~~~ force list to boolean ~~| | force evaluation of both operands
#  v    v~ walrus-assign list ~vv v
if bool(lines := list(some_file)) & ((today := datetime.today()) == 0):
   ...

由于赋值表达式需要括号来保证正确的优先级，different precedence在逻辑（and、or）和二进制（&、[=14]之间的常见问题=], ^) 运算符在这里无关紧要。

一个 If 语句中的两个海象运算符

Two Walrus Operators in one If Statement

python

python-3.x

walrus-operator