第05课:分支结构

迄今为止,我们写的 Python 代码都是一条一条语句按顺序向下执行的,这种代码结构叫做顺序结构。然而仅有顺序结构并不能解决所有的问题,比如我们设计一个游戏,游戏第一关的过关条件是玩家获得1000分,那么在第一关完成后,我们要根据玩家得到分数来决定是进入第二关,还是告诉玩家“Game Over”。在这样的场景下,我们的代码就会产生两个分支,而且这两个分支只有一个会被执行。类似的场景还有很多,我们将这种结构称之为“分支结构”或“选择结构”。给大家一分钟的时间,你应该可以想到至少5个以上类似的例子,赶紧试一试吧!

使用if和else构造分支结构

在 Python 中,要构造分支结构可以使用ifelifelse三个关键字。所谓关键字就是编程语言中有特殊含义的单词,像ifelse就是专门用于构造分支结构的关键字,很显然你不能够使用它作为变量名。当然,我们并不是每次构造分支结构都会把三个关键字全部用上,下面我们通过例子加以说明。

我们来写一个身体质量指数(BMI)的计算器。身体质量质数也叫体质指数,是国际上常用的衡量人体胖瘦程度以及是否健康的一个标准,计算公式如下所示。通常认为$\small{18.5 \le BMI < 24}$是正常范围,$\small{BMI < 18.5}$说明体重过轻,$\small{BMI \ge 24}$说明体重过重,$\small{BMI \ge 27}$就属于肥胖的范畴了。

$$ BMI = \frac{体重}{身高^{2}} $$

说明:上面公式中的体重以千克(kg)为单位,身高以米(m)为单位。

1"""
2BMI计算器
3
4Version: 1.0
5Author: 骆昊
6"""
7height = float(input('身高(cm):'))
8weight = float(input('体重(kg):'))
9bmi = weight / (height / 100) ** 2
10print(f'{bmi = :.1f}')
11if 18.5 <= bmi < 24:
12    print('你的身材很棒!')

提示if语句的最后面有一个:,它是用英文输入法输入的冒号;程序中输入的'"=() 等特殊字符,都是在英文输入法状态下输入的,这一点之前已经提醒过大家了。很多初学者经常会忽略这一点,等到执行代码时,就会看到一大堆错误提示。当然,认真读一下错误提示还是很容易发现哪里出了问题,但是 强烈建议大家在写代码的时候切换到英文输入法,这样可以避免很多不必要的麻烦。

上面的代码中,我们在计算和输出 BMI 之后,加上了一段分支结构,如果满足$\small{18.5 \le BMI < 24}$,程序会输出“你的身材很棒!”,但是如果不满足条件,这段输出就没有了。这就是我们上面说的代码有不同的执行路径,有些代码不一定会执行到。我们在if 关键字的后面给出了一个表达式18.5 <= bmi < 24,之前我们说过,关系运算会产生布尔值,如果if后面的布尔值为True,那么if 语句下方,有四个空格缩进的print('你的身材很棒!')就会被执行。我们先输入几组数据运行上面的代码,如下所示。

第一组输入:

身高(cm):175 体重(kg):68 bmi = 22.2 你的身材很棒!

第二组输入:

身高(cm):175 体重(kg):95 bmi = 31.0

第三组输入:

身高(cm):175 体重(kg):50 bmi = 16.3

只有第一组输入的身高和体重计算出的 BMI 在18.5到24这个范围值内,所以触发了if条件,输出了“你的身材很棒”。需要说明的是,不同于 C、C++、Java 等编程语言,Python 中没有用花括号来构造代码块而是使用缩进的方式来表示代码的层次结构,如果if 条件成立的情况下需要执行多条语句,只要保持多条语句具有相同的缩进就可以了。换句话说,若干行连续的语句如果保持了相同的缩进,那么它们就属于同一个 代码块,相当于是一个执行的整体。缩进可以使用任意数量的空格,但通常使用4个空格,强烈建议大家**不要使用制表键(Tab键)来缩进代码 **,如果你已经习惯了这么做,可以设置你的代码编辑器自动将1个制表键变成4个空格,很多代码编辑器都支持这项功能,PyCharm 中默认也是这样设定的。还有一点,在 C、C++、Java 等编程语言中,18.5 <= bmi < 24要写成两个条件bmi >= 18.5bmi < 24 ,然后把两个条件用逻辑与运算符连接起来,Python 中也可以这么做,例如刚才的if语句也可以写成if bmi >= 18.5 and bmi < 24: ,但是没有必要,难道if 18.5 <= bmi < 24:这个写法它不香吗?下面用 Java 代码做了同样的事情,看不懂 Java 代码没关系,感受一下它和 Python 语法的区别就可以了。

