Python 列表理解。通过重复调用函数直到返回预定义值来生成列表

Question

我想通过重复调用函数直到函数返回预定义值来生成列表。

myList = [i,r = foo() while i==0]

myList = [r1, r2, r3, r4...] as long as returned i value is == 0

有没有办法使用列表理解来做到这一点？

Answer 1

def stop_iteration():
    raise StopIteration

如果foo()是一个生成器:

myList = list((i,r) if i != 0 else stop_iteration() for (i,r) in foo())

其他:

def foo_generator():
    i,r = foo()
    while i != 0:
      yield i,r
      i,r = foo()

myList = list(foo_generator())