海象操作符（:=）

从python3.8开始，支持海象操作符（walrus operator :=），对应PEP572。

海象操作符的特点是将赋值和判断放在一起，简化了代码，提高了效率。其实，C语言早就是这样的使用，比如在if直接用一个=赋值，然后对赋值后的变量进行判断。Python从3.8开始支持，反正是有了吧。据说Go语言一开始就有。

>>> a = [1,2,3,4,5,6,7]
>>> if len(a) > 5:
...     print('length of a: %d' % len(a))  # call len twice
...
length of a: 7
>>>
>>> if n := len(a) > 5:  # wrong, > will be executed first
...     print('length of a: %d' % n)
...
length of a: 1
>>>
>>> if (n := len(a)) > 5:  # call len only once, assign n and compare
...     print('length of a: %d' % n)
...
length of a: 7

看上面的示例代码，第一段是传统写法，调用了2次len函数；第二段是错误的，大于符号的优先级高于海象符号（为什么是这样，请参考：python的连续赋值）；第三段才是主题，在if中直接给变量n赋值，然后对 n > 5进行判断。

>>> s
'abcde 12345 pynote.net'
>>> ore = re.search(r'\d+', s)
>>> if ore:
...     print(ore.group())
...
12345
>>>
>>> if ore := re.search(r'\d+', s):
...     print(ore.group())
...
12345
>>> if ore := re.search(r'tiantian', s):
...     print(ore.group())
... else:
...     print('not found')
...
not found
>>> ore
>>> ore is None
True

以上是另一段示例，注意最后一段在if中assign的ore对象是None。

>>> ss
['abcde', 'bcdef', 'a123', '000', 'b123', 'a444']
>>>
>>> def string_sum(ss):
...     return sum([ord(x) for x in ss])
>>>
>>> [x for i in ss if (x:=string_sum(i)) > 100]
[495, 500, 247, 144, 248, 253]

海象操作符在list comprehension中的应用。

while (line := file.readline()):
    pass  # your code here

-- EOF --

相关文章

• generator 生成器
• Python的布尔操作
• 迭代器（Iterator）和可迭代对象（Iterable）
• 对象比较用"=="还是"is"？
• Python带条件判断的赋值语句
• pass和Ellipsis（...）
• 匿名函数lambda
• python独有的交换算法
• 循环后的else分支
• python变量的作用域和引用，global和nonlocal申明
• python的空格缩进
• 在Python函数中定义嵌套函数
• from module import * 中的细节
• python的间隔切片技巧
• python的import module机制
• python的连续赋值语句
• Python实现do...while逻辑