Python进阶小技巧3：跳出多层循环

Python进阶小技巧3：跳出多层循环

以前学python的时候一度对如何跳出多层循环比较困惑，总觉得所有方法都不够pythonic。今天突然想起这茬，就觉得写篇文章讨论一下这个问题。

1. 你知道python也有goto吗？

1.1 官方的goto

先来看看python中的goto是怎么写的

from goto import goto, label
for i in range(9):
    for j in range(9):
        for k in range(9):
            print("I am trapped, please rescue!")
            if k == 2:
                goto .breakout # 从多重循环中跳出
label .breakout
print("Freedom!")

看完代码，不要急着复制出来运行，也不要急着pip install goto。首先这段代码在你的环境里肯定是跑不起来的，不管是python2.7还是python3，都不行，因为它属于古老的python2.3，而且即使在古老的2.3中，它也是作为一个愚人节玩笑存在的，并不是真的让你在生产环境里用goto的。

既然代码都贴了，顺便贴一下运行结果吧

I'm trapped, please rescue!
I'm trapped, please rescue!
Freedom!

1.2 非官方的goto

实在非要用goto的话，就看这里https://github.com/snoack/python-goto。这个项目的作者也表示了There is another module that has been released as April Fool's joke in 2004.

goto这个东西吧，就像泡面，偶尔来上一碗，香气扑鼻，但要你天天吃，那就恶心了。很多人都会教你千万别用goto，但依我浅见，这个东西偶尔用一下还是挺香的。

2. 为什么不试试return呢？

上面聊了半天goto，害得我忘记了这篇文章是要讲跳出多层循环的，回到主题。通常情况下，我会选择将多层循环封装成一个函数，然后在需要跳出循环的地方return。照这个道理把上面那段代码重构一下：

def foo():
    def sub_foo():
        for i in range(9):
            for j in range(9):
                for k in range(9):
                    print("I am trapped, please rescue!")
                    if k == 2:
                        return
    sub_foo()
    print("Freedom!")


foo()

虽然这么多缩进看起来有点恶心，但我觉得这是跳出多层循环最pythonic的写法了。

与return较为类似的，就是使用try-except，需要跳出的地方raise一下。但在我心目中raise是用来抛出异常的，用它来跳出循环的话，那就和科比的投篮选择一样，不合理。但是python官方文档在解释为什么不提供goto的时候，就抬出了使用raise来替代的方法。所以谈不上对错吧，用不用就看个人代码风格了。

3. for循环也能用else

合理使用else，也能顺利的跳出多层循环，但是代码可读性有点闹心，look:

for i in range(9):
    for j in range(9):
        for k in range(9):
            print("I am trapped, please rescue!")
            if k == 2:
                break
        else:
            continue
        break
    else:
        continue
    break
print("Freedom!")

是不是看着有点闹心，这种代码的可读性是比较差的，别说其他人了，你写完隔一个月自己都搞不清这段逻辑。

4. 立flag

答应我，别立flag，好吗。

在flag和goto之间二选一来跳出循环，我会选择goto。

Q：如何维护使用大量flag的代码？
A：加入更多的flag。

不知道是不是深受谭浩强影响的缘故，好多恶臭的代码都会充满一堆不明所以的flag，而且非要把它命名成flag吗？flag1，flag2，到flag正无穷，求求你当个人吧，别这么写了。

我觉得这样的flag尤其恶劣:

if k == 2:
    flag = True

不行给它换个名字呗

if k == 2:
    is_k_eq_2 = True

虽然这样有点长有点搓，但起码一看就知道这个标志用于记录k是否等于2。等等，我怎么开始写代码整洁之道了……