1import java.util.Scanner;
2
3class Test {
4
5    public static void main(String[] args) {
6        try (Scanner sc = new Scanner(System.in)) {
7            System.out.print("身高(cm): ");
8            double height = sc.nextDouble();
9            System.out.print("体重(kg): ");
10            double weight = sc.nextDouble();
11            double bmi = weight / Math.pow(height / 100, 2);
12            System.out.printf("bmi = %.1f\n", bmi);
13            if (bmi >= 18.5 && bmi < 24) {
14                System.out.println("你的身材很棒!");
15            }
16        }
17    }
18}

说明:上面就是 BMI 计算器1.0版本对应的Java代码,欢迎在评论区吐槽它。

接下来,我们对上面的代码稍作修改,在 BMI 不满足$\small{18.5 \le BMI < 24}$的情况下,也给出相信的提示信息。我们可以在if 代码块的后面增加一个else代码块,它会在if语句给出的条件没有达成时执行,如下所示。很显然,if 下面的print('你的身材很棒!')else下面的print('你的身材不够标准哟!')只有一个会被执行到。

1"""
2BMI计算器
3
4Version: 1.1
5Author: 骆昊
6"""
7height = float(input('身高(cm):'))
8weight = float(input('体重(kg):'))
9bmi = weight / (height / 100) ** 2
10print(f'{bmi = :.1f}')
11if 18.5 <= bmi < 24:
12    print('你的身材很棒!')
13else:
14    print('你的身材不够标准哟!')

如果要给出更为准确的提示信息,我们可以再次修改上面的代码,通过elif关键字为上面的分支结构增加更多的分支,如下所示。

1"""
2BMI计算器
3
4Version: 1.2
5Author: 骆昊
6"""
7height = float(input('身高(cm):'))
8weight = float(input('体重(kg):'))
9bmi = weight / (height / 100) ** 2
10print(f'{bmi = :.1f}')
11if bmi < 18.5:
12    print('你的体重过轻!')
13elif bmi < 24:
14    print('你的身材很棒!')
15elif bmi < 27:
16    print('你的体重过重!')
17elif bmi < 30:
18    print('你已轻度肥胖!')
19elif bmi < 35:
20    print('你已中度肥胖!')
21else:
22    print('你已重度肥胖!')

我们再用刚才的三组数据来测试下上面的代码,看看会得到怎样的结果。

第一组输入:

身高(cm):175 体重(kg):68 bmi = 22.2 你的身材很棒!

第二组输入:

身高(cm):175 体重(kg):95 bmi = 31.0 你已中度肥胖!

第三组输入:

身高(cm):175 体重(kg):50 bmi = 16.3 你的体重过轻!

使用math和case构造分支结构

Python 3.10 中增加了一种新的构造分支结构的方式,通过使用matchcase 关键字,我们可以轻松的构造出多分支结构。Python 的官方文档在介绍这个新语法时,举了一个 HTTP 响应状态码识别的例子,非常有意思。如果不知道什么是 HTTP 响应状态吗,可以看看 MDN 上面的文档 。下面我们对官方文档上的示例稍作修改,为大家讲解这个语法,先看看下面用if-else结构实现的代码。

1status_code = int(input('响应状态码: '))
2if status_code == 400:
3    description = 'Bad Request'
4elif status_code == 401:
5    description = 'Unauthorized'
6elif status_code == 403:
7    description = 'Forbidden'
8elif status_code == 404:
9    description = 'Not Found'
10elif status_code == 405:
11    description = 'Method Not Allowed'
12else:
13    description = 'Unknown status Code'
14print('状态码描述:', description)

运行结果:

响应状态码: 403 状态码描述: Forbidden

下面是使用match-case语法实现的代码,虽然作用完全相同,但是代码显得更加简单优雅。

1status_code = int(input('响应状态码: '))
2match status_code:
3    case 400: description = 'Bad Request'
4    case 401: description = 'Unauthorized'
5    case 403: description = 'Forbidden'
6    case 404: description = 'Not Found'
7    case 405: description = 'Method Not Allowed'
8    case _: description = 'Unknown Status Code'
9print('状态码描述:', description)

说明:带有_case语句在代码中起到通配符的作用,如果前面的分支都没有匹配上,代码就会来到case _case _ 的使用是可选的,并非每种分支结构都要给出通配符选项。如果分支中出现了case _,它只能放在分支结构的最后面,如果它的后面还有其他的分支,那么这些分支将是不可达的。

