不少人不知道的Python 炫技操做：海象运算符的三种用法

首发于公众号：《Python编程时光》，一个专一输出高质量 Python开发知识的专栏

系列导读

1. 不少人不知道的Python 炫技操做：条件语句的七种写法
2. 不少人不知道的Python 炫技操做：合并字典的七种方法

---

Python 版本发展很是快，现在最新的版本已是 Pyhton 3.9，即使如此，有不少人甚至还停留在 3.6 或者 3.7，连 3.8 还没用上。

不少 Python 3.8 的特性还没来得及了解，就已经成为旧知识了，好比今天要说的海象运算符。

海象运算符是在 PEP 572 被提出的，直到 3.8 版本合入发布。

它的英文原名叫 Assignment Expressions，翻译过来也就是 赋值表达式，不过如今你们更广泛地称之为海象运算符，就是由于它长得真的太像海象了。

1. 第一个用法：if/else

可能有朋友是第一次接触这个新特性，因此仍是简单的介绍一下这个海象运算符有什么用？

在 Golang 中的条件语句能够直接在 if 中运算变量的获取后直接对这个变量进行判断，可让你少写一行代码

import "fmt"

func main() {
    if age := 20;age > 18 {
        fmt.Println("已经成年了")
    }
}
复制代码

若在 Python 3.8 以前，Python 必须得这样子写

age = 20
if age > 18:
    print("已经成年了")
复制代码

但有了海象运算符以后，你能够和 Golang 同样（若是你没学过 Golang，那这里要注意，Golang 中的 := 叫短变量声明，意思是声明并初始化，它和 Python 中的 := 不是一个概念）

if (age:= 20) > 18:
    print("已经成年了")
复制代码

2. 第二个用法：while

在不使用 海象运算符以前，使用 while 循环来读取文件的时候，你也许会这么写

file = open("demo.txt", "r")
while True:
    line = file.readline()
    if not line:
        break
    print(line.strip())
复制代码

但有了海象运算符以后，你能够这样

file = open("demo.txt", "r")
while (line := file.readline()):
    print(line.strip())
复制代码

使用它替换以往的无限 while 循环写法更为惊艳

好比，实现一个须要命令行交互输入密码并检验的代码，你也许会这样子写

while True:
   p = input("Enter the password: ")
   if p == "youpassword":
      break
复制代码

有了海象运算符以后，这样子写更为舒服

while (p := input("Enter the password: ")) != "youpassword":
   continue
复制代码

3. 第三个用法：推导式

这个系列的文章，几乎每篇都能看到推导式的身影，这一篇依旧如此。

在编码过程当中，我很喜欢使用推导式，在简单的应用场景下，它简洁且不失高效。

以下这段代码中，我会使用列表推导式得出全部会员中过于肥胖的人的 bmi 指数

members = [
    {"name": "小五", "age": 23, "height": 1.75, "weight": 72},
    {"name": "小李", "age": 17, "height": 1.72, "weight": 63},
    {"name": "小陈", "age": 20, "height": 1.78, "weight": 82},
]

count = 0

def get_bmi(info):
    global count
    count += 1

    print(f"执行了 {count} 次")

    height = info["height"]
    weight = info["weight"]

    return weight / (height**2)

# 查出全部会员中过于肥胖的人的 bmi 指数
fat_bmis = [get_bmi(m) for m in members if get_bmi(m) > 24]

print(fat_bmis)
复制代码

输出以下

执行了 1 次
执行了 2 次
执行了 3 次
执行了 4 次
[25.88057063502083]
复制代码

能够看到，会员数只有 3 个，可是 get_bmi 函数却执行了 4 次，缘由是在判断时执行了 3 次，而在构造新的列表时又重复执行了一遍。

若是全部会员都是过于肥胖的，那最终将执行 6 次，这种在大量的数据下是比较浪费性能的，所以对于这种结构，我一般会使用传统的for 循环 + if 判断。

fat_bmis = []

# 查出全部会员中过于肥胖的人的 bmi 指数
for m in members:
    bmi = get_bmi(m)
    if bmi > 24:
        fat_bmis.append(bmi)
复制代码

在有了海象运算符以后，你就能够不用在这种场景下作出妥协。

# 查出全部会员中过于肥胖的人的 bmi 指数
fat_bmis = [bmi for m in members if (bmi := get_bmi(m)) > 24]
复制代码

最终从输出结果能够看出，只执行了 3 次

执行了 1 次
执行了 2 次
执行了 3 次
[25.88057063502083]
复制代码

这里仅介绍了列表推导式，但在字典推导式和集合推导式中一样适用。再也不演示。

海象运算符，是一个新奇的特性，有很多人以为这样这种特性会破坏代码的可读性。确实在一个新鲜事物刚出来时是会这样，但我相信通过时间的沉淀后，愈来愈多的人使用它并享受它带来的便利时，这种争议也会慢慢消失在历史的长河中。

---