python - for(i=0;True;i++) 在 python 中？

Question

有没有一种更 Pythonic 的方式，或者至少是一种更短更简单的方式来做到这一点：

i = 1
while True:
    res = lookup(i) # returns a value or None
    if res is None:
        break
    else:
        i += 1
        yield res

Answer 1

您可以使用itertools：

from itertools import takewhile, count

# ...
def myfunc():
    return takewhile(lambda x: x is not None, (lookup(i) for i in count(1)))

如果你不喜欢takewhile任何原因：

for i in count(1):
     res = lookup(i)
     if res is None: break
     yield res

Answer 2

无需进入itertools...

i = 1
res = lookup(i)
while res is not None:
    i += 1
    yield res
    res = lookup(i)

Answer 3

itertools.count可以无限计数：

for i in itertools.count(1):
    res = lookup(i)
    if res is None: break
    yield res