Python中斷多重循環的幾種思路
阿新 • • 發佈:2017-08-05
pre except keyword ctu cep from 集合 rod code
常見方法
II. 跳出多重循環
事實上,Python的標準語法是不支持跳出多重循環的,所以只能利用一些技巧,大概的思路有:寫成函數、利用笛卡爾積、利用調試。
寫成函數
在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中斷多重循環的幾種思路