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

2025-02-05 13:24:00
admin
原创
118
摘要:问题描述:和林克我会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大政策解读一、信创国产化的背景与意义信创国产化,即信息技术应用创新国产化,是当前中国信息技术领域的一个重要发展方向。其核心在于通过自主研发和创新,实现信息技术应用的自主可控,减少对外部技术的依赖,并规避潜在的技术制裁和风险。随着全球信息技术竞争的加剧,以及某些国家对中国在科技领域的打压,信创国产化显...
工程项目管理   2974  
  为什么项目管理通常仍然耗时且低效?您是否还在反复更新电子表格、淹没在便利贴中并参加每周更新会议?这确实是耗费时间和精力。借助软件工具的帮助,您可以一目了然地全面了解您的项目。如今,国内外有足够多优秀的项目管理软件可以帮助您掌控每个项目。什么是项目管理软件?项目管理软件是广泛行业用于项目规划、资源分配和调度的软件。它使项...
项目管理软件   1836  
  PLM(产品生命周期管理)系统在企业的产品研发、生产与管理过程中扮演着至关重要的角色。然而,在实际运行中,资源冲突是经常会遇到的难题。资源冲突可能导致项目进度延迟、成本增加以及产品质量下降等一系列问题,严重影响企业的效益与竞争力。因此,如何有效应对PLM系统中的资源冲突,成为众多企业关注的焦点。接下来,我们将详细探讨5...
plm项目管理系统   47  
  敏捷项目管理与产品生命周期管理(PLM)的融合,正成为企业在复杂多变的市场环境中提升研发效率、增强竞争力的关键举措。随着技术的飞速发展和市场需求的快速更迭,传统的研发流程面临着诸多挑战,而将敏捷项目管理理念融入PLM,有望在2025年实现研发流程的深度优化,为企业创造更大的价值。理解敏捷项目管理与PLM的核心概念敏捷项...
plm项目   47  
  模块化设计在现代产品开发中扮演着至关重要的角色,它能够提升产品开发效率、降低成本、增强产品的可维护性与可扩展性。而产品生命周期管理(PLM)系统作为整合产品全生命周期信息的关键平台,对模块化设计有着强大的支持能力。随着技术的不断发展,到 2025 年,PLM 系统在支持模块化设计方面将有一系列令人瞩目的技术实践。数字化...
plm软件   48  
热门文章
项目管理软件有哪些?
曾咪二维码

扫码咨询,免费领取项目管理大礼包!

云禅道AD
禅道项目管理软件

云端的项目管理软件

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

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

内置subversion和git源码管理

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

免费试用