一千萬個為什麽

搜索

Pythonic方法在列表中查找最大值及其索引?

如果我想要列表中的最大值,我可以寫 max(List),但是如果我還需要最大值的索引呢?

我可以寫這樣的東西:

maximum=0
for i,value in enumerate(List):
    if value>maximum:
        maximum=value
        index=i

但這對我來說看起來很乏味。

如果我寫:

List.index(max(List))

然後它將叠代列表兩次。

有沒有更好的辦法?

最佳答案

我認為接受的答案很好,但為什麽不明確地這樣做呢?我覺得更多的人會理解你的代碼,這與PEP 8一致:

max_value = max(my_list)
max_index = my_list.index(max_value)

這種方法也比接受的答案快三倍:

import random
from datetime import datetime
import operator

def explicit(l):
    max_val = max(l)
    max_idx = l.index(max_val)
    return max_idx, max_val

def implicit(l):
    max_idx, max_val = max(enumerate(l), key=operator.itemgetter(1))
    return max_idx, max_val

if __name__ == "__main__":
    from timeit import Timer
    t = Timer("explicit(l)", "from __main__ import explicit, implicit; "
          "import random; import operator;"
          "l = [random.random() for _ in xrange(100)]")
    print "Explicit: %.2f usec/pass" % (1000000 * t.timeit(number=100000)/100000)

    t = Timer("implicit(l)", "from __main__ import explicit, implicit; "
          "import random; import operator;"
          "l = [random.random() for _ in xrange(100)]")
    print "Implicit: %.2f usec/pass" % (1000000 * t.timeit(number=100000)/100000)

在我的計算機中運行的結果:

Explicit: 8.07 usec/pass
Implicit: 22.86 usec/pass

其他套裝:

Explicit: 6.80 usec/pass
Implicit: 19.01 usec/pass

轉載註明原文: Pythonic方法在列表中查找最大值及其索引?