我有一个列表列表,我的目标是创建与这些列表的切片对应的迭代器列表。
我最初的尝试如下(这是没有大部分不相关逻辑的最小“工作”示例):
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)
请注意,我需要的代码比第一个代码片段要复杂一些(即计算范围的开始和结束——即不是常量——我需要的不仅仅是第一个迭代器的第一个元素——特别是我将它们组合在一个具有内部逻辑的迭代器中,以选择正确的迭代器以从中选择下一个元素)。
请您参考如下方法:
原始代码的问题是生成器表达式需要使用外部范围的 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]
