Filtruj maksymalnie 20 wartości z listy liczb całkowitych

Question

Chciałbym utworzyć listę maxValues ​​ zawierającą górne 20 wartości z listy liczb całkowitych lst.

maxValues = [] 
for i in range(20): 
    maxValues.append(max(lst)) 
    lst.remove(max(lst)) 

Czy istnieje bardziej kompaktowy kod do realizacji tego zadania, a nawet funkcja wbudowana?

Answer 1

Jest heapq.nlargest():

maxvalues = heapq.nlargest(20, lst) 

Z Doc:

heapq.nlargest(n, iterable, key=None)

zwróci lista z n największymi elementami z th e zestaw danych zdefiniowany przez iterable. key, jeśli jest zapewnione, określa funkcję jednego argumentu, który jest używany, aby wyodrębnić klucza porównania z każdego elementu w iterable: key=str.lower Odpowiednik: sorted(iterable, key=key, reverse=True)[:n]

Albo w ten sam sposób wykorzystywać heapq.nsmallest() jeśli chcesz najmniejszy.

UWAGAfrom the doc:

Te dwie funkcje [nlargest i nsmallest] optymalnie dostosowane do mniejszych wartości n. W przypadku większych wartości bardziej efektywne jest użycie funkcji sorted(). Ponadto, gdy n==1, bardziej efektywne jest korzystanie z wbudowanych funkcji min() i max().

Answer 2

sorted(lst)[-20:] 

to najkrótsza rzecz, jaką mogę wymyślić. Prawdopodobnie też jest szybszy.

(edycja: najpierw spróbować uznało min zamiast max)