一旦发电机列表中的发电机耗尽，是否继续使用其他发电机？

Question

我在一个函数中有一个生成器列表alternate_all(*args)，该函数在列表中的每个生成器之间交替打印它们的第一个项目、第二个项目……等等，直到所有生成器都用完。

我的代码一直工作，直到生成器耗尽，一旦发生 StopIteration，当我希望它继续使用其余生成器并忽略耗尽的生成器时，它就会停止打印：

def alternate_all(*args):
    iter_list = []
    for iterable in args:
        iter_list.append(iter(iterable))
    try:
        while True:
            for iterable in iter_list:
                val = next(iter_list[0])
                iter_list.append(iter_list.pop(0))
                yield val
    except StopIteration:
        pass

            
if __name__ == '__main__':
    for i in alternate_all('abcde','fg','hijk'):
        print(i,end='')

我的输出是：

afhbgic

什么时候应该是：

afhbgicjdke

我怎样才能让它忽略耗尽的发电机？我不想使用 itertools 并保持相同的结构。

Answer 1

这有效。我试图接近您的原始代码的工作方式（尽管为了简单起见，我确实用列表理解替换了您的第一个循环）。

def alternate_all(*args):
    iter_list = [iter(arg) for arg in args]
    while iter_list:
        i = iter_list.pop(0)
        try:
            val = next(i)
        except StopIteration:
            pass
        else:
            yield val
            iter_list.append(i)

您的代码的主要问题是您的try/except位于循环之外，这意味着第一个耗尽的迭代器将从循环中退出。相反，您希望StopIteration在循环内部进行捕获，以便可以继续进行，并且循环应该iter_list在其中仍然有任何迭代器的情况下继续进行。