1. 程式人生 > 程式設計 >Python中斷多重迴圈的幾種方式詳解

Python中斷多重迴圈的幾種方式詳解

這篇文章主要介紹了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這一句就會停止,因此可以利用這一特性,將功能寫成函式,終止多重迴圈,例如

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

以上就是本文的全部內容,希望對大家的學習有所幫助,也希望大家多多支援我們。