python - 如何在 python 中使用 while(i = getNext()) 模式

Question

在 C++ 中，我经常使用以下模式:

while (int i = getNextElementPlease()) {
    printf("%d\n", i);
}

但是在 python 中我必须重复调用两次:

i = getNextElementPlease()
while (i):
    print(i)
    i = getNextElementPlease()

如何避免 getNextElementPlease 重复？

Answer 1

使用经典的 while True/break 模式:

while True:
    i = getNextElementPlease()
    if not i:
        break
    ...

或使用 two-argument form of iter从您的函数构建迭代器:

for i in iter(getNextElementPlease, sentinel):
    ...

其中 sentinel 是您期望终止循环的任何值。可能类似于 None。迭代器将调用 getNextElementPlease 获取元素并在 getNextElementPlease 返回标记值时停止。