代码中难免存在复杂的if-else条件逻辑，简化条件表达式是一种非常实用的技巧，能够帮助我么提高代码的可读性，减少不必要的if-else嵌套结构。由于涉及的主题较广，将分为上下篇进行介绍，本篇文章和大家分享优化if-else结构的7个实用小技巧，下一篇文章将分享一些更高级的主题，比如利用面向对象中的多态机制优化if-else结构。
使用三元运算符
与标准的if-else语句相比，三元运算符更加简洁，可读性更高，适用于简单的条件判断逻辑。
before:
def get_discount_price(self, price):
if self.is_discount_good():
return price*0.8
else:
return price
after:
def get_discount_price(self, price):
return price*0.8 if self.is_discount_good() else price提前return，去掉多余的else
同样是上面的例子，最后的else其实是多余的。正常情况下，商品是不打折的，如果发现顾客买的是打折商品就打8折。
before:
def get_discount_price(self, price):
if self.is_discount_good():
return price*0.8
else:
return priceafter:
def get_discount_price(self, price):
if self.is_discount_good():
return price*0.8
return price合并重复的条件片段
before:
def get_total_price(self, price, amount):
if self.is_discount_good():
price = price*0.95
total_price = price*amount
else:
total_price = price*amount
return total_priceafter:
def get_total_price(self, price，amount):
if self.is_discount_good():
price = price*0.95
total_price = price*amount
return total_price分解条件表达式
某些时候，我们的条件判断过于复杂，很难弄清楚这个条件判断的功能是什么，导致代码的可读性大大降低。这时候我们可以像处理过长的函数的时候那样，将复杂的条件判断提炼为独立函数，从而更清楚的表达自己的意图。
before:
def get_discount_price(self, price):
if date.today().strftime("%m%d")=="0618" or date.today().strftime("%m%d")=="1111":
return price*0.8
else:
return priceafter:
def get_discount_price(self, price):
def is_special_day(date):
if date.today().strftime("%m%d")=="0618" or date.today().strftime("%m%d")=="1111":
return True
return False
if is_special_day(self.order_date):
return price*0.8
else:
return price
虽然我们的代码长度增加了，但是我们提炼了新的函数，可读性更高，也便于单元测试。
合并条件表达式
合并与分解是两个相反的动作，如果我们发现有一系列的检查条件，最终的行为却是一致的，就应该使用逻辑或与非，将他们合并为一个条件表达式，让逻辑更加清晰。
before:
def get_discount_price(self, price):
if self.tax_free_good():
return price*0.8
if self.s_discount_good():
return price*0.8
return price既然免税商品和打折商品最后计算价格的时候都是打八折，那为什么不将他们合并到一个判断逻辑中呢？
after:
def get_discount_price(self, price):
if self.tax_free_good() or self.s_discount_good():
return price*0.8
return price
当然我们也反对强行合并条件表达式的做法，如果这些条件真的是彼此独立的，例如下一节员工工资发放的判断逻辑中，这些判断条件的确是彼此独立的，因此最好还是不要合并它们。
优化逻辑结构，让正常流程走主干
条件表达式通常有两种表现形式，第一种形式是：所有分支都属于正常行为；第二种形式则是：条件表达式中只有一种是正常行为，其他都是不常见的特殊情况。这两种表达式有着不同的用途，这一点应该通过代码表现出来。
before:
def get_pay_amount(self):
if self.isdead:
result = self.dead_amount()
else:
if self.is_dismiss:
result = self.dismiss_amount()
else:
if self.is_retired:
result = self.retired_amount()
else:
result = normal_pay_amount()
return result上面计算工资的例子中，工资发放有多种情况，但是其中死亡、解雇、退休都是特殊的情况，而大多数的时候是按照正常情况发工资的，我们应该在代码中体现出来。
after:
def get_pay_amount(self):
if self.isdead:
return self.dead_amount()
if self.is_dismiss:
return self.dismiss_amount()
if self.is_retired:
return self.retired_amount()
return normal_pay_amount()
换句话说，第一种形式中，我们潜在的表达了，我们对所有的if分支和else分支的重要程度是一样的，而第二种情况中，我们表达的是，if分支中的情况非常罕见，如果发生了做一些特别的处理，而大多数的时候我们都走主干流程，如此一来，可读性就高了。
使用枚举
在某些时候，使用枚举也可以优化if-else逻辑分支，这样做的好处在于维护起来更加方便，只需要维护一个枚举字典就好了，无需更改if-else逻辑分支。
before:
def print_order_info(status_code):
if status_code=="0":
print('订单未支付')
elif status_code=="1":
print("订单已支付")
elif status_code=="2":
print("订单已发货")
elif status_code=="3":
print("订单已收货")
elif status_code=="4":
print("订单已退货")after:
status_dict = {"0":'订单未支付',
"1":"订单已支付",
"2","订单已发货",
"3":"订单已收货",
"4":"订单已退货"
}
def print_order_info(status_code):
return status_dict.get(status_code)欢迎关注我的公众号“数据科学杂谈”，原创技术文章第一时间推送。
