上次迭代后关闭范围的 Pythonic 方式

Question

我有一个代码样式问题，我正在寻找我在下面写的内容的 Pythonic 实现。

我发布的（简化的）代码遍历一个序列并返回范围。每个范围都以特定条件开始和结束。范围不能重叠。我正在使用变量active 来跟踪是否已经找到了范围的开始。如果在序列的末尾未关闭活动范围，则应添加（以input_length 作为结束索引）

下面的代码按预期工作，但我不喜欢我必须编写代码将范围附加到结果列表两次的事实。在我遇到的真正问题中，这个块要长得多，我不想在 for 循环之后再写一次。

您有什么建议可以改进吗？

谢谢！

input_length = 100
results = []

active = False
start = None
for i in range(input_length):
    condition = i % 9 == 0
    if not active and condition:
        active = True
        start = i

    condition2 = i % 13 == 0
    if active and condition2:
        active = False
        # do some additional calculations...
        results.append((start, i))

if active:
    # do some additional calculations...
    results.append((start, input_length))

print(active)
print(results)

Answer 1

我找到了一个很好的方法：

import itertools as it

il = 100
results = []
def fun1():
    active = False
    start = None
    for i in range(il):
        condition = i % 9 == 0
        if not active and condition:
            active = True
            start = i
        condition2 = i % 13 == 0
        if active and condition2:
            active = False
            # do some additional calculations...
            results.append((start, i))
    if active:
        # do some additional calculations...
        results.append((start, il))
    return results

def fun2():
    a=b=0
    while b<il:
        yield (a,b)
        b=b+13
        a=a+9
        while a<=b-13:
            a=a+9
    if a<il:
        yield (a,il)

print fun1()
print
print [(a,b) for (a,b) in fun2()]

请检查代码中不同的 il 值，并在使用前将其与您的函数进行比较。

Answer 2

简单的方法是编辑内部条件：

    condition2 = i % 13 == 0
    if active and (condition2 or i == input_length-1))
        active = False
        # do some additional calculations...
        results.append((start, i if condition2 else i + 1))

并移除外层。

如果你想避免i if condition2 else i + 1，也许你可以迭代range(input_length + 1)？ （这取决于您在循环中执行的其他操作）