跳出嵌套循环

Question

有没有比抛出异常更简单的方法来打破嵌套循环？ （在 Perl 中，您可以为每个循环提供标签，并至少继续一个外循环。）

for x in range(10):
    for y in range(10):
        print x*y
        if x*y > 50:
            "break both loops"

即，有没有比以下更好的方法：

class BreakIt(Exception): pass

try:
    for x in range(10):
        for y in range(10):
            print x*y
            if x*y > 50:
                raise BreakIt
except BreakIt:
    pass

Answer 1

for x in xrange(10):
    for y in xrange(10):
        print x*y
        if x*y > 50:
            break
    else:
        continue  # only executed if the inner loop did NOT break
    break  # only executed if the inner loop DID break

这同样适用于更深的循环：

for x in xrange(10):
    for y in xrange(10):
        for z in xrange(10):
            print x,y,z
            if x*y*z == 30:
                break
        else:
            continue
        break
    else:
        continue
    break

Answer 2

它至少被建议过，但也被拒绝。 我认为除了重复测试或重新组织代码之外，没有其他方法。 有时有点烦人。

在拒绝消息中，van Rossum 先生提到使用 < code>return，这非常明智，也是我个人需要记住的。 :)

Answer 3

如果您能够将循环代码提取到函数中，则可以使用 return 语句随时退出最外层循环。

def foo():
    for x in range(10):
        for y in range(10):
            print(x*y)
            if x*y > 50:
                return
foo()

如果很难提取该函数，您可以使用内部函数，如 @bjd2385 建议的那样，例如

def your_outer_func():
    ...
    def inner_func():
        for x in range(10):
            for y in range(10):
                print(x*y)
                if x*y > 50:
                    return
    inner_func()
    ...

Answer 4

使用 itertools.product！

from itertools import product
for x, y in product(range(10), range(10)):
    #do whatever you want
    break

以下是 python 文档中 itertools.product 的链接：
http://docs.python.org/library/itertools.html#itertools.product

您还可以循环遍历包含 2 个 for 的数组理解，并在需要时随时中断。

>>> [(x, y) for y in ['y1', 'y2'] for x in ['x1', 'x2']]
[
    ('x1', 'y1'), ('x2', 'y1'),
    ('x1', 'y2'), ('x2', 'y2')
]

Answer 5

有时我使用布尔变量。 天真，如果你愿意的话，但我发现它非常灵活且读起来很舒服。 测试变量可以避免再次测试复杂的条件，并且还可以收集内部循环中的多次测试的结果。

    x_loop_must_break = False
    for x in range(10):
        for y in range(10):
            print x*y
            if x*y > 50:
                x_loop_must_break = True
                break
        if x_loop_must_break: break

Answer 6

如果您要引发异常，则可以引发 StopIteration 异常。 这至少会让意图变得明显。

Answer 7

您还可以重构代码以使用生成器。 但这可能不是适用于所有类型的嵌套循环的解决方案。

Answer 8

在这种特殊情况下，您可以使用 itertools.product 将循环与现代 python（3.0，也可能是 2.6）合并。

我自己认为这是一条经验法则，如果您嵌套太多循环（例如，超过 2 个），您通常可以将其中一个循环提取到另一种方法中，或者将循环合并为一个，如下所示案件。