如何从生成器或列表中取出前 N 个项目?[重复]

2025-02-05 13:24:00
admin
原创
96
摘要:问题描述:和林克我会var top5 = array.Take(5); 如何用 Python 做到这一点?解决方案 1:列表切片top5 = array[:5] 要对列表进行切片,有一个简单的语法:array[start:stop:step]您可以省略任何参数。以下均有效:array[start:]、arra...

问题描述:

和林克我会

var top5 = array.Take(5);

如何用 Python 做到这一点?


解决方案 1:

列表切片

top5 = array[:5]
  • 要对列表进行切片,有一个简单的语法:array[start:stop:step]

  • 您可以省略任何参数。以下均有效:array[start:]array[:stop]array[::step]

切片生成器

import itertools
top5 = itertools.islice(my_list, 5) # grab the first five elements
  • 你不能在 Python 中直接对生成器进行切片。itertools.islice()将使用以下语法将对象包装在新的切片生成器中itertools.islice(generator, start, stop, step)

  • 请记住,对生成器进行切片会使其部分耗尽。如果要保持整个生成器完好无损,可以先将其转换为元组或列表,例如:result = tuple(generator)

解决方案 2:

import itertools

top5 = itertools.islice(array, 5)

解决方案 3:

@Shaikovsky 的回答非常好,但我想澄清几点。

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

这是最简单的方法,但StopIteration如果发电机过早耗尽,就会抛出异常。


另一方面,以下方法返回最多可取的 n项目,这在许多情况下是更好的选择:

列表:
[x for _, x in zip(range(n), records)]

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

解决方案 4:

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)

解决方案 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()而不是使得推导式在抛出的异常[]上终止。StopIteration`next()`

解决方案 6:

你是指N 个项目,还是最大的N 个项目?

如果你想要第一个:

top5 = sequence[:5]

假设您的序列按降序排序,这也适用于最大的 N 个项目。 (您的 LINQ 示例似乎也假设了这一点。)

如果您想要最大的,并且它没有排序,最明显的解决方案是先对其进行排序:

l = list(sequence)
l.sort(reverse=True)
top5 = l[:5]

为了获得更高性能的解决方案,请使用最小堆(感谢 Thijs):

import heapq
top5 = heapq.nlargest(5, sequence)

解决方案 7:

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

使用列表理解:

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

否则:

first_n_elements = list(generator)[:n]

其中n是要获取的元素数量(例如,前五个元素为 n=5)。

解决方案 8:

这应该有效

top5 = array[:5] 
相关推荐
  政府信创国产化的10大政策解读一、信创国产化的背景与意义信创国产化,即信息技术应用创新国产化,是当前中国信息技术领域的一个重要发展方向。其核心在于通过自主研发和创新,实现信息技术应用的自主可控,减少对外部技术的依赖,并规避潜在的技术制裁和风险。随着全球信息技术竞争的加剧,以及某些国家对中国在科技领域的打压,信创国产化显...
工程项目管理   1565  
  为什么项目管理通常仍然耗时且低效?您是否还在反复更新电子表格、淹没在便利贴中并参加每周更新会议?这确实是耗费时间和精力。借助软件工具的帮助,您可以一目了然地全面了解您的项目。如今,国内外有足够多优秀的项目管理软件可以帮助您掌控每个项目。什么是项目管理软件?项目管理软件是广泛行业用于项目规划、资源分配和调度的软件。它使项...
项目管理软件   1354  
  信创国产芯片作为信息技术创新的核心领域,对于推动国家自主可控生态建设具有至关重要的意义。在全球科技竞争日益激烈的背景下,实现信息技术的自主可控,摆脱对国外技术的依赖,已成为保障国家信息安全和产业可持续发展的关键。国产芯片作为信创产业的基石,其发展水平直接影响着整个信创生态的构建与完善。通过不断提升国产芯片的技术实力、产...
国产信创系统   21  
  信创生态建设旨在实现信息技术领域的自主创新和安全可控,涵盖了从硬件到软件的全产业链。随着数字化转型的加速,信创生态建设的重要性日益凸显,它不仅关乎国家的信息安全,更是推动产业升级和经济高质量发展的关键力量。然而,在推进信创生态建设的过程中,面临着诸多复杂且严峻的挑战,需要深入剖析并寻找切实可行的解决方案。技术创新难题技...
信创操作系统   27  
  信创产业作为国家信息技术创新发展的重要领域,对于保障国家信息安全、推动产业升级具有关键意义。而国产芯片作为信创产业的核心基石,其研发进展备受关注。在信创国产芯片的研发征程中,面临着诸多复杂且艰巨的难点,这些难点犹如一道道关卡,阻碍着国产芯片的快速发展。然而,科研人员和相关企业并未退缩,积极探索并提出了一系列切实可行的解...
国产化替代产品目录   28  
热门文章
项目管理软件有哪些?
云禅道AD
禅道项目管理软件

云端的项目管理软件

尊享禅道项目软件收费版功能

无需维护,随时随地协同办公

内置subversion和git源码管理

每天备份,随时转为私有部署

免费试用