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

Sun*_*y88 138 python

如果我想要列表中的最大值,我可以写max(List),但如果我还需要最大值的索引怎么办?

我可以这样写:

maximum=0
for i,value in enumerate(List):
    if value>maximum:
        maximum=value
        index=i
Run Code Online (Sandbox Code Playgroud)

但这对我来说看起来很乏味.

如果我写:

List.index(max(List))
Run Code Online (Sandbox Code Playgroud)

然后它将迭代列表两次.

有没有更好的办法?

Esc*_*alo 301

我认为接受的答案很棒,但为什么不明确地这样做呢?我觉得更多的人会理解你的代码,这与PEP 8一致:

max_value = max(my_list)
max_index = my_list.index(max_value)
Run Code Online (Sandbox Code Playgroud)

这种方法也比接受的答案快三倍:

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)
Run Code Online (Sandbox Code Playgroud)

在我的电脑中运行的结果:

Explicit: 8.07 usec/pass
Implicit: 22.86 usec/pass
Run Code Online (Sandbox Code Playgroud)

其他套装:

Explicit: 6.80 usec/pass
Implicit: 19.01 usec/pass
Run Code Online (Sandbox Code Playgroud)

  • @ Sunny88:对于一个简单的数字列表,简单的方法更快.如果你在这种情况下表现出色,我建议使用`numpy.argmax()`,这在我的机器上要快30倍.如果列表包含比单纯数字更复杂的对象,我的答案中的方法可以变得更快.该方法的另一个优点是它可以用于任意迭代器,而不仅仅用于列表. (14认同)
  • 如果有重复的最大值,list.index()会不会有问题? (8认同)
  • 没想到它会更快.即使我用"l = [random.random()for _ in xrange(10000000)] + [2]"替换l也更快,这保证了最后一个元素是最大的. (2认同)

Sve*_*ach 174

有很多选择,例如:

import operator
index, value = max(enumerate(my_list), key=operator.itemgetter(1))
Run Code Online (Sandbox Code Playgroud)

  • @SvenMarnach为什么不使用key = lambda e:e [1]代替,从而避免导入? (6认同)
  • @lifebalance使用`itemgetter()`更快,避免导入不是值得追求的目标.在某些情况下,避免外部依赖性是值得的,但从标准库导入是不成问题的. (6认同)
  • 嗯,我在其他地方也看到了这一点,但我认为它只会返回一个值,而不是元组。 (2认同)

por*_*ast 19

这个答案比@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 )
Run Code Online (Sandbox Code Playgroud)

我的电脑上的结果:

Npmax: 8.78 msec/pass
Explicit: 290.01 msec/pass
Implicit: 790.27 msec/pass
Run Code Online (Sandbox Code Playgroud)

  • 好吧,`numpy.argmax` 的速度看起来很惊人,除非你让它处理一个 *标准* python 列表。然后速度介于显式和隐式版本之间。我猜 `np.array` 不仅创建了一个列表,而且还保存了一些额外的信息 - 例如最小值和最大值(只是一个假设)。 (2认同)

Sun*_*pil 16

使用Python内置库,它非常简单:

a = [2, 9, -10, 5, 18, 9] 
max(xrange(len(a)), key = lambda x: a[x])
Run Code Online (Sandbox Code Playgroud)


Lui*_*eva 10

max([(v,i) for i,v in enumerate(my_list)])
Run Code Online (Sandbox Code Playgroud)

  • 谢谢,你可以在答案中发帖吗?在计算最大值时,`max`如何知道将每个元组的第一项(`v`)考虑在内? (4认同)

小智 6

我建议一种非常简单的方法:

import numpy as np
l = [10, 22, 8, 8, 11]
print(np.argmax(l))
print(np.argmin(l))
Run Code Online (Sandbox Code Playgroud)

希望能帮助到你。