编写干净的 Pythonic 代码就是尽可能使其既简洁又易于理解。所以当你学习编程到一定阶段之后,重构代码是很有必要的。
但不少经验尚不丰富的学习者对于自己写的冗长代码,想重构但又无思路。今天我们就来分享几个python代码重构优化的小技巧,助你一臂之力。
本文重点是想跟大家探讨为什么这些重构是好想法,而不仅仅是如何做。
- 合并嵌套的 if 条件
太多的嵌套会使代码难以理解,这在 Python 中尤为如此,因为 Python 没有括号来帮助区隔不同的嵌套级别。
阅读深度嵌套的代码容易让人烦躁,因为你必须理清哪些条件属于哪一级。因此,我们应尽可能减少嵌套,如果两个条件可以用 and 合在一起,会比嵌套更易读。
合并之前:
合并后:
- 将重复的代码移到条件语句之外
我们应该始终寻找移除重复代码的方法。这是提升代码能力的好办法。
有时,在条件的两个分支上出现重复代码,这意味这段代码将始终执行。因此这段重复的代码可以从条件中移出,放在条件之外执行。
通过将对label变量赋值移到到条件之外,我们删除了重复的代码行,并明确了条件实际控制的内容,就是计算total。
- 将内部循环中的 yield 替换为 yield from
经常忽略的一个小窍门是 Python 的yield关键字有对应的为collections准备的yield from。因此无需使用 for 循环遍历集合。这使代码变短,并删除 for 中的额外变量。而且消除 for 循环后,yield from使程序运行效率提高约 15%。
重构前:
重构后: