如何在python的while循环语句中使用迭代器

Question

是否可以在 Python 的 while 循环中使用生成器或迭代器？例如，类似：

i = iter(range(10))
while next(i):
    # your code

这样做的目的是在 while 循环语句中构建迭代，使其类似于 for 循环，不同之处在于您现在可以在 while 语句中添加逻辑：

i = iter(range(10))
while next(i) and {some other logic}:
    # your code

然后它变成了一个很好的 for 循环/while 循环混合。

有人知道怎么做吗？

Answer 1

你只需要安排你的迭代器在它过期时返回一个类似假的值。例如，如果我们反转 range 使其倒计时到 0：

>>> i = iter(range(5, -1, -1))
>>> while val := next(i):
...     print('doing something here with value', val)
...

这将导致：

doing something here with value 5
doing something here with value 4
doing something here with value 3
doing something here with value 2
doing something here with value 1

Answer 2

你可以的

    a = iter(range(10))

    try:
        a.next()
        while True and {True or False logic}:
            print("Bonjour")
            a.next()
    except StopIteration:
        print("a.next() Stop iteration")

Answer 3

while next(i):有两个问题

1. 与for 循环不同，while 循环不会捕获在没有next 值时引发的StopIteration 异常；在这种情况下，您可以使用 next(i, None) 返回一个“假”值，但是只要迭代器返回一个实际的假值，while 循环也会停止
2. next 返回的值将被消耗，循环体中不再可用。 （在 Python 3.8+ 中，这可以通过赋值表达式来解决，请参阅其他答案。）

相反，您可以使用带有itertools.takewhile 的for 循环，从可迭代对象中测试当前元素，或者只是任何其他条件。这将循环直到迭代器用尽，或者条件评估为假。

from itertools import takewhile
i = iter(range(10))
r = 0
for x in takewhile(lambda x: r < 10, i):
    print("using", x)
    r += x
print("result", r)

输出：

using 0
...
using 4
result 10

Answer 4

在 Python >= 3.8 中，您可以使用 assignment expressions 执行以下操作：

i = iter(range(10))
while (x := next(i, None)) is not None and x < 5:
    print(x)

在 Python itertools.takewhile:

from itertools import takewhile

i = iter(range(10))
for x in takewhile({some logic}, i):
    # do stuff

这里的“一些逻辑”是一个 1-arg 可调用的接收任何next(i) 产生的东西：

for x in takewhile(lambda e: 5 > e, i):
    print(x)
0
1
2
3
4

Answer 5

a = iter(range(10))

try:
    next(a)
    while True:
        print(next(a))
except StopIteration:
    print("Stop iteration")