我已经被这个数据结构困住了一段时间了:
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
我正在努力“翻译”代码。
请您参考如下方法:
如果数据嵌套两层深,我们可以使用chain函数将可迭代对象连接在一起,然后让sum(..) 计算迭代的总和。所以:
<b>from itertools import chain</b>
sum(<b>chain.from_iterable(</b>outer_iter<b>)</b>)
chain.from_iterable 将可迭代对象的可迭代对象作为输入,并将其转换为一个可迭代对象,一次从可迭代对象中延迟获取元素。我们可以在 chain 上使用 iterable 解包,但如果外部 iterable 是一个无限列表,算法就会卡住(并最终耗尽内存)。
