python - 使用 map-reduce/itertools 对嵌套迭代求和

Question

我已经被这个数据结构困扰了一段时间了:

iter([iter([1,0]),iter([1,1]),iter([0,0])])

我想使用 map-reduce/itertools 计算最内部元素的总和。

我能够使用 for 循环相当快地得到答案:

outer_iter = iter([iter([1,0]),iter([1,1]),iter([0,0])])

for inner_iter in outer_iter:
    for inner_list in inner_iter:
        total = total + inner_list

我正在努力“翻译”代码。

Answer 1

如果数据嵌套两层深度，我们可以使用chain函数将迭代连接在一起，然后让sum(..) 计算可迭代的总和。所以:

<b>from itertools import chain</b>

sum(<b>chain.from_iterable(</b>outer_iter<b>)</b>)

chain.from_iterable 将可迭代的可迭代作为输入，并将其转换为一个可迭代，每次从可迭代中延迟获取元素。我们可以在链上使用可迭代解包，但如果外部可迭代是一个无限列表，算法就会陷入困境(并最终耗尽内存)。