Python中断多重循环的几种思路
程序员文章站
2022-07-16 23:43:01
I. 跳出单循环 不管是什么编程语言,都有可能会有跳出循环的需求,比如枚举时,找到一个满足条件的数就终止。跳出单循环是很简单的,比如 然而,我们有时候会需要跳出多重循环,而break只能够跳出一层循环,比如 这样的代码并非说找到一组i+j 5就停止,而是连续找到10组,因为break只跳出了for ......
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,例如
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