python - 从列表列表生成迭代器列表

Question

我有一个列表列表，我的目标是创建与这些列表的切片对应的迭代器列表。

我最初的尝试如下(这是没有大部分不相关逻辑的最小“工作”示例):

my_lists = [1,2,3], [6,7,8]

iterators = []
for my_list in my_lists:
    it = (my_list[i] for i in range(1,3))
    iterators.append(it)

print(next(iterators[0]))

不幸的是，这并没有像预期的那样工作，因为迭代器的范围是共享的，因此 my_list 变量在所有迭代器中都是相同的。这意味着打印的消息是 6 而不是所需的 1。

我很难想出一些干净的解决方案。我所有的尝试都失败了。他们要么创建不必要的列表，例如:

    it = iter(my_list[1:3])

和

    it = iter([my_list[i] for i in range(1,3)])

不必要地困惑:

def apply_range(l, start, stop):
    return (l[i] for i in range(start, stop))
...
    it = apply_range(my_list, 1, 3)

或完全 hack-ish:

    it = (lambda l: (l[i] for i in range(1,3)))(my_list)

或从列表的开头开始迭代:

from itertools import islice
...
    it = islice(my_list, 1, 3)

请注意，我需要的代码比第一个代码片段要复杂一些(即计算范围的开始和结束——即不是常量——我需要的不仅仅是第一个迭代器的第一个元素——特别是我将它们组合在一个具有内部逻辑的迭代器中，以选择正确的迭代器以从中选择下一个元素)。

Answer 1

原始代码的问题是生成器表达式需要使用外部范围的 my_list 变量。因为该变量随着循环的继续运行而改变，所以所有迭代器最终都会从输入中的最后一个列表中产生值。

您可以通过将生成器的创建封装在一个函数中来解决这个问题。该函数将拥有自己的命名空间，其中要迭代的特定列表将绑定(bind)到局部变量(不会根据外部代码更改)。该函数可以是生成器函数，也可以是返回生成器表达式的普通函数(这两种方法几乎相同)。

def make_iter(lst, start, stop):
    # this loop could be replaced by `return (lst[i] for i in range(start, stop))`
    for i in range(start, stop):
        yield lst[i]

my_lists = [1,2,3], [6,7,8]

start = 1
stop = 3
iterators = [make_iter(lst, start, stop) for lst in my_lists]