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)這一重循環。那麼,要怎麼跳出多重呢?在此記錄備忘一下。
相關推薦:《Python教程》
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
以上是盤點Python中斷多重循環的思路的詳細內容。更多資訊請關注PHP中文網其他相關文章!