Python: take max N elements from some list

前端 未结 4 1384
后悔当初
后悔当初 2020-12-05 13:41

Is there some function which would return me the N highest elements from some list?

I.e. if max(l) returns the single highest element, sth. like m

4条回答
  •  独厮守ぢ
    2020-12-05 13:49

    Start with the first 10 from L, call that X. Note the minimum value of X.

    Loop over L[i] for i over the rest of L.

    If L[i] is greater than min(X), drop min(X) from X and insert L[i]. You may need to keep X as a sorted linked list and do an insertion. Update min(X).

    At the end, you have the 10 largest values in X.

    I suspect that will be O(kN) (where k is 10 here) since insertion sort is linear. Might be what gsl uses, so if you can read some C code:

    http://www.gnu.org/software/gsl/manual/html_node/Selecting-the-k-smallest-or-largest-elements.html

    Probably something in numpy that does this.

提交回复
热议问题