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

前端之家收集整理的这篇文章主要介绍了Python中断多重循环的思路总结前端之家小编觉得挺不错的,现在分享给大家,也给大家做个参考。

I. 跳出单循环

不管是什么编程语言,都有可能会有跳出循环的需求,比如枚举时,找到一个满足条件的数就终止。跳出单循环是很简单的,比如:

  1. for i in range(10):
  2.  
  3. if i > 5:
  4.  
  5. print i
  6.  
  7. break

然而,我们有时候会需要跳出多重循环,而break只能够跳出一层循环,比如:

  1. for i in range(10):
  2.  
  3. for j in range(10):
  4.  
  5. if i+j > 5:
  6.  
  7. print i,j
  8.  
  9. break

这样的代码并非说找到一组 i+j > 5就停止,而是连续找到10组,因为break只跳出了for j in range(10)这一重循环。那么,怎么才能跳出多重呢?在此记录备忘一下。

II. 跳出多重循环

事实上,Python的标准语法是不支持跳出多重循环的,所以只能利用一些技巧,大概的思路有:写成函数、利用笛卡尔积、利用调试。

当然最常用的思路是使用变量标记

  1. def f():
  2.  
  3. flag = 0
  4.  
  5. for i in range(10):
  6.  
  7. for j in range(i):
  8.  
  9. if i+j>5:
  10.  
  11. print i,j
  12.  
  13. flag = 1
  14.  
  15. break
  16.  
  17. if flag == 1:
  18.  
  19. break
  20.  
  21. if __name__ == "__main__":
  22.  
  23. f()

写成函数

在Python中,函数运行到return这一句就会停止,因此可以利用这一特性,将功能写成函数,终止多重循环。

例如:

  1. def work():
  2.  
  3. for i in range(10):
  4.  
  5. for j in range(10):
  6.  
  7. if i+j > 5:
  8.  
  9. return i,j
  10.  
  11. print work()

利用笛卡尔积

这种方法的思路就是,既然可以跳出单循环,我就将多重循环改写为单循环,这可以利用itertools中的笛卡尔积函数product,例如:

  1. from itertools import product
  2.  
  3. for i,j in product(range(10),range(10)):
  4.  
  5. if i+j > 5:
  6.  
  7. print i,j
  8.  
  9. break

利用调试模式

笛卡尔积的方式很巧妙,也很简洁,但它只能用于每次循环的集合都是独立的情形,假如每层循环都与前一层紧密相关,就不能用这种技巧了。这时候可以用第一种方法,将它写成函数,另外,还可以利用调试模式。这个利用了调试模式中,只要出现报错就退出的原理,它伪装了一个错误出来。

  1. class Found(Exception):
  2.  
  3. pass
  4.  
  5. try:
  6.  
  7. for i in range(10):
  8.  
  9. for j in range(i): #第二重循环跟第一重有关
  10.  
  11. if i + j > 5:
  12.  
  13. raise Found
  14.  
  15. except Found:
  16.  
  17. print i,j

以上就是盘点Python中断多重循环的思路的详细内容,更多请关注我们其它相关文章

猜你在找的Python相关文章