如何做到这一点的答案可以在这里找到

>>> generator = (i for i in xrange(10))
>>> list(next(generator) for _ in range(4))
[0, 1, 2, 3]
>>> list(next(generator) for _ in range(4))
[4, 5, 6, 7]
>>> list(next(generator) for _ in range(4))
[8, 9]

注意，最后一个调用要求下一个4，而此时只剩下2个。使用list()而不是[]是为了让理解在next()抛出的StopIteration异常上终止。

import itertools

top5 = itertools.islice(array, 5)

如何做到这一点的答案可以在这里找到

>>> generator = (i for i in xrange(10))
>>> list(next(generator) for _ in range(4))
[0, 1, 2, 3]
>>> list(next(generator) for _ in range(4))
[4, 5, 6, 7]
>>> list(next(generator) for _ in range(4))
[8, 9]

注意，最后一个调用要求下一个4，而此时只剩下2个。使用list()而不是[]是为了让理解在next()抛出的StopIteration异常上终止。

@Shaikovsky的回答很好，但我想澄清两点。

[next(generator) for _ in range(n)]

这是最简单的方法，但如果生成器过早耗尽，则抛出StopIteration。

另一方面，以下方法最多返回n个项，这在许多情况下是可取的:

列表: [x for _， x in zip(范围(n)，记录)]

发电机: (x for _， x in zip(range(n)， records))

这应该可以

top5 = array[:5] 

使用itertools，您将获得另一个生成器对象，因此在大多数情况下，您将需要另一个步骤来获取前n个元素。至少有两种更简单的解决方案(在性能方面效率稍低，但非常方便)可以从生成器中获取元素:

使用列表推导式:

first_n_elements = [generator.next() for i in range(n)]

否则:

first_n_elements = list(generator)[:n]

其中n是你想要取的元素数量(例如，n=5表示前五个元素)。