当然,match-case 语法还有很多高级玩法,我们等用到时候再为大家讲解,有一个合并模式可以在这里分享给大家。例如,我们要将响应状态码401403404归入一个分支,400405归入到一个分支,其他保持不变,代码还可以这么写。

1status_code = int(input('响应状态码: '))
2match status_code:
3    case 400 | 405: description = 'Invalid Request'
4    case 401 | 403 | 404: description = 'Not Allowed'
5    case _: description = 'Unknown Status Code'
6print('状态码描述:', description)

运行结果:

响应状态码: 403 状态码描述: Not Allowed

分支结构应用举例

例子1:分段函数求值

有如下所示的分段函数,要求输入x,计算出y。 $$ y = \begin{cases} 3x - 5, & (x \gt 1) \ x + 2, & (-1 \le x \le 1) \ 5x + 3, & (x \lt -1) \end{cases} $$

1"""
2分段函数求值
3
4Version: 1.0
5Author: 骆昊
6"""
7x = float(input('x = '))
8if x > 1:
9    y = 3 * x - 5
10elif x >= -1:
11    y = x + 2
12else:
13    y = 5 * x + 3
14print(f'{y = }')

根据实际开发的需要,分支结构是可以嵌套的,也就是说在分支结构的ifelifelse代码块中还可以再次引入分支结构。例如if 条件成立表示玩家过关,但过关以后还要根据你获得宝物或者道具的数量对你的表现给出评价(比如点亮一颗、两颗或三颗星星),那么我们就需要在if 的内部再构造一个新的分支结构。同理,我们在elifelse中也可以构造新的分支,我们称之为嵌套的分支结构。按照这样的思路,上面的分段函数求值也可以用下面的代码来实现。

1"""
2分段函数求值
3
4Version: 1.1
5Author: 骆昊
6"""
7x = float(input('x = '))
8if x > 1:
9    y = 3 * x - 5
10else:
11    if x >= -1:
12        y = x + 2
13    else:
14        y = 5 * x + 3
15print(f'{y = }')

说明:大家可以自己感受和评判一下上面两种写法哪一种更好。在Python之禅 中有这么一句话:“Flat is better than nested”。之所以认为“扁平化”的代码更好,是因为代码嵌套的层次如果很多,会严重的影响代码的可读性。所以,我认为上面的代码第一种写法是更好的选择。

例子2:百分制成绩转换为等级制成绩

要求:如果输入的成绩在90分以上(含90分),则输出A;输入的成绩在80分到90分之间(不含90分),则输出B ;输入的成绩在70分到80分之间(不含80分),则输出C;输入的成绩在60分到70分之间(不含70分),则输出D ;输入的成绩在60分以下,则输出E

1"""
2百分制成绩转换为等级制成绩
3
4Version: 1.0
5Author: 骆昊
6"""
7score = float(input('请输入成绩: '))
8if score >= 90:
9    grade = 'A'
10elif score >= 80:
11    grade = 'B'
12elif score >= 70:
13    grade = 'C'
14elif score >= 60:
15    grade = 'D'
16else:
17    grade = 'E'
18print(f'{grade = }')

例子3:计算三角形的周长和面积。

要求:输入三条边的长度,如果能构成三角形就计算周长和面积;否则给出“不能构成三角形”的提示。

1"""
2计算三角形的周长和面积
3
4Version: 1.0
5Author: 骆昊
6"""
7a = float(input('a = '))
8b = float(input('b = '))
9c = float(input('c = '))
10if a + b > c and a + c > b and b + c > a:
11    perimeter = a + b + c
12    print(f'周长: {perimeter}')
13    s = perimeter / 2
14    area = (s * (s - a) * (s - b) * (s - c)) ** 0.5
15    print(f'面积: {area}')
16else:
17    print('不能构成三角形')

说明: 上面的if 条件表示任意两边之和大于第三边,这是构成三角形的必要条件。当这个条件成立时,我们要计算并输出周长和面积,所以if 下方有五条语句都保持了相同的缩进,它们是一个整体,只要if

条件成立,它们都会被执行,这就是我们之前提到的代码块的概念。另外,上面计算三角形面积的公式叫做海伦公式,假设有一个三角形,边长分别为$\small{a}$、$\small{b}$、$\small{c}$,那么三角的面积$\small{A}$可以由下面的公式得到,其中,$\small{s=\frac{a+b+c}{2}}$。

$$ A = \sqrt{s(s-a)(s-b)(s-c)} $$

总结

学会了 Python 中的分支结构和循环结构,我们就可以解决很多实际的问题了。这一节课相信已经帮助大家掌握了构造分支结构的方法,下一节课我们为大家介绍循环结构,学完这两次课你一定会发现,你能写出很多很有意思的代码,继续加油吧!