Python中断多重循环的几种思路exit_flag

最新推荐文章于 2021-01-28 16:09:29 发布

weixin_34107955

最新推荐文章于 2021-01-28 16:09:29 发布

阅读量959

点赞数

文章标签： python

事实上，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

关注