Python中斷多重循環的幾種思路

轉自： https://www.cnblogs.com/cuijianxin/p/6611751.html

I. 跳出單循環
不管是什么編程語言，都有可能會有跳出循環的需求，比如枚舉時，找到一個滿足條件的數就終止。跳出單循環是很簡單的，比如

for i in range(10):
    if i > 5:
        print i
        break

然而，我們有時候會需要跳出多重循環，而break只能夠跳出一層循環，比如

for i in range(10):
    for j in range(10):
        if i+j > 5:
            print i,j
            break

這樣的代碼並非說找到一組i+j > 5就停止，而是連續找到10組，因為break只跳出了for j in range(10)這一重循環。那么，怎么才能跳出多重呢？在此記錄備忘一下。

II. 跳出多重循環
事實上，Python的標准語法是不支持跳出多重循環的，所以只能利用一些技巧，大概的思路有：寫成函數、利用笛卡爾積、利用調試。

當然最常用的思路是使用變量標記法

def f():
    flag = 0
    for i in range(10):
        for j in range(i):
            if i+j>5:
                print i,j
                flag = 1
                break
        if flag == 1:
            break

if __name__ == "__main__":
    f()

寫成函數
在Python中，函數運行到return這一句就會停止，因此可以利用這一特性，將功能寫成函數，終止多重循環，例如

def work():
    for i in range(10):
        for j in range(10):
            if i+j > 5:
                return i,j

print work()

利用笛卡爾積
這種方法的思路就是，既然可以跳出單循環，我就將多重循環改寫為單循環，這可以利用itertools中的笛卡爾積函數product，例如

from itertools import product

for i,j in product(range(10), range(10)):
    if i+j > 5:
        print i,j
        break

利用調試模式
笛卡爾積的方式很巧妙，也很簡潔，但它只能用於每次循環的集合都是獨立的情形，假如每層循環都與前一層緊密相關，就不能用這種技巧了。這時候可以用第一種方法，將它寫成函數，另外，還可以利用調試模式。這個利用了調試模式中，只要出現報錯就退出的原理，它偽裝了一個錯誤出來。

class Found(Exception):
    pass

try:
    for i in range(10):
        for j in range(i): #第二重循環跟第一重有關
            if i + j > 5:
                raise Found
except Found:
    print i, j