Python中断多重循环的思路总结

跳出单循环的基础概念

I. 跳出单循环

在任何编程语言中，跳出循环都是常见需求，例如在枚举过程中，一旦找到满足条件的数值便立即终止。对于单层循环，实现这一目标非常简单，参考以下示例：

for i in range(10):

  if i > 5:

    print i

    break

然而，在实际开发中，我们有时需要跳出多重嵌套循环。由于Python中的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的标准语法并不直接支持跳出多重循环，因此需要借助一些技巧。主要的解决思路包括：将逻辑封装为函数、利用itertools模块中的笛卡尔积功能，或者利用调试模式中的异常机制。

当然，最常用且直观的思路是使用变量标记法。

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语句时会立即终止并返回结果。利用这一特性，我们可以将需要跳出多重循环的功能封装在一个函数中，通过return来终止整个循环过程。

例如：

def work():

  for i in range(10):

    for j in range(10):

      if i+j > 5:

        return i,j

print work()

方法二：利用itertools笛卡尔积

利用笛卡尔积

这种方法的思路是：既然可以轻松跳出单循环，那么将多重循环改写为单循环即可。这可以通过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中断多重循环的思路的详细内容，更多请关注本站其它相关文章！