从Python中的生成器获取多个单个值

Question

如何从迭代器的不同索引位置获取多个任意值？

如何在列表（python）中 获取 生成器的n个下一个值以及如何 在Python中获取生成器的n个项，描述了itertools.islice用于从迭代器获取任意元素或连续子集的用法。 但是，如果我想要迭代器中不同位置的多个任意元素，而不能仅仅使用islice的step参数呢？

我正在尝试解决Euler项目的问题40 。 我生成了一个由串联的整数组成的字符串

iteration = (i for i in ''.join(map(str, (i for i in xrange(1,10**6)))))

现在，我想从索引1开始获得索引为islice在这里我不能使用islice ，因为每次调用next都会使当前值右移。 例如

next(islice(iteration, 1, 2)) + next(islice(iteration, 3, 4))

产生“ 26”而不是“ 24”。

更新（25.11.12，4:43 UTC + 0） ：

感谢所有的建议。 我当前的代码如下：

it = (i for i in ''.join(map(str, (i for i in xrange(1,10**6)))))
ds = [int(nth(it, 10**i-10**(i-1)-1)) for i in range(7)]
return product(ds)

nth的丑陋参数是生成0、8、89、899、8999等的序列。

Answer 1

（请注意，有更快的方法可以解决Euler＃40。）

我的工作方式会有所不同。 而不是使用nth ：

>>> from itertools import chain, count, islice
>>> 
>>> it = chain.from_iterable(str(i) for i in count(1))
>>> wanted = {10**i for i in range(7)}
>>> scan_region = islice(it, max(wanted)+1)
>>> digits = [int(x) for i, x in enumerate(scan_region, 1) if i in wanted]
>>> digits
[1, 1, 5, 3, 7, 2, 1]

这样，我不必进行任何减法操作即可确保索引正确。

Answer 2

这是来自itertools文档的“食谱”部分 。 它返回iterable第n个元素，并随其使用：

def nth(iterable, n, default=None):
    "Returns the nth item or a default value"
    return next(islice(iterable, n, None), default)

您可以通过依次调用它来获得1st，10th，100th等元素（请注意，迭代器已消耗，并且索引为零）：

first = nth(iteration, 0)
tenth = nth(iteration, 8)  # since we've already taken one
hundredth = nth(iteration, 89)  # since we've already taken ten
# etc

另外，您可以每次使用tee并使用nth和不同的迭代器。 这样，您不必担心单个迭代器被消耗的事实。 另一方面，如果迭代器很长，则可能开始吞下内存。

Answer 3

就像提到的那样查看nth我将着眼于简化生成器：

from itertools import count

def concat():
    for i in count(1):
        for j in str(i):
            yield int(j)