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)

或直接破解：

    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变量。由于该变量会随着循环继续运行而发生变化，因此所有迭代器最终都会从输入的最后一个列表中产生值。

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

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]

Answer 2

这是一个简单的整洁代码，我认为您可以满足您进行一些迭代的需求；

my_lists = [1,2,3], [6,7,8]
iterators = []
for my_list in my_lists:
  iterators.append(iter(my_list))

允许您使用“ next”访问每个“迭代器”和值；

next(iterators[1])
>>> 6
next(iterators[1])
>>> 7

从列表列表中生成迭代器列表

2 个答案: