python列表前几个_如何从Python中的生成器或列表中获取前N个项目？

根据我的喜好，将'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)