在列表中以Pythonic方式找到最大值及其索引？

Question

如果我想找出一个列表中的最大值，我可以直接写 max(List)，但是如果我还需要知道这个最大值的索引位置该怎么办呢？

我可以写类似这样的代码：

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

不过我觉得这样写起来有点麻烦。

如果我写：

List.index(max(List))

那么这个方法会让列表被遍历两次。

有没有更好的办法呢？

Answer 1

这个答案的速度比@Escualo快了33倍，前提是列表非常大，并且假设它已经是一个np.array()。我不得不减少测试的运行次数，因为测试是针对10000000个元素，而不仅仅是100个。

import random
from datetime import datetime
import operator
import numpy as np

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

def npmax(l):
    max_idx = np.argmax(l)
    max_val = l[max_idx]
    return (max_idx, max_val)

if __name__ == "__main__":
    from timeit import Timer

t = Timer("npmax(l)", "from __main__ import explicit, implicit, npmax; "
      "import random; import operator; import numpy as np;"
      "l = np.array([random.random() for _ in xrange(10000000)])")
print "Npmax: %.2f msec/pass" % (1000  * t.timeit(number=10)/10 )

t = Timer("explicit(l)", "from __main__ import explicit, implicit; "
      "import random; import operator;"
      "l = [random.random() for _ in xrange(10000000)]")
print "Explicit: %.2f msec/pass" % (1000  * t.timeit(number=10)/10 )

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

我电脑上的测试结果：

Npmax: 8.78 msec/pass
Explicit: 290.01 msec/pass
Implicit: 790.27 msec/pass

Answer 2

我觉得这个被认可的答案很好，但为什么不把它写得更清楚一些呢？这样更多的人会理解你的代码，这也符合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

Answer 3

有很多选择，比如：

import operator
index, value = max(enumerate(my_list), key=operator.itemgetter(1))