如何从python列表中获得n个最大的列表
我正在使用heapq从列表中获取最大的元素。 我写的程序如下。
import csv
import heapq
f = open("E:/output.csv","r")
read = csv.reader(f)
allrows = [row for row in read]
for i in xrange(0,2):
print allrows[i]
allrows.sort(key=lambda x: x[2]) #this is working properly
it=heapq.nlargest(20,enumerate(allrows),key=lambda x:x[2]) #error
我只想要前20名元素。 所以,而不是排序我想用堆。 我得到的错误是,
Traceback (most recent call last):
File "D:eclipse_progsDaDvIMDBAssignment1.py", line 42, in <module>
it=heapq.nlargest(2,enumerate(allrows),key=lambda x:x[2])
File "C:Python27libheapq.py", line 470, in nlargest
result = _nlargest(n, it)
File "D:eclipse_progsDaDvIMDBAssignment1.py", line 42, in <lambda>
it=heapq.nlargest(2,enumerate(allrows),key=lambda x:x[2])
IndexError: tuple index out of range
我可以知道为什么我收到错误以及如何解决它。 有没有使用heapq的任何属性,我缺少。
enumerate()
返回一个超过2元组的迭代。 因此,在第二个例子中访问x[2]
总是会超出范围(唯一有效的索引是0和1)。
要使第二个示例等同于第一个示例,应该直接传递allrows
,而不是使用enumerate()
:
it = heapq.nlargest(20, allrows, key=lambda x:x[2])
如果你需要保留原始的索引, enumerate()
就是要走的路。 但是,在关键函数中还需要额外的间接级别:
it = heapq.nlargest(20, enumerate(allrows), key=lambda x:x[1][2])
^^^^^^^^^ ^^^
感谢NPE来解决问题,作为替代答案,您可以将所有行与itertools.chain()
连接在一起,并获得具有排序功能的前20个元素,这些元素比heapq
具有更高的性能:
from itertools import chain
sorted(chain(*allrows))[-20:]
如果您试图查找相对较少的项目,则nlargest()
和nsmallest()
函数是最合适的。 如果您只是试图找到单个最小或最大的项目(N=1)
,则使用min()
和max()
会更快。 同样,如果N与集合本身的大小大致相同,则首先对它进行排序并采取切片(即,使用sorted(items)[:N]
或sorted(items)[-N:]
)通常会更快。
上一篇: How can I get n largest lists from a list of lists in python
下一篇: How to integrate npm/gulp/bower building process into sbt?