Python：如何在while循环条件中同时检查并加载下一个元素？

1. 支持赋值表达式的语言（比如Python 3.8+） #

如果你用的是Python 3.8及以上版本，刚好可以用海象运算符（:=）实现你想要的效果，写法如下：

while b := a.get_next():
    fn(b)

这个运算符的作用是：先执行a.get_next()把结果赋值给b，再判断b的布尔值是否为真——如果a.get_next()在没有元素时返回None、空值这类“假值”，循环就会自动终止。

对比你原来的写法while a.is_next(): fn(a.get_next())，这种方式少了一次is_next()的调用，逻辑上更紧凑，也避免了is_next()和get_next()可能出现的状态不一致问题（比如某些迭代器中这两个方法的状态不同步）。

2. 不支持赋值表达式的语言（比如Python 3.7及以下、Java、C++等） #

如果你的语言不支持在循环条件里直接赋值，也可以用「无限循环+条件break」的方式实现同样的逻辑，本质上还是“先加载再检查”：

Python 3.7及以下示例： #

while True:
    b = a.get_next()
    if not b:
        break
    fn(b)

Java示例： #

while (true) {
    Object b = a.getNext();
    if (b == null) {
        break;
    }
    fn(b);
}

这种写法和你想要的逻辑完全一致，只是把赋值和判断拆成了两步而已。

注意事项 #

不管用哪种写法，都要确保a.get_next()的行为符合预期：

• 当还有元素时，返回有效的元素（布尔值为真）；
• 当没有元素时，返回一个布尔值为假的结果（比如None、null、空对象等），否则循环会陷入死循环或者跳过判断。

内容的提问来源于stack exchange，提问作者zephyrus