индекс heapq.nlargest возвращенного результата в исходной последовательности

Как я возвращаю индекс в исходном списке энных самых больших объектов повторяемого

heapq.nlargest(2, [100, 2, 400, 500, 400])

output = [(3,500), (2, 400)]

Это уже стоило мне пары часов. Я не могу понять это.

5
задан miku 29 June 2010 в 10:42
поделиться

2 ответа

>>> seq = [100, 2, 400, 500, 400]
>>> heapq.nlargest(2, enumerate(seq), key=lambda x: x[1])
[(3, 500), (2, 400)]
22
ответ дан 18 December 2019 в 06:21
поделиться

Вы можете использовать list.index в сочетании с map , что быстро для малых n (остерегайтесь list.index возвращает индекс в списке первого элемента, значение которого равно x):

>>> iterable = [100, 2, 400, 500, 400]
>>> map(iterable.index, heapq.nlargest(2, iterable))
[3, 2]

Чтобы увидеть связанные значения ...

>>> map(lambda n: (n, iterable.index(n)), heapq.nlargest(2, iterable))
[(500, 3), (400, 2)]

Для большего n см. Сообщение @ SilentGhost.


Редактировать: Сравнение некоторых решений:

#!/usr/bin/env python
import heapq
from timeit import Timer

seq = [100, 2, 400, 500, 400]

def a(seq):
    """returns [(3, 500), (2, 400)]"""
    return heapq.nlargest(2, enumerate(seq), key=lambda x: x[1])

def b(seq):
    """returns [3, 2]"""
    return map(seq.index, heapq.nlargest(2, seq))

def c(seq):
    """returns [(500, 3), (400, 2)]"""
    map(lambda n: (n, seq.index(n)), heapq.nlargest(2, seq))

if __name__ == '__main__':
    _a = Timer("a(seq)", "from __main__ import a, seq")
    _b = Timer("b(seq)", "from __main__ import b, seq")
    _c = Timer("c(seq)", "from __main__ import c, seq") 

    loops = 1000000

    print _a.timeit(number=loops)
    print _b.timeit(number=loops)
    print _c.timeit(number=loops)

    # Core i5, 2.4GHz, Python 2.6, Darwin
    # 8.92712688446
    # 5.64332985878
    # 6.50824809074
3
ответ дан 18 December 2019 в 06:21
поделиться
Другие вопросы по тегам:

Похожие вопросы: