파이썬 목록에서 n 개의 가장 큰 목록을 얻는 방법

Question

목록 목록에서 nlargest 요소를 얻기 위해 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_progs\DaDv\IMDB\Assignment1.py", line 42, in <module> 
    it=heapq.nlargest(2,enumerate(allrows),key=lambda x:x[2]) 
File "C:\Python27\lib\heapq.py", line 470, in nlargest 
    result = _nlargest(n, it) 
File "D:\eclipse_progs\DaDv\IMDB\Assignment1.py", line 42, in <lambda> 
    it=heapq.nlargest(2,enumerate(allrows),key=lambda x:x[2]) 
IndexError: tuple index out of range 

오류가 발생하는 이유와 해결 방법을 알 수 있습니까? heapq를 사용하는 속성이 누락되었습니다.

Answer 1

enumerate()은 2 튜플에 대한 반복 가능을 반환합니다. 따라서 두 번째 예제에서 x[2]에 액세스하는 것은 항상 범위를 벗어납니다 (유일한 유효한 인덱스는 0과 1입니다).

it = heapq.nlargest(20, allrows, key=lambda x:x[2]) 

원래 인덱스를 보존해야하는 경우, enumerate() 갈 방법은 다음과 같습니다

당신은 enumerate()를 사용하는 대신 직접 allrows을 통과해야한다, 처음에 두 번째 예제와 동등한를 확인하십시오. 그러나, 당신은 또한 핵심 기능에 간접적 인 추가 레벨을 필요

it = heapq.nlargest(20, enumerate(allrows), key=lambda x:x[1][2]) 
         ^^^^^^^^^       ^^^ 

Answer 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:])를 사용하는 것이 더 빠릅니다.