Python中如何优雅地使用switch语句
我们知道Python中没有类似C++或者Java中的switch...case
语句,我们可以使用多个if...elif...else
进行模拟,但是这样的写法让代码看起来很凌乱,个人不是很推荐在代码中大量使用if
语句。
那么解决的办法是什么呢?答曰:字典(dict
)。下面我们以两个典型案例进行说明。
案例一(简单情况)
第一种简单情况就是一对一,给定一个值,返回一个值,这是C++和Java中的switch
语句支持的情况。
下面的案例是将英文日期翻译为中文日期:
dates = {
'Sun': '星期天', 'Mon': '星期一', 'Tues': '星期二', 'Wed': '星期三',
'Thurs': '星期四', 'Fri': '星期五', 'Sat': '星期六'}
day = dates.get('Fri', '未知')
print(day) # 输出结果为星期五
案例二(带条件判断)
第二种情况是多对一,反映在编程上就是case
语句中带条件判断,这个是诸如Scala中的switch
和Kotlin中的when
支持的情况。
下面给出的案例是给定一个数字,如果该数字在某个范围之类,则返回一个指定的数字。
# 这里的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
语句的使用了,让我们的代码更加干净一些!