Python
March 7, 2019

Python中如何优雅地使用switch语句

版权声明：本文为博主原创文章，转载请注明原文出处！

写作时间：2019-03-07 13:49:45

Python中如何优雅地使用switch语句

我们知道Python中没有类似C++或者Java中的switch...case语句，我们可以使用多个if...elif...else进行模拟，但是这样的写法让代码看起来很凌乱，个人不是很推荐在代码中大量使用if语句。

那么解决的办法是什么呢？答曰：字典（dict）。下面我们以两个典型案例进行说明。

案例一（简单情况）

第一种简单情况就是一对一，给定一个值，返回一个值，这是C++和Java中的switch语句支持的情况。

下面的案例是将英文日期翻译为中文日期：

1
2
3
4
5
6
dates = {
    'Sun': '星期天', 'Mon': '星期一', 'Tues': '星期二', 'Wed': '星期三',
    'Thurs': '星期四', 'Fri': '星期五', 'Sat': '星期六'}

day = dates.get('Fri', '未知')
print(day)  # 输出结果为星期五

案例二（带条件判断）

第二种情况是多对一，反映在编程上就是case语句中带条件判断，这个是诸如Scala中的switch和Kotlin中的when支持的情况。

下面给出的案例是给定一个数字，如果该数字在某个范围之类，则返回一个指定的数字。

1
2
3
4
5
6
7
8
9
# 这里的conditions是一个函数
conditions = lambda x: {
    x < -1: 0, -1 <= x <= 1: 0.5, x > 1: 1
}

num = conditions(0.25)[True]
print(num)
num = conditions(10)[True]
print(num)

这里我们的dict不是一个普通的字典，其key是一个lambda表达式（一个函数）。如果我们调用该函数，则会返回一个字典，该字典中有两个元素：一个元素的键是True，另一个是False。True元素包含的值是对应lambda函数中满足条件的给定值，False元素包含的值是对应lambda函数中最后一个不满足条件的给定值（这句话写得比较拗口，不好理解。动手实践一下，可以加深理解）。

经过上面的介绍，我们以后可以大大减少对if...else语句的使用了，让我们的代码更加干净一些！