这应该可以

top5 = array[:5] 

import itertools

top5 = itertools.islice(array, 5)

这应该可以

top5 = 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))

在我看来，将zip()与xrange(n)(或Python3中的range(n))结合使用也非常简洁，这在生成器上也很好，而且对于一般的更改似乎更灵活。

# Option #1: taking the first n elements as a list
[x for _, x in zip(xrange(n), generator)]

# Option #2, using 'next()' and taking care for 'StopIteration'
[next(generator) for _ in xrange(n)]

# Option #3: taking the first n elements as a new generator
(x for _, x in zip(xrange(n), generator))

# Option #4: yielding them by simply preparing a function
# (but take care for 'StopIteration')
def top_n(n, generator):
    for _ in xrange(n):
        yield next(generator)