如何在python中找到列表的第二个最大值的位置？

Question

我试图从列表中找到n个最大值,然后在列表中打印出它们的位置.

如果我只关注最大值.价值,它看起来像这样:

>>>>fr = [8,4,1,1,12]

>>>>print fr.index(max(fr))

4

但是我的目标是获得如下输出:4,0,1,如果它是n = 3

谢谢您的帮助!

Answer 1

heapq.nlargest与key =一起使用fr.__getitem__:

>>> import heapq
>>> fr = [8,4,1,1,12]
>>> heapq.nlargest(3, xrange(len(fr)), key=fr.__getitem__)
[4, 0, 1]

如果您想要值本身,那么:

>>> heapq.nlargest(3, fr)
[12, 8, 4]

Answer 2

另一种方式是：

[fr.index(x) for x in sorted(fr, reverse=True)[:3]]

当我们比较两者的速度时......

import heapq

fr = [8, 4, 1, 1, 12]


def method_one():
    for i in xrange(10000):
        res = [fr.index(x) for x in sorted(fr, reverse=True)[:3]]


def method_two():
    for i in xrange(10000):
        heapq.nlargest(3, xrange(len(fr)), key=fr.__getitem__)


if __name__ == '__main__':
    import timeit

    print timeit.repeat(stmt='method_one()',
                    setup='from __main__ import method_one',
                    number=100)
    print timeit.repeat(stmt='method_two()',
                    setup='from __main__ import method_two',
                    number=100)

我们得到：

[1.1253619194030762, 1.1268768310546875, 1.128382921218872]
[2.5129621028900146, 2.529547929763794, 2.492828130722046]