如何在Python中WHILE循环语句中使用迭代器

问题描述

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

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

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

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

然后它成为一个很好的for循环/While循环混合。

有人知道如何执行此操作吗？

---

解决方案

在Python>；=3.8中，可以使用assignment expressions：

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

在Python<；3.8中，您可以使用itertools.takewhile：

from itertools import takewhile

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

&QOOT；一些逻辑&QOOT；此处将是一个1参数可调回函数，无论next(i)产生什么结果：

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