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

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這一句就會停止,因此可以利用這一特性,將功能寫成函數,終止多重循環,

例如

'''
遇到問題沒人解答?小編創建了一個Python學習交流QQ羣:857662006 尋找有志同道合的小夥伴,互幫互助,
羣裏還有不錯的視頻學習教程和PDF電子書!
'''
def work():
    for i in range(10):
        for j in range(10):
            if i+j > 5:
                return i,j

print work()

利用笛卡爾積

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

'''
遇到問題沒人解答?小編創建了一個Python學習交流QQ羣:857662006 尋找有志同道合的小夥伴,互幫互助,
羣裏還有不錯的視頻學習教程和PDF電子書!
'''
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
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章